SocketError: merge duplicate FormatMessage() calls

Make a class that contains the formatting code and the buffer.
This commit is contained in:
Max Kellermann 2013-01-15 22:32:13 +01:00
parent bc66dc45e6
commit 5bb2d64ddb

View File

@ -87,27 +87,49 @@ IsSocketErrorClosed(socket_error_t code)
#endif #endif
} }
/**
* Helper class that formats a socket error message into a
* human-readable string. On Windows, a buffer is necessary for this,
* and this class hosts the buffer.
*/
class SocketErrorMessage {
#ifdef WIN32
char msg[256];
#else
const char *const msg;
#endif
public:
#ifdef WIN32
explicit SocketErrorMessage(socket_error_t code=GetSocketError()) {
DWORD nbytes = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS |
FORMAT_MESSAGE_MAX_WIDTH_MASK,
NULL, code, 0,
(LPSTR)msg, sizeof(msg), NULL);
if (nbytes == 0)
strcpy(msg, "Unknown error");
}
#else
explicit SocketErrorMessage(socket_error_t code=GetSocketError())
:msg(g_strerror(code)) {}
#endif
operator const char *() const {
return msg;
}
};
static inline void static inline void
SetSocketError(GError **error_r, socket_error_t code) SetSocketError(GError **error_r, socket_error_t code)
{ {
#ifdef WIN32 #ifdef WIN32
if (error_r == NULL) if (error_r == NULL)
return; return;
char buffer[256];
DWORD nbytes = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS |
FORMAT_MESSAGE_MAX_WIDTH_MASK,
NULL, code, 0,
(LPSTR)buffer, sizeof(buffer), NULL);
const char *msg = nbytes > 0
? buffer
: "Unknown error";
g_set_error_literal(error_r, SocketErrorQuark(), code, msg);
#else
g_set_error_literal(error_r, SocketErrorQuark(), code,
g_strerror(code));
#endif #endif
const SocketErrorMessage msg(code);
g_set_error_literal(error_r, SocketErrorQuark(), code, msg);
} }
static inline void static inline void
@ -120,20 +142,8 @@ gcc_malloc
static inline GError * static inline GError *
NewSocketError(socket_error_t code) NewSocketError(socket_error_t code)
{ {
#ifdef WIN32 const SocketErrorMessage msg(code);
char buffer[256];
DWORD nbytes = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS |
FORMAT_MESSAGE_MAX_WIDTH_MASK,
NULL, code, 0,
(LPSTR)buffer, sizeof(buffer), NULL);
const char *msg = nbytes > 0
? buffer
: "Unknown error";
return g_error_new_literal(SocketErrorQuark(), code, msg); return g_error_new_literal(SocketErrorQuark(), code, msg);
#else
return g_error_new_literal(SocketErrorQuark(), code, g_strerror(code));
#endif
} }
gcc_malloc gcc_malloc