summary refs log tree commit homepage
path: root/ext/unicorn_http/httpdate.c
blob: 06598e261bc5a1826ddbbb53c3373b18dc6ae866 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#define RSTRING_MODIFIED 1 /* we modify RSTRING_PTR */
#include <ruby.h>
#include <time.h>
#include <stdio.h>

static const size_t buf_capa = sizeof("Thu, 01 Jan 1970 00:00:00 GMT");
static const char week[] = "Sun\0Mon\0Tue\0Wed\0Thu\0Fri\0Sat";
static const char months[] = "Jan\0Feb\0Mar\0Apr\0May\0Jun\0"
                             "Jul\0Aug\0Sep\0Oct\0Nov\0Dec";

/* for people on wonky systems only */
#ifndef HAVE_GMTIME_R
static struct tm * my_gmtime_r(time_t *now, struct tm *tm)
{
	struct tm *global = gmtime(now);
	if (global)
		*tm = *global;
	return tm;
}
#  define gmtime_r my_gmtime_r
#endif

/* TODO: update this in case other implementations lose the GVL */
#if defined(RUBINIUS)
#  define UH_HAVE_GVL (0)
#else
#  define UH_HAVE_GVL (1)
#endif

#if defined(__GNUC__) && (__GNUC__ >= 3)
# define UH_ATTRIBUTE_CONST __attribute__ ((__const__))
#else
# define UH_ATTRIBUTE_CONST /* empty */
#endif

#if UH_HAVE_GVL
static VALUE g_buf;
static char *g_buf_ptr;
static VALUE UH_ATTRIBUTE_CONST get_buf(void) { return g_buf; }
static char * UH_ATTRIBUTE_CONST get_buf_ptr(VALUE ign) { return g_buf_ptr; }
static void init_buf(void)
{
	g_buf = rb_str_new(0, buf_capa - 1);
	g_buf_ptr = RSTRING_PTR(g_buf);
	rb_global_variable(&g_buf);
}
#else /* !UH_HAVE_GVL */
static VALUE buf_key;
static VALUE get_buf(void)
{
	VALUE buf = rb_thread_local_aref(rb_thread_current(), buf_key);

	/*
	 * we must validate this, otherwise some bad code could muck
	 * with local vars in Thread.current and crash us
	 */
	if (TYPE(buf) != T_STRING) {
		buf = rb_str_new(0, buf_capa - 1);
		rb_thread_local_aset(rb_thread_current(), buf_key, buf);
	} else if (RSTRING_LEN(buf) != (long)buf_capa) {
		rb_str_modify(buf);
		rb_str_resize(buf, buf_capa - 1);
	}

	return buf;
}

static char *get_buf_ptr(VALUE buf) { return RSTRING_PTR(buf); }

static void init_buf(void)
{
	buf_key = ID2SYM(rb_intern("uh_httpdate_buf"));
	rb_global_variable(&buf_key); /* in case symbols ever get GC-ed */
}
#endif /* !UH_HAVE_GVL */

/*
 * Returns a string which represents the time as rfc1123-date of HTTP-date
 * defined by RFC 2616:
 *
 *   day-of-week, DD month-name CCYY hh:mm:ss GMT
 *
 * Note that the result is always GMT.
 *
 * This method is identical to Time#httpdate in the Ruby standard library,
 * except it is implemented in C for performance.  We always saw
 * Time#httpdate at or near the top of the profiler output so we
 * decided to rewrite this in C.
 */
static VALUE httpdate(VALUE self)
{
	static time_t last;
	time_t now = time(NULL); /* not a syscall on modern 64-bit systems */
	struct tm tm;
	VALUE buf = get_buf();

	if (last == now)
		return buf;
	last = now;
	gmtime_r(&now, &tm);

	snprintf(get_buf_ptr(buf), buf_capa,
	         "%s, %02d %s %4d %02d:%02d:%02d GMT",
	         week + (tm.tm_wday * 4),
	         tm.tm_mday,
	         months + (tm.tm_mon * 4),
	         tm.tm_year + 1900,
	         tm.tm_hour,
	         tm.tm_min,
	         tm.tm_sec);

	return buf;
}

void init_unicorn_httpdate(void)
{
	VALUE mod = rb_const_get(rb_cObject, rb_intern("Unicorn"));
	mod = rb_define_module_under(mod, "HttpResponse");
	init_buf();

	/*
	 * initialize, gmtime_r uses a lot of stack on FreeBSD for
	 * loading tzinfo data, so this allows a Ruby implementation
	 * to use * smaller thread stacks in child threads.
	 */
	httpdate(Qnil);

	rb_define_method(mod, "httpdate", httpdate, 0);
}