i3status/src/print_time.c
Orestis Floros 980164ccca
Fix various memory leaks
Fixes #362
2020-05-01 13:55:19 +02:00

89 lines
2.2 KiB
C

// vim:ts=4:sw=4:expandtab
#include <config.h>
#include <time.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
#include <locale.h>
#include <yajl/yajl_gen.h>
#include <yajl/yajl_version.h>
#include "i3status.h"
#define STRING_SIZE 50
static bool local_timezone_init = false;
static const char *local_timezone = NULL;
static const char *current_timezone = NULL;
void set_timezone(const char *tz) {
if (!local_timezone_init) {
/* First call, initialize. */
local_timezone = getenv("TZ");
local_timezone_init = true;
}
if (tz == NULL || tz[0] == '\0') {
/* User wants localtime. */
tz = local_timezone;
}
if (tz != current_timezone) {
if (tz) {
setenv("TZ", tz, 1);
} else {
unsetenv("TZ");
}
current_timezone = tz;
}
tzset();
}
void print_time(yajl_gen json_gen, char *buffer, const char *title, const char *format, const char *tz, const char *locale, const char *format_time, bool hide_if_equals_localtime, time_t t) {
char *outwalk = buffer;
struct tm local_tm, tm;
if (title != NULL)
INSTANCE(title);
set_timezone(NULL);
localtime_r(&t, &local_tm);
set_timezone(tz);
localtime_r(&t, &tm);
// When hide_if_equals_localtime is true, compare local and target time to display only if different
time_t local_t = mktime(&local_tm);
double diff = difftime(local_t, t);
if (hide_if_equals_localtime && diff == 0.0) {
goto out;
}
if (locale != NULL) {
setlocale(LC_ALL, locale);
}
char string_time[STRING_SIZE];
if (format_time == NULL) {
strftime(string_time, sizeof(string_time), format, &tm);
maybe_escape_markup(string_time, &outwalk);
} else {
strftime(string_time, sizeof(string_time), format_time, &tm);
placeholder_t placeholders[] = {
{.name = "%time", .value = string_time}};
const size_t num = sizeof(placeholders) / sizeof(placeholder_t);
buffer = format_placeholders(format_time, &placeholders[0], num);
}
if (locale != NULL) {
setlocale(LC_ALL, "");
}
out:
*outwalk = '\0';
OUTPUT_FULL_TEXT(buffer);
if (format_time != NULL) {
free(buffer);
}
}