[solanum.git] / libratbox / src / ratbox_lib.c

/*
 *  ircd-ratbox: A slightly useful ircd.
 *  ratbox_lib.c: libircd initialization functions at the like
 *
 *  Copyright (C) 2005,2006 ircd-ratbox development team
 *  Copyright (C) 2005,2006 Aaron Sethman <androsyn@ratbox.org>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301
 *  USA
 *
 *  $Id: ratbox_lib.c 26282 2008-12-10 20:33:21Z androsyn $
 */

#include <libratbox_config.h>
#include <ratbox_lib.h>
#include <commio-int.h>
#include <commio-ssl.h>

static log_cb *rb_log;
static restart_cb *rb_restart;
static die_cb *rb_die;

static struct timeval rb_time;
static char errbuf[512];

/* this doesn't do locales...oh well i guess */

static const char *months[] = {
	"January", "February", "March", "April",
	"May", "June", "July", "August",
	"September", "October", "November", "December"
};

static const char *weekdays[] = {
	"Sunday", "Monday", "Tuesday", "Wednesday",
	"Thursday", "Friday", "Saturday"
};

static const char *s_month[] = {
	"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul",
	"Aug", "Sep", "Oct", "Nov", "Dec"
};

static const char *s_weekdays[] = {
	"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"
};

char *
rb_ctime(const time_t t, char *buf, size_t len)
{
	char *p;
	struct tm *tp;
	static char timex[128];
	size_t tlen;
#if defined(HAVE_GMTIME_R)
	struct tm tmr;
	tp = gmtime_r(&t, &tmr);
#else
	tp = gmtime(&t);
#endif
	if(rb_unlikely(tp == NULL))
	{
		strcpy(buf, "");
		return (buf);
	}

	if(buf == NULL)
	{
		p = timex;
		tlen = sizeof(timex);
	}
	else
	{
		p = buf;
		tlen = len;
	}

	rb_snprintf(p, tlen, "%s %s %d %02u:%02u:%02u %d",
		    s_weekdays[tp->tm_wday], s_month[tp->tm_mon],
		    tp->tm_mday, tp->tm_hour, tp->tm_min, tp->tm_sec, tp->tm_year + 1900);
	return (p);
}


/* I hate this..but its sort of ircd standard now.. */
char *
rb_date(const time_t t, char *buf, size_t len)
{
	struct tm *gm;
#if defined(HAVE_GMTIME_R)
	struct tm gmbuf;
	gm = gmtime_r(&t, &gmbuf);
#else
	gm = gmtime(&t);
#endif

	if(rb_unlikely(gm == NULL))
	{
		rb_strlcpy(buf, "", len);
		return (buf);
	}

	rb_snprintf(buf, len, "%s %s %d %d -- %02u:%02u:%02u +00:00",
		    weekdays[gm->tm_wday], months[gm->tm_mon], gm->tm_mday,
		    gm->tm_year + 1900, gm->tm_hour, gm->tm_min, gm->tm_sec);
	return (buf);
}

time_t
rb_current_time(void)
{
	return rb_time.tv_sec;
}

const struct timeval *
rb_current_time_tv(void)
{
	return &rb_time;
}

void
rb_lib_log(const char *format, ...)
{
	va_list args;
	if(rb_log == NULL)
		return;
	va_start(args, format);
	rb_vsnprintf(errbuf, sizeof(errbuf), format, args);
	va_end(args);
	rb_log(errbuf);
}

void
rb_lib_die(const char *format, ...)
{
	va_list args;
	if(rb_die == NULL)
		abort();
	va_start(args, format);
	rb_vsnprintf(errbuf, sizeof(errbuf), format, args);
	va_end(args);
	rb_die(errbuf);
}

void
rb_lib_restart(const char *format, ...)
{
	va_list args;
	if(rb_restart == NULL)
		abort();
	va_start(args, format);
	rb_vsnprintf(errbuf, sizeof(errbuf), format, args);
	va_end(args);
	rb_restart(errbuf);
}

