[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 26092 2008-09-19 15:13:52Z androsyn $
 */

#include <libratbox_config.h>
#include <ratbox_lib.h>
#include <commio-int.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)
{
	return libratbox_serno;
}

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