pitchfork 0.1.0

data/ext/pitchfork_http/httpdate.c

@@ -0,0 +1,79 @@
+#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 VALUE buf;
+static char *buf_ptr;
+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
+# warning using fake 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
+
+
+/*
+ * 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.
+ *
+ * Caveats: it relies on a Ruby implementation with the global VM lock,
+ * a thread-safe version will be provided when a Unix-only, GVL-free Ruby
+ * implementation becomes viable.
+ */
+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;
+
+	if (last == now)
+		return buf;
+	last = now;
+	gmtime_r(&now, &tm);
+
+	/* we can make this thread-safe later if our Ruby loses the GVL */
+	snprintf(buf_ptr, 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_pitchfork_httpdate(void)
+{
+	VALUE mod = rb_define_module("Pitchfork");
+	mod = rb_define_module_under(mod, "HttpResponse");
+
+	buf = rb_str_new(0, buf_capa - 1);
+	rb_gc_register_mark_object(buf);
+	buf_ptr = RSTRING_PTR(buf);
+	httpdate(Qnil);
+
+	rb_define_method(mod, "httpdate", httpdate, 0);
+}