void
rb_set_time(void)
{
	struct timeval newtime;

	if(rb_unlikely(rb_gettimeofday(&newtime, NULL) == -1))
	{
		rb_lib_log("Clock Failure (%s)", strerror(errno));
		rb_lib_restart("Clock Failure");
	}

	if(newtime.tv_sec < rb_time.tv_sec)
		rb_set_back_events(rb_time.tv_sec - newtime.tv_sec);

	memcpy(&rb_time, &newtime, sizeof(struct timeval));
}

extern const char *libratbox_serno;

const char *
rb_lib_version(void)
{
	static char version_info[512];
	char ssl_info[512];
	rb_get_ssl_info(ssl_info, sizeof(ssl_info));
	rb_snprintf(version_info, sizeof(version_info), "libratbox version: %s - %s", libratbox_serno, ssl_info);
	return version_info;
}

void
rb_lib_init(log_cb * ilog, restart_cb * irestart, die_cb * idie, int closeall, int maxcon,
	    size_t dh_size, size_t fd_heap_size)
{
	rb_set_time();
	rb_log = ilog;
	rb_restart = irestart;
	rb_die = idie;
	rb_event_init();
	rb_init_bh();
	rb_fdlist_init(closeall, maxcon, fd_heap_size);
	rb_init_netio();
	rb_init_rb_dlink_nodes(dh_size);
	if(rb_io_supports_event())
	{
		rb_io_init_event();
	}
}

void
rb_lib_loop(long delay)
{
	time_t next;
	rb_set_time();

	if(rb_io_supports_event())
	{
		if(delay == 0)
			delay = -1;
		while(1)
			rb_select(-1);
	}


	while(1)
	{
		if(delay == 0)
		{
			if((next = rb_event_next()) > 0)
			{
				next -= rb_current_time();
				if(next <= 0)
					next = 1000;
				else
					next *= 1000;
			}
			else
				next = -1;
			rb_select(next);
		}
		else
			rb_select(delay);
		rb_event_run();
	}
}

#ifndef HAVE_STRTOK_R
char *
rb_strtok_r(char *s, const char *delim, char **save)
{
	char *token;

	if(s == NULL)
		s = *save;

	/* Scan leading delimiters.  */
	s += strspn(s, delim);

	if(*s == '\0')
	{
		*save = s;
		return NULL;
	}

	token = s;
	s = strpbrk(token, delim);

	if(s == NULL)
		*save = (token + strlen(token));
	else
	{
		*s = '\0';
		*save = s + 1;
	}
	return token;
}
#else
char *
rb_strtok_r(char *s, const char *delim, char **save)
{
	return strtok_r(s, delim, save);
}
#endif


static const char base64_table[] =
	{ 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
	'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
	'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
	'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
	'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/', '\0'
};

static const char base64_pad = '=';

static const short base64_reverse_table[256] = {
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, -1, -1, 63,
	52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1,
	-1, 0, 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, -1, -1, -1, -1, -1,
	-1, 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, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1
};

unsigned char *
rb_base64_encode(const unsigned char *str, int length)
{
	const unsigned char *current = str;
	unsigned char *p;
	unsigned char *result;

	if((length + 2) < 0 || ((length + 2) / 3) >= (1 << (sizeof(int) * 8 - 2)))
	{
		return NULL;
	}

	result = rb_malloc(((length + 2) / 3) * 5);
	p = result;

	while(length > 2)
	{
		*p++ = base64_table[current[0] >> 2];
		*p++ = base64_table[((current[0] & 0x03) << 4) + (current[1] >> 4)];
		*p++ = base64_table[((current[1] & 0x0f) << 2) + (current[2] >> 6)];
		*p++ = base64_table[current[2] & 0x3f];

		current += 3;
		length -= 3;
	}

	if(length != 0)
	{
		*p++ = base64_table[current[0] >> 2];
		if(length > 1)
		{
			*p++ = base64_table[((current[0] & 0x03) << 4) + (current[1] >> 4)];
			*p++ = base64_table[(current[1] & 0x0f) << 2];
			*p++ = base64_pad;
		}
		else
		{
			*p++ = base64_table[(current[0] & 0x03) << 4];
			*p++ = base64_pad;
			*p++ = base64_pad;
		}
	}
	*p = '\0';
	return result;
}

