580d49
diff --git a/mime.c b/mime.c
580d49
index 45de80a..ecac85d 100644
580d49
--- a/mime.c
580d49
+++ b/mime.c
580d49
@@ -1109,16 +1109,34 @@ fromhdr_end:
580d49
 }
580d49
 
580d49
 /*
580d49
+ * return length of this UTF-8 codepoint in bytes
580d49
+ */
580d49
+static size_t
580d49
+codepointsize(char tc)
580d49
+{
580d49
+	int rv = 0;
580d49
+	if ( ! ( tc & 0x80 ) )
580d49
+		return 1;
580d49
+	while ( tc & 0x80 )
580d49
+	{
580d49
+		rv++;
580d49
+		tc = tc<<1;
580d49
+	}
580d49
+	return rv;
580d49
+}
580d49
+
580d49
+/*
580d49
  * Convert header fields to RFC 1522 format and write to the file fo.
580d49
  */
580d49
 static size_t
580d49
 mime_write_tohdr(struct str *in, FILE *fo)
580d49
 {
580d49
 	char *upper, *wbeg, *wend, *charset, *lastwordend = NULL, *lastspc, b,
580d49
-		*charset7;
580d49
+		*charset7, *cp;
580d49
 	struct str cin, cout;
580d49
-	size_t sz = 0, col = 0, wr, charsetlen, charset7len;
580d49
+	size_t sz = 0, col = 0, wr, charsetlen, charset7len, cpsz;
580d49
 	int quoteany, mustquote, broken,
580d49
+		maxin, maxout, curin, cps,
580d49
 		maxcol = 65 /* there is the header field's name, too */;
580d49
 
580d49
 	upper = in->s + in->l;
580d49
@@ -1134,41 +1152,75 @@ mime_write_tohdr(struct str *in, FILE *fo)
580d49
 		if (mustquote_hdr(wbeg, wbeg == in->s, wbeg == &upper[-1]))
580d49
 			quoteany++;
580d49
 	}
580d49
+
580d49
+	/*
580d49
+	 * rfc2047 says we cannot split multi-byte characters over
580d49
+	 * encoded words, so we need to know if we're a multi-byte
580d49
+	 * source stream (UTF-8 specifically) or just an 8 bit
580d49
+	 * stream like ISO-8859-15
580d49
+	 * so test beginning of charset since it is valid to include
580d49
+	 * language in charset "UTF-8*DE" etc as per rfc 2184/2231
580d49
+	 */
580d49
+	char *thisset = b&0200 ? charset : charset7;
580d49
+	int is_utf8 = ( strncasecmp( thisset, "utf-8", 5 ) == 0 );
580d49
+
580d49
 	if (2 * quoteany > in->l) {
580d49
 		/*
580d49
 		 * Print the entire field in base64.
580d49
 		 */
580d49
-		for (wbeg = in->s; wbeg < upper; wbeg = wend) {
580d49
+		for (wbeg = in->s; wbeg < upper; ) {
580d49
 			wend = upper;
580d49
 			cin.s = wbeg;
580d49
-			for (;;) {
580d49
-				cin.l = wend - wbeg;
580d49
-				if (cin.l * 4/3 + 7 + charsetlen
580d49
-						< maxcol - col) {
580d49
-					fprintf(fo, "=?%s?B?",
580d49
-						b&0200 ? charset : charset7);
580d49
-					wr = mime_write_tob64(&cin, fo, 1);
580d49
-					fwrite("?=", sizeof (char), 2, fo);
580d49
-					wr += 7 + charsetlen;
580d49
-					sz += wr, col += wr;
580d49
-					if (wend < upper) {
580d49
-						fwrite("\n ", sizeof (char),
580d49
-								2, fo);
580d49
-						sz += 2;
580d49
-						col = 0;
580d49
-						maxcol = 76;
580d49
+			/*
580d49
+			 * we calculate the maximum number of bytes
580d49
+			 * we can use on this output line, and then what
580d49
+			 * this equates to as base64 encoded source bytes
580d49
+			 */
580d49
+			maxout = maxcol - col - 7 - charsetlen;
580d49
+			maxin = (maxout - (maxout & 0x03)) * 3/4;
580d49
+
580d49
+			/* short enough to finish ? */
580d49
+			if (maxin > upper - wbeg )
580d49
+			{
580d49
+				curin = upper - wbeg;
580d49
+				wbeg += curin;
580d49
+			}else
580d49
+			{
580d49
+				if (is_utf8)
580d49
+				{
580d49
+					/*
580d49
+					 * now scan the input from the beginning
580d49
+					 * to see how many codepoints will fit
580d49
+					 */
580d49
+					curin = 0;
580d49
+					while (curin < maxin
580d49
+						&& (cpsz = codepointsize(*wbeg)) <= (maxin - curin))
580d49
+					{
580d49
+						curin += cpsz;
580d49
+						wbeg += cpsz;
580d49
 					}
580d49
-					break;
580d49
-				} else {
580d49
-					if (col) {
580d49
-						fprintf(fo, "\n ");
580d49
-						sz += 2;
580d49
-						col = 0;
580d49
-						maxcol = 76;
580d49
-					} else
580d49
-						wend -= 4;
580d49
+				}else
580d49
+				{
580d49
+					curin = maxin;
580d49
+					wbeg += maxin;
580d49
 				}
580d49
 			}
580d49
+			cin.l = curin;
580d49
+			fprintf(fo, "%s=?%s?B?", (cin.s != in->s) ? " " : "", thisset );
580d49
+			wr = mime_write_tob64(&cin, fo, 1);
580d49
+
580d49
+			if (wbeg < upper)
580d49
+			{
580d49
+				wr += fwrite("?=\n ", sizeof (char), 4, fo) * sizeof (char);
580d49
+			}else
580d49
+			{
580d49
+				wr += fwrite("?=", sizeof (char), 2, fo) * sizeof (char);
580d49
+			}
580d49
+
580d49
+			/* and shuffle pointers and counts */
580d49
+			col = 1;
580d49
+			maxcol = 76;
580d49
+			sz += wr + 7 + charsetlen + ((cin.s != in->s) ? 1 : 0 );
580d49
 		}
580d49
 	} else {
580d49
 		/*
580d49
@@ -1243,7 +1295,29 @@ mime_write_tohdr(struct str *in, FILE *fo)
580d49
 								maxcol -= wbeg -
580d49
 									lastspc;
580d49
 						} else {
580d49
-							wend -= 4;
580d49
+							if (is_utf8)
580d49
+							{
580d49
+								/*
580d49
+								 * make sure wend is not pointing to
580d49
+								 * the middle of a codepoint
580d49
+								 */
580d49
+								cp = wend;
580d49
+								while (--cp > wbeg)
580d49
+								{
580d49
+									cps = codepointsize(*cp);
580d49
+									if (cps > 1)
580d49
+									{
580d49
+										if (wend - cp - cps > 4)
580d49
+											wend -= 4;
580d49
+										else
580d49
+											wend = cp;
580d49
+										break;
580d49
+									}
580d49
+								}
580d49
+								if (cp == wbeg)
580d49
+									wend -= 4;
580d49
+							} else
580d49
+								wend -= 4;
580d49
 						}
580d49
 						free(cout.s);
580d49
 					}