1 files changed, 141 insertions, 0 deletions
diff --git a/lib/dpkg/strwide.c b/lib/dpkg/strwide.c
new file mode 100644
index 0000000..0a510b7
--- /dev/null
+++ b/lib/dpkg/strwide.c
@@ -0,0 +1,141 @@
+/*
+ * libdpkg - Debian packaging suite library routines
+ * strwide.c - wide character string handling routines
+ *
+ * Copyright © 2004 Changwoo Ryu <cwryu@debian.org>
+ * Copyright © 2012-2013 Guillem Jover <guillem@debian.org>
+ *
+ * This is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+#include <config.h>
+#include <compat.h>
+
+#include <string.h>
+#include <stdlib.h>
+#ifdef ENABLE_NLS
+#include <wchar.h>
+#endif
+
+#include <dpkg/i18n.h>
+#include <dpkg/dpkg.h>
+#include <dpkg/string.h>
+
+/**
+ * Compute the screen width of a string.
+ *
+ * @param str The multibyte string.
+ *
+ * @return The width of the string.
+ */
+int
+str_width(const char *str)
+{
+#ifdef ENABLE_NLS
+	mbstate_t state;
+	wchar_t *wcs;
+	const char *mbs = str;
+	size_t len, res;
+	int width;
+
+	len = strlen(str) + 1;
+	wcs = m_malloc(sizeof(wcs[0]) * len);
+
+	memset(&state, 0, sizeof(state));
+
+	res = mbsrtowcs(wcs, &mbs, len, &state);
+	if (res == (size_t)-1) {
+#ifdef DPKG_UNIFORM_ENCODING
+		ohshit(_("cannot convert multibyte string '%s' "
+		         "to a wide-character string"), str);
+#else
+		/* Cannot convert, fallback to ASCII method. */
+		free(wcs);
+		return strlen(str);
+#endif
+	}
+
+	width = wcswidth(wcs, res);
+
+	free(wcs);
+
+	return width;
+#else
+	return strlen(str);
+#endif
+}
+
+/**
+ * Generate the crop values for a string given a maximum screen width.
+ *
+ * This function analyzes the string passed and computes the correct point
+ * where to crop the string, returning the amount of string and maximum
+ * bytes to use for padding for example.
+ *
+ * On NLS enabled builds, in addition the string will be cropped on any
+ * newline.
+ *
+ * @param str        The string to crop.
+ * @param max_width  The max screen width to use.
+ * @param[out] crop  The generated crop values for the string.
+ */
+void
+str_gen_crop(const char *str, int max_width, struct str_crop_info *crop)
+{
+#ifdef ENABLE_NLS
+	mbstate_t state;
+	size_t str_bytes;
+	int mbs_bytes = 0;
+	int mbs_width = 0;
+
+	str_bytes = strlen(str) + 1;
+	memset(&state, 0, sizeof(state));
+
+	for (;;) {
+		wchar_t wc;
+		int wc_width;
+		size_t mb_bytes;
+
+		mb_bytes = mbrtowc(&wc, str, str_bytes, &state);
+		if (mb_bytes == (size_t)-1 || mb_bytes == (size_t)-2) {
+#ifdef DPKG_UNIFORM_ENCODING
+			ohshit(_("cannot convert multibyte sequence '%s' "
+			         "to a wide character"), str);
+#else
+			/* Cannot convert, fallback to ASCII method. */
+			crop->str_bytes = crop->max_bytes = max_width;
+			return;
+#endif
+		}
+		if (mb_bytes == 0)
+			break;
+
+		wc_width = wcwidth(wc);
+		if (wc_width < 0)
+			break;
+		if (mbs_width + wc_width > max_width)
+			break;
+
+		mbs_width += wc_width;
+		mbs_bytes += mb_bytes;
+		str_bytes -= mb_bytes;
+		str += mb_bytes;
+	}
+
+	crop->str_bytes = mbs_bytes;
+	crop->max_bytes = mbs_bytes + max_width - mbs_width;
+#else
+	crop->str_bytes = crop->max_bytes = max_width;
+#endif
+}