unsigned char *
rb_base64_decode(const unsigned char *str, int length, int *ret)
{
	const unsigned char *current = str;
	int ch, i = 0, j = 0, k;
	unsigned char *result;

	result = rb_malloc(length + 1);

	while((ch = *current++) != '\0' && length-- > 0)
	{
		if(ch == base64_pad)
			break;

		ch = base64_reverse_table[ch];
		if(ch < 0)
			continue;

		switch (i % 4)
		{
		case 0:
			result[j] = ch << 2;
			break;
		case 1:
			result[j++] |= ch >> 4;
			result[j] = (ch & 0x0f) << 4;
			break;
		case 2:
			result[j++] |= ch >> 2;
			result[j] = (ch & 0x03) << 6;
			break;
		case 3:
			result[j++] |= ch;
			break;
		}
		i++;
	}

	k = j;

	if(ch == base64_pad)
	{
		switch (i % 4)
		{
		case 1:
			free(result);
			return NULL;
		case 2:
			k++;
		case 3:
			result[k++] = 0;
		}
	}
	result[j] = '\0';
	*ret = j;
	return result;
}
Commit	Line	Data
db137867 AC	1	/*
	2	* ircd-ratbox: A slightly useful ircd.
	3	* ratbox_lib.c: libircd initialization functions at the like
	4	*
	5	* Copyright (C) 2005,2006 ircd-ratbox development team
	6	* Copyright (C) 2005,2006 Aaron Sethman <androsyn@ratbox.org>
	7	*
	8	* This program is free software; you can redistribute it and/or modify
	9	* it under the terms of the GNU General Public License as published by
	10	* the Free Software Foundation; either version 2 of the License, or
	11	* (at your option) any later version.
	12	*
	13	* This program is distributed in the hope that it will be useful,
	14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	16	* GNU General Public License for more details.
	17	*
	18	* You should have received a copy of the GNU General Public License
	19	* along with this program; if not, write to the Free Software
	20	* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
	21	* USA
	22	*
030272f3	23	* $Id: ratbox_lib.c 26282 2008-12-10 20:33:21Z androsyn $
db137867 AC	24	*/
	25
	26	#include <libratbox_config.h>
	27	#include <ratbox_lib.h>
	28	#include <commio-int.h>
