Fix invalid timestamps crashing on Windows

This commit is contained in:
TingPing 2013-11-02 02:05:55 -04:00
parent 7e6f37b4cd
commit af248ce2c1
5 changed files with 82 additions and 20 deletions

View File

@ -542,7 +542,7 @@ log_create_pathname (char *servname, char *channame, char *netname)
/* insert time/date */ /* insert time/date */
now = time (NULL); now = time (NULL);
tm = localtime (&now); tm = localtime (&now);
strftime (fnametime, sizeof (fnametime), fname, tm); strftime_validated (fnametime, sizeof (fnametime), fname, tm);
/* create final path/filename */ /* create final path/filename */
if (logmask_is_fullpath ()) if (logmask_is_fullpath ())
@ -649,14 +649,7 @@ get_stamp_str (char *fmt, time_t tim, char **ret)
fmt = loc; fmt = loc;
} }
len = strftime (dest, sizeof (dest), fmt, localtime (&tim)); len = strftime_validated (dest, sizeof (dest), fmt, localtime (&tim));
#ifdef WIN32
if (!len)
{
/* use failsafe format until a correct one is specified */
len = strftime (dest, sizeof (dest), "[%H:%M:%S]", localtime (&tim));
}
#endif
if (len) if (len)
{ {
if (prefs.utf8_locale) if (prefs.utf8_locale)

View File

@ -25,6 +25,7 @@
#include <stdio.h> #include <stdio.h>
#include <string.h> #include <string.h>
#include <stdlib.h> #include <stdlib.h>
#include <time.h>
#include <sys/types.h> #include <sys/types.h>
#include <sys/stat.h> #include <sys/stat.h>
@ -2287,3 +2288,80 @@ challengeauth_response (char *username, char *password, char *challenge)
return (char *) digest; return (char *) digest;
} }
#endif #endif
/**
* \brief Wrapper around strftime for Windows
*
* Prevents crashing when using an invalid format by escaping them.
*
* Behaves the same as strftime with the addition that
* it returns 0 if the escaped format string is too large.
*
* Based upon work from znc-msvc project.
*/
size_t
strftime_validated (char *dest, size_t destsize, const char *format, const struct tm *time)
{
#ifndef WIN32
return strftime (dest, destsize, format, time);
#else
char safe_format[64];
const char *p = format;
int i = 0;
if (strlen (format) >= sizeof(safe_format))
return 0;
memset (safe_format, 0, sizeof(safe_format));
while (*p)
{
if (*p == '%')
{
int has_hash = (*(p + 1) == '#');
char c = *(p + (has_hash ? 2 : 1));
if (i >= sizeof (safe_format))
return 0;
switch (c)
{
case 'a': case 'A': case 'b': case 'B': case 'c': case 'd': case 'H': case 'I': case 'j': case 'm': case 'M':
case 'p': case 'S': case 'U': case 'w': case 'W': case 'x': case 'X': case 'y': case 'Y': case 'z': case 'Z':
case '%':
/* formatting code is fine */
break;
default:
/* replace bad formatting code with itself, escaped, e.g. "%V" --> "%%V" */
g_strlcat (safe_format, "%%", sizeof(safe_format));
i += 2;
p++;
break;
}
/* the current loop run will append % (and maybe #) and the next one will do the actual char. */
if (has_hash)
{
safe_format[i] = *p;
p++;
i++;
}
if (c == '%')
{
safe_format[i] = *p;
p++;
i++;
}
}
if (*p)
{
safe_format[i] = *p;
p++;
i++;
}
}
return strftime (dest, destsize, safe_format, time);
#endif
}

View File

@ -83,5 +83,5 @@ char *encode_sasl_pass_plain (char *user, char *pass);
char *encode_sasl_pass_blowfish (char *user, char *pass, char *data); char *encode_sasl_pass_blowfish (char *user, char *pass, char *data);
char *encode_sasl_pass_aes (char *user, char *pass, char *data); char *encode_sasl_pass_aes (char *user, char *pass, char *data);
char *challengeauth_response (char *username, char *password, char *challenge); char *challengeauth_response (char *username, char *password, char *challenge);
size_t strftime_validated (char *dest, size_t destsize, const char *format, const struct tm *time);
#endif #endif

View File

@ -2114,7 +2114,6 @@ setup_apply (struct hexchatprefs *pr)
PangoFontDescription *old_desc; PangoFontDescription *old_desc;
PangoFontDescription *new_desc; PangoFontDescription *new_desc;
char buffer[4 * FONTNAMELEN + 1]; char buffer[4 * FONTNAMELEN + 1];
time_t rawtime;
#endif #endif
int new_pix = FALSE; int new_pix = FALSE;
int noapply = FALSE; int noapply = FALSE;
@ -2192,14 +2191,6 @@ setup_apply (struct hexchatprefs *pr)
g_free (old_desc); g_free (old_desc);
g_free (new_desc); g_free (new_desc);
*/ */
/* workaround for strftime differences between POSIX and MSVC */
time (&rawtime);
if (!strftime (buffer, sizeof (buffer), prefs.hex_stamp_text_format, localtime (&rawtime)) || !strftime (buffer, sizeof (buffer), prefs.hex_stamp_log_format, localtime (&rawtime)))
{
fe_message (_("Invalid time stamp format! See the strftime MSDN article for details."), FE_MSG_ERROR);
}
#endif #endif
if (prefs.hex_irc_real_name[0] == 0) if (prefs.hex_irc_real_name[0] == 0)

View File

@ -121,7 +121,7 @@ fe_new_window (struct session *sess, int focus)
static int static int
get_stamp_str (time_t tim, char *dest, int size) get_stamp_str (time_t tim, char *dest, int size)
{ {
return strftime (dest, size, prefs.hex_stamp_text_format, localtime (&tim)); return strftime_validated (dest, size, prefs.hex_stamp_text_format, localtime (&tim));
} }
static int static int