unix-time-0.5.0: cbits/conv.c

#include "config.h"

#if IS_LINUX
/* Linux cheats AC_CHECK_FUNCS(strptime_l), sigh. */
#define THREAD_SAFE 0
#define _XOPEN_SOURCE
#define _DEFAULT_SOURCE
#define _BSD_SOURCE
#elif HAVE_STRPTIME_L
#define THREAD_SAFE 1
#define _GNU_SOURCE
#else
#define THREAD_SAFE 0
#endif

#include <string.h>
#include <time.h>
#include <locale.h>
#include <stdlib.h>
#include <stdio.h>

#if defined(_WIN32)
#include "win_patch.h"
#endif // _WIN32

#if THREAD_SAFE
#if HAVE_XLOCALE_H
#include <xlocale.h>
#endif

locale_t c_locale = NULL;

void init_locale() {
    if (c_locale == NULL) c_locale = newlocale(LC_TIME_MASK, "C", NULL);
}
#else
void init_locale() {
    static int initialized = 0;
    if (initialized == 0) {
        setlocale(LC_TIME, "C");
        initialized = 1;
    }
}
#endif

/*
 * Set the value of the TZ environment variable to UTC
 * and return the old value.
 */
char *set_tz_utc() {
    char *tz;
    tz = getenv("TZ");
#if defined(_WIN32)
    _patch_setenv("TZ", "UTC", 1);
#else
    setenv("TZ", "", 1);
#endif
#if defined(HAVE_TZSET)
    tzset();
#endif
    return tz;
}

/*
 * Set the value of the TZ environment variable to tz or
 * unset the variable if tz is null;
 */
void set_tz(char *local_tz) {
    if (local_tz) {
#if defined(_WIN32)
      _patch_setenv("TZ", local_tz, 1);
#else
      setenv("TZ", local_tz, 1);
#endif
    } else {
#if defined(_WIN32)
      _patch_unsetenv("TZ");
#else
      unsetenv("TZ");
#endif
    }
#if defined(HAVE_TZSET)
    tzset();
#endif
}

time_t c_parse_unix_time(char *fmt, char *src) {
    struct tm dst;
    init_locale();
    memset(&dst, 0, sizeof(struct tm));
#if THREAD_SAFE
    strptime_l(src, fmt, &dst, c_locale);
#else
    strptime(src, fmt, &dst);
#endif
    return mktime(&dst);
}

#if !defined(HAVE_TIMEGM)
/* This part is for Solaris that doesn't have timegm.
 * The copyright notice of timegm and is_leap is as below:
 *
 * Copyright (c) 1997 Kungliga Tekniska H.gskolan
 * (Royal Institute of Technology, Stockholm, Sweden).
 * All rights reserved.
 */

static time_t
timegm (struct tm *tm)
{
  static const unsigned ndays[2][12] ={
    {31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31},
    {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31}};
  time_t res = 0;
  unsigned i;

  for (i = 70; i < tm->tm_year; ++i)
    res += isleap(i + 1900) ? 366 : 365;

  for (i = 0; i < tm->tm_mon; ++i)
    res += ndays[isleap(tm->tm_year + 1900)][i];
  res += tm->tm_mday - 1;
  res *= 24;
  res += tm->tm_hour;
  res *= 60;
  res += tm->tm_min;
  res *= 60;
  res += tm->tm_sec;
  return res;
}
#endif /* HAVE_TIMEGM */

time_t c_parse_unix_time_gmt(char *fmt, char *src) {
    struct tm dst;
    char *local_tz;
    init_locale();
    memset(&dst, 0, sizeof(struct tm));
    local_tz = set_tz_utc();
#if THREAD_SAFE
    strptime_l(src, fmt, &dst, c_locale);
#else
    strptime(src, fmt, &dst);
#endif
    set_tz(local_tz);
    return timegm(&dst);
}

size_t c_format_unix_time(char *fmt, time_t src, char* dst, int siz) {
    struct tm tim;
    init_locale();
    localtime_r(&src, &tim);
#if THREAD_SAFE
#if defined(_WIN32)
    return _patch_strftime_l(dst, siz, fmt, &tim, c_locale);
#else
    return strftime_l(dst, siz, fmt, &tim, c_locale);
#endif // _WIN32
#else
#if defined(_WIN32)
    return _patch_strftime(dst, siz, fmt, &tim);
#else
    return strftime(dst, siz, fmt, &tim);
#endif // _WIN32
#endif
}

size_t c_format_unix_time_gmt(char *fmt, time_t src, char* dst, int siz) {
    struct tm tim;
    char *local_tz;
    size_t dst_size;

    init_locale();
    gmtime_r(&src, &tim);

    local_tz = set_tz_utc();
#if THREAD_SAFE
#if defined(_WIN32)
    dst_size = _patch_strftime_l(dst, siz, fmt, &tim, c_locale);
#else
    dst_size = strftime_l(dst, siz, fmt, &tim, c_locale);
#endif // _WIN32
#else
#if defined(_WIN32)
    dst_size = _patch_strftime(dst, siz, fmt, &tim);
#else
    dst_size = strftime(dst, siz, fmt, &tim);
#endif // _WIN32
#endif
    set_tz(local_tz);
    return dst_size;
}