030272f3	29	#include <commio-ssl.h>
db137867 AC	30
	31	static log_cb *rb_log;
	32	static restart_cb *rb_restart;
	33	static die_cb *rb_die;
	34
	35	static struct timeval rb_time;
	36	static char errbuf[512];
	37
	38	/* this doesn't do locales...oh well i guess */
	39
	40	static const char *months[] = {
3202e249 VY	41	"January", "February", "March", "April",
	42	"May", "June", "July", "August",
	43	"September", "October", "November", "December"
db137867 AC	44	};
	45
	46	static const char *weekdays[] = {
3202e249 VY	47	"Sunday", "Monday", "Tuesday", "Wednesday",
3202e249 VY	48	"Thursday", "Friday", "Saturday"
db137867 AC	49	};
	50
	51	static const char *s_month[] = {
	52	"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul",
	53	"Aug", "Sep", "Oct", "Nov", "Dec"
3202e249	54	};
db137867 AC	55
	56	static const char *s_weekdays[] = {
	57	"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"
	58	};
	59
	60	char *
	61	rb_ctime(const time_t t, char *buf, size_t len)
	62	{
	63	char *p;
	64	struct tm *tp;
	65	static char timex[128];
	66	size_t tlen;
	67	#if defined(HAVE_GMTIME_R)
	68	struct tm tmr;
	69	tp = gmtime_r(&t, &tmr);
	70	#else
	71	tp = gmtime(&t);
	72	#endif
c2ac22cc	73	if(rb_unlikely(tp == NULL))
db137867 AC	74	{
db137867 AC	75	strcpy(buf, "");
3202e249	76	return (buf);
db137867 AC	77	}
	78
	79	if(buf == NULL)
	80	{
	81	p = timex;
	82	tlen = sizeof(timex);
3202e249 VY	83	}
	84	else
	85	{
db137867 AC	86	p = buf;
	87	tlen = len;
	88	}
	89
	90	rb_snprintf(p, tlen, "%s %s %d %02u:%02u:%02u %d",
3202e249 VY	91	s_weekdays[tp->tm_wday], s_month[tp->tm_mon],
	92	tp->tm_mday, tp->tm_hour, tp->tm_min, tp->tm_sec, tp->tm_year + 1900);
	93	return (p);
db137867 AC	94	}
	95
	96
	97	/* I hate this..but its sort of ircd standard now.. */
	98	char *
	99	rb_date(const time_t t, char *buf, size_t len)
	100	{
	101	struct tm *gm;
	102	#if defined(HAVE_GMTIME_R)
	103	struct tm gmbuf;
	104	gm = gmtime_r(&t, &gmbuf);
	105	#else
	106	gm = gmtime(&t);
	107	#endif
	108
c2ac22cc	109	if(rb_unlikely(gm == NULL))
db137867	110	{
3202e249 VY	111	rb_strlcpy(buf, "", len);
3202e249 VY	112	return (buf);
db137867	113	}
3202e249 VY	114
	115	rb_snprintf(buf, len, "%s %s %d %d -- %02u:%02u:%02u +00:00",
	116	weekdays[gm->tm_wday], months[gm->tm_mon], gm->tm_mday,
	117	gm->tm_year + 1900, gm->tm_hour, gm->tm_min, gm->tm_sec);
	118	return (buf);
db137867 AC	119	}
	120
	121	time_t
	122	rb_current_time(void)
	123	{
	124	return rb_time.tv_sec;
	125	}
	126
	127	const struct timeval *
	128	rb_current_time_tv(void)
	129	{
	130	return &rb_time;
	131	}
	132
	133	void
	134	rb_lib_log(const char *format, ...)
	135	{
	136	va_list args;
	137	if(rb_log == NULL)
	138	return;
	139	va_start(args, format);
3202e249	140	rb_vsnprintf(errbuf, sizeof(errbuf), format, args);
db137867 AC	141	va_end(args);
	142	rb_log(errbuf);
	143	}
	144
	145	void
	146	rb_lib_die(const char *format, ...)
	147	{
	148	va_list args;
	149	if(rb_die == NULL)
	150	abort();
	151	va_start(args, format);
3202e249	152	rb_vsnprintf(errbuf, sizeof(errbuf), format, args);
db137867 AC	153	va_end(args);
	154	rb_die(errbuf);
	155	}
	156
	157	void
	158	rb_lib_restart(const char *format, ...)
	159	{
	160	va_list args;
	161	if(rb_restart == NULL)
	162	abort();
	163	va_start(args, format);
3202e249	164	rb_vsnprintf(errbuf, sizeof(errbuf), format, args);
db137867 AC	165	va_end(args);
	166	rb_restart(errbuf);
	167	}
	168
	169	void
	170	rb_set_time(void)
	171	{
	172	struct timeval newtime;
	173
c2ac22cc	174	if(rb_unlikely(rb_gettimeofday(&newtime, NULL) == -1))
db137867 AC	175	{
	176	rb_lib_log("Clock Failure (%s)", strerror(errno));
	177	rb_lib_restart("Clock Failure");
	178	}
	179
	180	if(newtime.tv_sec < rb_time.tv_sec)
	181	rb_set_back_events(rb_time.tv_sec - newtime.tv_sec);
	182
	183	memcpy(&rb_time, &newtime, sizeof(struct timeval));
	184	}
	185
