Commit bd25a302d3232dea375f226602fdcdb3a83abcdc

Sven Strickroth 2013-02-01T22:22:26

Improved error handling Signed-off-by: Sven Strickroth <email@cs-ware.de>

diff --git a/src/errors.c b/src/errors.c
index 1ab2894..56f07b6 100644
--- a/src/errors.c
+++ b/src/errors.c
@@ -64,8 +64,15 @@ void giterr_set(int error_class, const char *string, ...)
 				int utf8_size = WideCharToMultiByte(CP_UTF8, 0, lpMsgBuf, -1, NULL, 0, NULL, NULL);
 
 				char *lpMsgBuf_utf8 = git__malloc(utf8_size * sizeof(char));
-				GITERR_CHECK_ALLOC(lpMsgBuf_utf8);
-				WideCharToMultiByte(CP_UTF8, 0, lpMsgBuf, -1, lpMsgBuf_utf8, utf8_size, NULL, NULL);
+				if (lpMsgBuf_utf8 == NULL) {
+					LocalFree(lpMsgBuf);
+					return;
+				}
+				if (!WideCharToMultiByte(CP_UTF8, 0, lpMsgBuf, -1, lpMsgBuf_utf8, utf8_size, NULL, NULL)) {
+					LocalFree(lpMsgBuf);
+					git__free(lpMsgBuf_utf8);
+					return;
+				}
 
 				git_buf_PUTS(&buf, ": ");
 				git_buf_puts(&buf, lpMsgBuf_utf8);
diff --git a/src/netops.c b/src/netops.c
index c5554ef..86579c7 100644
--- a/src/netops.c
+++ b/src/netops.c
@@ -48,8 +48,16 @@ static void net_set_error(const char *str)
 
 	int utf8_size = WideCharToMultiByte(CP_UTF8, 0, err_str, -1, NULL, 0, NULL, NULL);
 	char * err_str_utf8 = git__malloc(utf8_size * sizeof(char));
-	GITERR_CHECK_ALLOC(err_str_utf8);
-	WideCharToMultiByte(CP_UTF8, 0, err_str, -1, err_str_utf8, utf8_size, NULL, NULL);
+	if (err_str_utf8 == NULL) {
+		LocalFree(err_str);
+		return;
+	}
+
+	if (!WideCharToMultiByte(CP_UTF8, 0, err_str, -1, err_str_utf8, utf8_size, NULL, NULL)) {
+		LocalFree(err_str);
+		git__free(err_str_utf8);
+		return;
+	}
 
 	giterr_set(GITERR_NET, "%s: %s", str, err_str_utf8);
 	LocalFree(err_str);