Permalink
Fetching contributors…
Cannot retrieve contributors at this time
243 lines (206 sloc) 5.49 KB
/*
* jim-clock.c
*
* Implements the clock command
*/
#include "jimautoconf.h"
/* For strptime() - currently nothing sets this */
#ifdef STRPTIME_NEEDS_XOPEN_SOURCE
#ifndef _XOPEN_SOURCE
#define _XOPEN_SOURCE 500
#endif
#endif
/* For timegm() */
#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <time.h>
#include <jim-subcmd.h>
#ifdef HAVE_SYS_TIME_H
#include <sys/time.h>
#endif
struct clock_options {
int gmt;
const char *format;
};
/* Parses the options ?-format string? ?-gmt boolean? and fills in *opts.
* Any options not present are not set.
* argc must be even.
*
* Returns JIM_OK or JIM_ERR and sets an error result.
*/
static int parse_clock_options(Jim_Interp *interp, int argc, Jim_Obj *const *argv, struct clock_options *opts)
{
static const char * const options[] = { "-gmt", "-format", NULL };
enum { OPT_GMT, OPT_FORMAT, };
int i;
for (i = 0; i < argc; i += 2) {
int option;
if (Jim_GetEnum(interp, argv[i], options, &option, NULL, JIM_ERRMSG | JIM_ENUM_ABBREV) != JIM_OK) {
return JIM_ERR;
}
switch (option) {
case OPT_GMT:
if (Jim_GetBoolean(interp, argv[i + 1], &opts->gmt) != JIM_OK) {
return JIM_ERR;
}
break;
case OPT_FORMAT:
opts->format = Jim_String(argv[i + 1]);
break;
}
}
return JIM_OK;
}
static int clock_cmd_format(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
{
/* How big is big enough? */
char buf[100];
time_t t;
jim_wide seconds;
struct clock_options options = { 0, "%a %b %d %H:%M:%S %Z %Y" };
struct tm *tm;
if (Jim_GetWide(interp, argv[0], &seconds) != JIM_OK) {
return JIM_ERR;
}
if (argc % 2 == 0) {
return -1;
}
if (parse_clock_options(interp, argc - 1, argv + 1, &options) == JIM_ERR) {
return JIM_ERR;
}
t = seconds;
tm = options.gmt ? gmtime(&t) : localtime(&t);
if (tm == NULL || strftime(buf, sizeof(buf), options.format, tm) == 0) {
Jim_SetResultString(interp, "format string too long or invalid time", -1);
return JIM_ERR;
}
Jim_SetResultString(interp, buf, -1);
return JIM_OK;
}
#ifdef HAVE_STRPTIME
#ifndef HAVE_TIMEGM
/* Implement a basic timegm() for system's that don't have it */
static time_t timegm(struct tm *tm)
{
time_t t;
const char *tz = getenv("TZ");
setenv("TZ", "", 1);
tzset();
t = mktime(tm);
if (tz) {
setenv("TZ", tz, 1);
}
else {
unsetenv("TZ");
}
tzset();
return t;
}
#endif
static int clock_cmd_scan(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
{
char *pt;
struct tm tm;
/*time_t now = time(NULL);*/
/* No default format */
struct clock_options options = { 0, NULL };
if (argc % 2 == 0) {
return -1;
}
if (parse_clock_options(interp, argc - 1, argv + 1, &options) == JIM_ERR) {
return JIM_ERR;
}
if (options.format == NULL) {
return -1;
}
/* Set unspecified fields to 0, e.g. HH:MM becomes 00:00 */
memset(&tm, 0, sizeof(tm));
/* But this is 1-based */
tm.tm_mday = 1;
pt = strptime(Jim_String(argv[0]), options.format, &tm);
if (pt == 0 || *pt != 0) {
Jim_SetResultString(interp, "Failed to parse time according to format", -1);
return JIM_ERR;
}
/* Now convert into a time_t */
Jim_SetResultInt(interp, options.gmt ? timegm(&tm) : mktime(&tm));
return JIM_OK;
}
#endif
static int clock_cmd_seconds(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
{
Jim_SetResultInt(interp, time(NULL));
return JIM_OK;
}
static int clock_cmd_micros(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
{
struct timeval tv;
gettimeofday(&tv, NULL);
Jim_SetResultInt(interp, (jim_wide) tv.tv_sec * 1000000 + tv.tv_usec);
return JIM_OK;
}
static int clock_cmd_millis(Jim_Interp *interp, int argc, Jim_Obj *const *argv)
{
struct timeval tv;
gettimeofday(&tv, NULL);
Jim_SetResultInt(interp, (jim_wide) tv.tv_sec * 1000 + tv.tv_usec / 1000);
return JIM_OK;
}
static const jim_subcmd_type clock_command_table[] = {
{ "clicks",
NULL,
clock_cmd_micros,
0,
0,
/* Description: Returns the current time in 'clicks' */
},
{ "format",
"seconds ?-format string? ?-gmt boolean?",
clock_cmd_format,
1,
5,
/* Description: Format the given time */
},
{ "microseconds",
NULL,
clock_cmd_micros,
0,
0,
/* Description: Returns the current time in microseconds */
},
{ "milliseconds",
NULL,
clock_cmd_millis,
0,
0,
/* Description: Returns the current time in milliseconds */
},
#ifdef HAVE_STRPTIME
{ "scan",
"str -format format ?-gmt boolean?",
clock_cmd_scan,
3,
5,
/* Description: Determine the time according to the given format */
},
#endif
{ "seconds",
NULL,
clock_cmd_seconds,
0,
0,
/* Description: Returns the current time as seconds since the epoch */
},
{ NULL }
};
int Jim_clockInit(Jim_Interp *interp)
{
if (Jim_PackageProvide(interp, "clock", "1.0", JIM_ERRMSG))
return JIM_ERR;
Jim_CreateCommand(interp, "clock", Jim_SubCmdProc, (void *)clock_command_table, NULL);
return JIM_OK;
}