3202e249 VY	186	extern const char *libratbox_serno;
3202e249 VY	187
db137867 AC	188	const char *
	189	rb_lib_version(void)
	190	{
030272f3 VY	191	static char version_info[512];
	192	char ssl_info[512];
	193	rb_get_ssl_info(ssl_info, sizeof(ssl_info));
	194	rb_snprintf(version_info, sizeof(version_info), "libratbox version: %s - %s", libratbox_serno, ssl_info);
	195	return version_info;
db137867 AC	196	}
	197
	198	void
3202e249 VY	199	rb_lib_init(log_cb * ilog, restart_cb * irestart, die_cb * idie, int closeall, int maxcon,
3202e249 VY	200	size_t dh_size, size_t fd_heap_size)
db137867 AC	201	{
	202	rb_set_time();
	203	rb_log = ilog;
	204	rb_restart = irestart;
	205	rb_die = idie;
	206	rb_event_init();
	207	rb_init_bh();
	208	rb_fdlist_init(closeall, maxcon, fd_heap_size);
	209	rb_init_netio();
	210	rb_init_rb_dlink_nodes(dh_size);
	211	if(rb_io_supports_event())
	212	{
	213	rb_io_init_event();
	214	}
	215	}
	216
	217	void
	218	rb_lib_loop(long delay)
	219	{
	220	time_t next;
	221	rb_set_time();
	222
	223	if(rb_io_supports_event())
	224	{
	225	if(delay == 0)
	226	delay = -1;
	227	while(1)
	228	rb_select(-1);
	229	}
3202e249	230
db137867 AC	231
	232	while(1)
	233	{
	234	if(delay == 0)
	235	{
3202e249	236	if((next = rb_event_next()) > 0)
db137867 AC	237	{
	238	next -= rb_current_time();
	239	if(next <= 0)
	240	next = 1000;
3202e249	241	else
db137867 AC	242	next *= 1000;
	243	}
	244	else
	245	next = -1;
	246	rb_select(next);
3202e249 VY	247	}
3202e249 VY	248	else
db137867 AC	249	rb_select(delay);
	250	rb_event_run();
	251	}
	252	}
	253
	254	#ifndef HAVE_STRTOK_R
	255	char *
3202e249	256	rb_strtok_r(char s, const char delim, char **save)
db137867 AC	257	{
	258	char *token;
	259
3202e249	260	if(s == NULL)
db137867 AC	261	s = *save;
	262
	263	/* Scan leading delimiters. */
	264	s += strspn(s, delim);
	265
3202e249	266	if(*s == '\0')
db137867 AC	267	{
	268	*save = s;
	269	return NULL;
	270	}
	271
	272	token = s;
	273	s = strpbrk(token, delim);
3202e249 VY	274
3202e249 VY	275	if(s == NULL)
db137867 AC	276	*save = (token + strlen(token));
	277	else
	278	{
3202e249	279	*s = '\0';
db137867 AC	280	*save = s + 1;
	281	}
	282	return token;
	283	}
	284	#else
3202e249 VY	285	char *
3202e249 VY	286	rb_strtok_r(char s, const char delim, char **save)
db137867 AC	287	{
	288	return strtok_r(s, delim, save);
	289	}
	290	#endif
	291
	292
	293	static const char base64_table[] =
	294	{ 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
3202e249 VY	295	'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
	296	'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
	297	'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
	298	'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/', '\0'
	299	};
db137867 AC	300
	301	static const char base64_pad = '=';
	302
	303	static const short base64_reverse_table[256] = {
	304	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	305	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	306	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, -1, -1, 63,
	307	52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1,
3202e249	308	-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
db137867 AC	309	15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1,
	310	-1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
	311	41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -1, -1, -1, -1, -1,
	312	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	313	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	314	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	315	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	316	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	317	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	318	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
	319	-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1
	320	};
	321
	322	unsigned char *
	323	rb_base64_encode(const unsigned char *str, int length)
	324	{
	325	const unsigned char *current = str;
	326	unsigned char *p;
	327	unsigned char *result;
	328
3202e249 VY	329	if((length + 2) < 0 \|\| ((length + 2) / 3) >= (1 << (sizeof(int) * 8 - 2)))
3202e249 VY	330	{
db137867 AC	331	return NULL;
	332	}
	333
	334	result = rb_malloc(((length + 2) / 3) * 5);
	335	p = result;
	336
3202e249 VY	337	while(length > 2)
3202e249 VY	338	{
db137867 AC	339	*p++ = base64_table[current[0] >> 2];
	340	*p++ = base64_table[((current[0] & 0x03) << 4) + (current[1] >> 4)];
	341	*p++ = base64_table[((current[1] & 0x0f) << 2) + (current[2] >> 6)];
	342	*p++ = base64_table[current[2] & 0x3f];
	343
	344	current += 3;
3202e249	345	length -= 3;
db137867 AC	346	}
db137867 AC	347
3202e249 VY	348	if(length != 0)
3202e249 VY	349	{
db137867	350	*p++ = base64_table[current[0] >> 2];
3202e249 VY	351	if(length > 1)
3202e249 VY	352	{
db137867 AC	353	*p++ = base64_table[((current[0] & 0x03) << 4) + (current[1] >> 4)];
	354	*p++ = base64_table[(current[1] & 0x0f) << 2];
	355	*p++ = base64_pad;
3202e249 VY	356	}
	357	else
	358	{
db137867 AC	359	*p++ = base64_table[(current[0] & 0x03) << 4];
	360	*p++ = base64_pad;
	361	*p++ = base64_pad;
	362	}
	363	}
	364	*p = '\0';
	365	return result;
	366	}
	367
	368	unsigned char *
	369	rb_base64_decode(const unsigned char str, int length, int ret)
	370	{
	371	const unsigned char *current = str;
	372	int ch, i = 0, j = 0, k;
	373	unsigned char *result;
3202e249	374
db137867 AC	375	result = rb_malloc(length + 1);
db137867 AC	376
3202e249 VY	377	while((ch = *current++) != '\0' && length-- > 0)
	378	{
	379	if(ch == base64_pad)
	380	break;
db137867 AC	381
db137867 AC	382	ch = base64_reverse_table[ch];
3202e249 VY	383	if(ch < 0)
3202e249 VY	384	continue;
db137867	385
3202e249 VY	386	switch (i % 4)
3202e249 VY	387	{
db137867 AC	388	case 0:
	389	result[j] = ch << 2;
	390	break;
	391	case 1:
	392	result[j++] \|= ch >> 4;
	393	result[j] = (ch & 0x0f) << 4;
	394	break;
	395	case 2:
3202e249	396	result[j++] \|= ch >> 2;
db137867 AC	397	result[j] = (ch & 0x03) << 6;
	398	break;
	399	case 3:
	400	result[j++] \|= ch;
	401	break;
	402	}
	403	i++;
	404	}
	405
	406	k = j;
	407
3202e249 VY	408	if(ch == base64_pad)
	409	{
	410	switch (i % 4)
	411	{
db137867 AC	412	case 1:
	413	free(result);
	414	return NULL;
	415	case 2:
	416	k++;
	417	case 3:
	418	result[k++] = 0;
	419	}
	420	}
	421	result[j] = '\0';
	422	*ret = j;
	423	return result;
	424	}