mirror of
https://github.com/mruby/mruby
synced 2026-06-08 16:11:16 +00:00
841 lines
19 KiB
C
841 lines
19 KiB
C
/*
|
|
** pack.c - Array#pack, String#unpack
|
|
*/
|
|
|
|
#include "mruby.h"
|
|
#include "error.h"
|
|
#include "mruby/array.h"
|
|
#include "mruby/class.h"
|
|
#include "mruby/numeric.h"
|
|
#include "mruby/string.h"
|
|
#include "mruby/variable.h"
|
|
|
|
#include <ctype.h>
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
|
|
struct tmpl {
|
|
mrb_value str;
|
|
int idx;
|
|
};
|
|
|
|
enum {
|
|
PACK_DIR_CHAR, /* C */
|
|
PACK_DIR_SHORT, /* S */
|
|
PACK_DIR_LONG, /* L */
|
|
//PACK_DIR_QUAD, /* Q */
|
|
//PACK_DIR_INT, /* i */
|
|
//PACK_DIR_VAX,
|
|
//PACK_DIR_UTF8, /* U */
|
|
//PACK_DIR_BER,
|
|
PACK_DIR_DOUBLE, /* E */
|
|
//PACK_DIR_FLOAT, /* f */
|
|
PACK_DIR_STR, /* A */
|
|
PACK_DIR_HEX, /* h */
|
|
PACK_DIR_BASE64, /* m */
|
|
PACK_DIR_INVALID
|
|
};
|
|
|
|
enum {
|
|
PACK_TYPE_INTEGER,
|
|
PACK_TYPE_FLOAT,
|
|
PACK_TYPE_STRING,
|
|
PACK_TYPE_NONE
|
|
};
|
|
|
|
#define PACK_FLAG_s 0x00000001 /* native size ("_" "!") */
|
|
#define PACK_FLAG_a 0x00000002 /* null padding ("a") */
|
|
#define PACK_FLAG_Z 0x00000004 /* append nul char ("z") */
|
|
#define PACK_FLAG_SIGNED 0x00000008 /* native size ("_" "!") */
|
|
#define PACK_FLAG_GT 0x00000010 /* big endian (">") */
|
|
#define PACK_FLAG_LT 0x00000020 /* little endian ("<") */
|
|
#define PACK_FLAG_WIDTH 0x00000040 /* */
|
|
#define PACK_FLAG_LSB 0x00000080 /* LSB / low nibble first */
|
|
#define PACK_FLAG_COUNT2 0x00000100 /* "count" is special... */
|
|
#define PACK_FLAG_LITTLEENDIAN 0x00000200 /* little endian actually */
|
|
|
|
#define PACK_BASE64_IGNORE 0xff
|
|
#define PACK_BASE64_PADDING 0xfe
|
|
|
|
static int littleendian = 0;
|
|
|
|
const static unsigned char base64chars[] =
|
|
"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
|
static signed char base64_dec_tab[128];
|
|
|
|
|
|
static int
|
|
check_little_endian(void)
|
|
{
|
|
unsigned int n = 1;
|
|
return (*(unsigned char *)&n == 1);
|
|
}
|
|
|
|
static unsigned int
|
|
hex2int(unsigned char ch)
|
|
{
|
|
if (ch >= '0' && ch <= '9')
|
|
return ch - '0';
|
|
else if (ch >= 'A' && ch <= 'F')
|
|
return 10 + (ch - 'A');
|
|
else if (ch >= 'a' && ch <= 'f')
|
|
return 10 + (ch - 'a');
|
|
else
|
|
return 0;
|
|
}
|
|
|
|
static void
|
|
make_base64_dec_tab(void)
|
|
{
|
|
int i;
|
|
memset(base64_dec_tab, PACK_BASE64_IGNORE, sizeof(base64_dec_tab));
|
|
for (i = 0; i < 26; i++)
|
|
base64_dec_tab['A' + i] = i;
|
|
for (i = 0; i < 26; i++)
|
|
base64_dec_tab['a' + i] = i + 26;
|
|
for (i = 0; i < 10; i++)
|
|
base64_dec_tab['0' + i] = i + 52;
|
|
base64_dec_tab['+'] = 62;
|
|
base64_dec_tab['/'] = 63;
|
|
base64_dec_tab['='] = PACK_BASE64_PADDING;
|
|
}
|
|
|
|
static mrb_value
|
|
str_len_ensure(mrb_state *mrb, mrb_value str, int len)
|
|
{
|
|
int n = RSTRING_LEN(str);
|
|
if (len > n) {
|
|
do {
|
|
n *= 2;
|
|
} while (len > n);
|
|
str = mrb_str_resize(mrb, str, n);
|
|
}
|
|
return str;
|
|
}
|
|
|
|
|
|
static int
|
|
pack_c(mrb_state *mrb, mrb_value o, mrb_value str, mrb_int sidx, unsigned int flags)
|
|
{
|
|
str = str_len_ensure(mrb, str, sidx + 1);
|
|
RSTRING_PTR(str)[sidx] = mrb_fixnum(o);
|
|
return 1;
|
|
}
|
|
|
|
static int
|
|
unpack_c(mrb_state *mrb, const void *src, int srclen, mrb_value ary, unsigned int flags)
|
|
{
|
|
if (flags & PACK_FLAG_SIGNED)
|
|
mrb_ary_push(mrb, ary, mrb_fixnum_value(*(signed char *)src));
|
|
else
|
|
mrb_ary_push(mrb, ary, mrb_fixnum_value(*(unsigned char *)src));
|
|
return 1;
|
|
}
|
|
|
|
static int
|
|
pack_s(mrb_state *mrb, mrb_value o, mrb_value str, mrb_int sidx, unsigned int flags)
|
|
{
|
|
unsigned short n;
|
|
|
|
str = str_len_ensure(mrb, str, sidx + 2);
|
|
n = mrb_fixnum(o);
|
|
if (flags & PACK_FLAG_LITTLEENDIAN) {
|
|
RSTRING_PTR(str)[sidx+0] = n % 256;
|
|
RSTRING_PTR(str)[sidx+1] = n / 256;
|
|
} else {
|
|
RSTRING_PTR(str)[sidx+0] = n / 256;
|
|
RSTRING_PTR(str)[sidx+1] = n % 256;
|
|
}
|
|
return 2;
|
|
}
|
|
|
|
static int
|
|
unpack_s(mrb_state *mrb, const unsigned char *src, int srclen, mrb_value ary, unsigned int flags)
|
|
{
|
|
int n;
|
|
|
|
if (flags & PACK_FLAG_LITTLEENDIAN) {
|
|
n = src[1] * 256 + src[0];
|
|
} else {
|
|
n = src[0] * 256 + src[1];
|
|
}
|
|
if ((flags & PACK_FLAG_SIGNED) && (n >= 0x8000)) {
|
|
n -= 0x10000;
|
|
}
|
|
mrb_ary_push(mrb, ary, mrb_fixnum_value(n));
|
|
return 2;
|
|
}
|
|
|
|
static int
|
|
pack_l(mrb_state *mrb, mrb_value o, mrb_value str, mrb_int sidx, unsigned int flags)
|
|
{
|
|
unsigned long n;
|
|
str = str_len_ensure(mrb, str, sidx + 4);
|
|
n = mrb_fixnum(o);
|
|
if (flags & PACK_FLAG_LITTLEENDIAN) {
|
|
RSTRING_PTR(str)[sidx+0] = n & 0xff;
|
|
RSTRING_PTR(str)[sidx+1] = n >> 8;
|
|
RSTRING_PTR(str)[sidx+2] = n >> 16;
|
|
RSTRING_PTR(str)[sidx+3] = n >> 24;
|
|
} else {
|
|
RSTRING_PTR(str)[sidx+0] = n >> 24;
|
|
RSTRING_PTR(str)[sidx+1] = n >> 16;
|
|
RSTRING_PTR(str)[sidx+2] = n >> 8;
|
|
RSTRING_PTR(str)[sidx+3] = n & 0xff;
|
|
}
|
|
return 4;
|
|
}
|
|
|
|
static int
|
|
pack_double(mrb_state *mrb, mrb_value o, mrb_value str, mrb_int sidx, unsigned int flags)
|
|
{
|
|
int i;
|
|
double d;
|
|
str = str_len_ensure(mrb, str, sidx + 8);
|
|
d = mrb_float(o);
|
|
uint8_t *buffer = (uint8_t *)&d;
|
|
|
|
#ifdef MRB_ENDIAN_BIG
|
|
#error unsupported
|
|
#else
|
|
for(i = 0; i< 8; i++){
|
|
RSTRING_PTR(str)[sidx+i] = buffer[i];
|
|
}
|
|
#endif
|
|
|
|
return 8;
|
|
}
|
|
|
|
static int
|
|
unpack_l(mrb_state *mrb, const unsigned char *src, int srclen, mrb_value ary, unsigned int flags)
|
|
{
|
|
unsigned long n;
|
|
|
|
if (flags & PACK_FLAG_LITTLEENDIAN) {
|
|
n = (src[3] << 24) + (src[2] << 16) + (src[1] << 8) + src[0];
|
|
} else {
|
|
n = (src[0] << 24) + (src[1] << 16) + (src[2] << 8) + src[3];
|
|
}
|
|
if (flags & PACK_FLAG_SIGNED) {
|
|
if (!FIXABLE((mrb_int)n))
|
|
mrb_raisef(mrb, E_ARGUMENT_ERROR, "cannot unpack to 32bit signed number: %ld", (mrb_int)n);
|
|
} else {
|
|
if (!POSFIXABLE(n))
|
|
mrb_raisef(mrb, E_ARGUMENT_ERROR, "cannot unpack to 32bit unsigned number: %lu", n);
|
|
}
|
|
mrb_ary_push(mrb, ary, mrb_fixnum_value(n));
|
|
return 4;
|
|
}
|
|
|
|
static int
|
|
pack_a(mrb_state *mrb, mrb_value src, mrb_value dst, mrb_int didx, long count, unsigned int flags)
|
|
{
|
|
int copylen, slen, padlen;
|
|
char *dptr, *dptr0, pad, *sptr;
|
|
|
|
sptr = RSTRING_PTR(src);
|
|
slen = RSTRING_LEN(src);
|
|
|
|
if ((flags & PACK_FLAG_a) || (flags & PACK_FLAG_Z))
|
|
pad = '\0';
|
|
else
|
|
pad = ' ';
|
|
|
|
if (count == -1) {
|
|
copylen = slen;
|
|
padlen = (flags & PACK_FLAG_Z) ? 1 : 0;
|
|
} else if (count < slen) {
|
|
copylen = count;
|
|
padlen = 0;
|
|
} else {
|
|
copylen = slen;
|
|
padlen = count - slen;
|
|
}
|
|
|
|
dst = str_len_ensure(mrb, dst, didx + copylen + padlen);
|
|
dptr0 = dptr = RSTRING_PTR(dst) + didx;
|
|
memcpy(dptr, sptr, copylen);
|
|
dptr += copylen;
|
|
while (padlen-- > 0) {
|
|
*dptr++ = pad;
|
|
}
|
|
|
|
return dptr - dptr0;
|
|
}
|
|
|
|
static int
|
|
unpack_a(mrb_state *mrb, const void *src, int slen, mrb_value ary, long count, unsigned int flags)
|
|
{
|
|
mrb_value dst;
|
|
const char *sptr;
|
|
char *dptr, *dptr0;
|
|
|
|
sptr = src;
|
|
|
|
if (count == -1 || count > slen)
|
|
count = slen;
|
|
|
|
dst = mrb_str_new(mrb, NULL, count);
|
|
dptr0 = dptr = RSTRING_PTR(dst);
|
|
|
|
memcpy(dptr, sptr, count);
|
|
if (flags & PACK_FLAG_Z) {
|
|
if (count > 0 && dptr[count - 1] == '\0')
|
|
count--;
|
|
} else if (!(flags & PACK_FLAG_a)) {
|
|
while (count > 0 && (dptr[count - 1] == '\0' || isspace(dptr[count - 1])))
|
|
count--;
|
|
}
|
|
|
|
dst = mrb_str_resize(mrb, dst, count);
|
|
mrb_ary_push(mrb, ary, dst);
|
|
return count;
|
|
}
|
|
|
|
|
|
static int
|
|
pack_h(mrb_state *mrb, mrb_value src, mrb_value dst, mrb_int didx, long count, unsigned int flags)
|
|
{
|
|
unsigned int a, ashift, b, bshift;
|
|
int slen;
|
|
char *dptr, *dptr0, *sptr;
|
|
|
|
sptr = RSTRING_PTR(src);
|
|
slen = RSTRING_LEN(src);
|
|
|
|
if (flags & PACK_FLAG_LSB) {
|
|
ashift = 0;
|
|
bshift = 4;
|
|
} else {
|
|
ashift = 4;
|
|
bshift = 0;
|
|
}
|
|
|
|
if (count == -1) {
|
|
count = slen;
|
|
} else if (slen > count) {
|
|
slen = count;
|
|
}
|
|
|
|
dst = str_len_ensure(mrb, dst, didx + count);
|
|
dptr = RSTRING_PTR(dst) + didx;
|
|
|
|
dptr0 = dptr;
|
|
for (; count > 0; count -= 2) {
|
|
a = b = 0;
|
|
if (slen > 0) {
|
|
a = hex2int(*sptr++);
|
|
slen--;
|
|
}
|
|
if (slen > 0) {
|
|
b = hex2int(*sptr++);
|
|
slen--;
|
|
}
|
|
*dptr++ = (a << ashift) + (b << bshift);
|
|
}
|
|
|
|
return dptr - dptr0;
|
|
}
|
|
|
|
static int
|
|
unpack_h(mrb_state *mrb, const void *src, int slen, mrb_value ary, int count, unsigned int flags)
|
|
{
|
|
mrb_value dst;
|
|
int a, ashift, b, bshift, padding;
|
|
const char *sptr, *sptr0;
|
|
char *dptr, *dptr0;
|
|
const char hexadecimal[] = "0123456789abcdef";
|
|
|
|
if (flags & PACK_FLAG_LSB) {
|
|
ashift = 0;
|
|
bshift = 4;
|
|
} else {
|
|
ashift = 4;
|
|
bshift = 0;
|
|
}
|
|
|
|
sptr = src;
|
|
|
|
if (count == -1)
|
|
count = slen * 2;
|
|
|
|
dst = mrb_str_new(mrb, NULL, count);
|
|
dptr = RSTRING_PTR(dst);
|
|
|
|
sptr0 = sptr;
|
|
dptr0 = dptr;
|
|
padding = 0;
|
|
while (slen > 0 && count > 0) {
|
|
a = (*sptr >> ashift) & 0x0f;
|
|
b = (*sptr >> bshift) & 0x0f;
|
|
sptr++;
|
|
slen--;
|
|
|
|
*dptr++ = hexadecimal[a];
|
|
count--;
|
|
|
|
if (count > 0) {
|
|
*dptr++ = hexadecimal[b];
|
|
count--;
|
|
}
|
|
}
|
|
|
|
dst = mrb_str_resize(mrb, dst, dptr - dptr0);
|
|
mrb_ary_push(mrb, ary, dst);
|
|
return sptr - sptr0;
|
|
}
|
|
|
|
|
|
static int
|
|
pack_m(mrb_state *mrb, mrb_value src, mrb_value dst, mrb_int didx, long count, unsigned int flags)
|
|
{
|
|
mrb_int dstlen;
|
|
unsigned long l;
|
|
int column, srclen;
|
|
char *srcptr, *dstptr, *dstptr0;
|
|
|
|
srcptr = RSTRING_PTR(src);
|
|
srclen = RSTRING_LEN(src);
|
|
|
|
if (srclen == 0) /* easy case */
|
|
return 0;
|
|
|
|
if (count != 0 && count < 3) { /* -1, 1 or 2 */
|
|
count = 45;
|
|
} else if (count >= 3) {
|
|
count -= count % 3;
|
|
}
|
|
|
|
dstlen = srclen / 3 * 4;
|
|
if (count > 0) {
|
|
dstlen += (srclen / count) + ((srclen % count) == 0 ? 0 : 1);
|
|
}
|
|
dst = str_len_ensure(mrb, dst, didx + dstlen);
|
|
dstptr = RSTRING_PTR(dst) + didx;
|
|
|
|
dstptr0 = dstptr;
|
|
for (column = 3; srclen >= 3; srclen -= 3, column += 3) {
|
|
l = (unsigned char)*srcptr++ << 16;
|
|
l += (unsigned char)*srcptr++ << 8;
|
|
l += (unsigned char)*srcptr++;
|
|
|
|
*dstptr++ = base64chars[(l >> 18) & 0x3f];
|
|
*dstptr++ = base64chars[(l >> 12) & 0x3f];
|
|
*dstptr++ = base64chars[(l >> 6) & 0x3f];
|
|
*dstptr++ = base64chars[ l & 0x3f];
|
|
|
|
if (column == count) {
|
|
*dstptr++ = '\n';
|
|
column = 0;
|
|
}
|
|
}
|
|
if (srclen == 1) {
|
|
l = (unsigned char)*srcptr++ << 16;
|
|
*dstptr++ = base64chars[(l >> 18) & 0x3f];
|
|
*dstptr++ = base64chars[(l >> 12) & 0x3f];
|
|
*dstptr++ = '=';
|
|
*dstptr++ = '=';
|
|
column += 3;
|
|
} else if (srclen == 2) {
|
|
l = (unsigned char)*srcptr++ << 16;
|
|
l += (unsigned char)*srcptr++ << 8;
|
|
*dstptr++ = base64chars[(l >> 18) & 0x3f];
|
|
*dstptr++ = base64chars[(l >> 12) & 0x3f];
|
|
*dstptr++ = base64chars[(l >> 6) & 0x3f];
|
|
*dstptr++ = '=';
|
|
column += 3;
|
|
}
|
|
if (column > 0 && count > 0) {
|
|
*dstptr++ = '\n';
|
|
}
|
|
|
|
return dstptr - dstptr0;
|
|
}
|
|
|
|
static int
|
|
unpack_m(mrb_state *mrb, const void *src, int slen, mrb_value ary, unsigned int flags)
|
|
{
|
|
mrb_value dst;
|
|
int dlen;
|
|
unsigned long l;
|
|
int i, padding;
|
|
unsigned char c, ch[4];
|
|
const char *sptr, *sptr0;
|
|
char *dptr, *dptr0;
|
|
|
|
sptr0 = sptr = src;
|
|
|
|
dlen = slen / 4 * 3; /* an estimated value - may be shorter */
|
|
dst = mrb_str_new(mrb, NULL, dlen);
|
|
dptr0 = dptr = RSTRING_PTR(dst);
|
|
|
|
padding = 0;
|
|
while (slen >= 4) {
|
|
for (i = 0; i < 4; i++) {
|
|
do {
|
|
if (slen-- == 0)
|
|
goto done;
|
|
c = *sptr++;
|
|
if (c >= sizeof(base64_dec_tab))
|
|
continue;
|
|
ch[i] = base64_dec_tab[c];
|
|
if (ch[i] == PACK_BASE64_PADDING) {
|
|
ch[i] = 0;
|
|
padding++;
|
|
}
|
|
} while (ch[i] == PACK_BASE64_IGNORE);
|
|
}
|
|
|
|
l = (ch[0] << 18) + (ch[1] << 12) + (ch[2] << 6) + ch[3];
|
|
|
|
if (padding == 0) {
|
|
*dptr++ = (l >> 16) & 0xff;
|
|
*dptr++ = (l >> 8) & 0xff;
|
|
*dptr++ = l & 0xff;
|
|
} else if (padding == 1) {
|
|
*dptr++ = (l >> 16) & 0xff;
|
|
*dptr++ = (l >> 8) & 0xff;
|
|
break;
|
|
} else {
|
|
*dptr++ = (l >> 16) & 0xff;
|
|
break;
|
|
}
|
|
}
|
|
|
|
done:
|
|
dst = mrb_str_resize(mrb, dst, dptr - dptr0);
|
|
mrb_ary_push(mrb, ary, dst);
|
|
return sptr - sptr0;
|
|
}
|
|
|
|
|
|
static void
|
|
prepare_tmpl(mrb_state *mrb, struct tmpl *tmpl)
|
|
{
|
|
mrb_get_args(mrb, "S", &tmpl->str);
|
|
tmpl->idx = 0;
|
|
}
|
|
|
|
static int
|
|
has_tmpl(const struct tmpl *tmpl)
|
|
{
|
|
return (tmpl->idx < RSTRING_LEN(tmpl->str));
|
|
}
|
|
|
|
static void
|
|
read_tmpl(mrb_state *mrb, struct tmpl *tmpl, int *dirp, int *typep, int *sizep, long *countp, unsigned int *flagsp)
|
|
{
|
|
int ch, dir, t, tlen, type;
|
|
int size = 0;
|
|
long count = 1;
|
|
unsigned int flags = 0;
|
|
const char *tptr;
|
|
|
|
tptr = RSTRING_PTR(tmpl->str);
|
|
tlen = RSTRING_LEN(tmpl->str);
|
|
|
|
t = tptr[tmpl->idx++];
|
|
switch (t) {
|
|
case 'A':
|
|
dir = PACK_DIR_STR;
|
|
type = PACK_TYPE_STRING;
|
|
flags |= PACK_FLAG_WIDTH | PACK_FLAG_COUNT2;
|
|
break;
|
|
case 'a':
|
|
dir = PACK_DIR_STR;
|
|
type = PACK_TYPE_STRING;
|
|
flags |= PACK_FLAG_WIDTH | PACK_FLAG_COUNT2 | PACK_FLAG_a;
|
|
break;
|
|
case 'C':
|
|
dir = PACK_DIR_CHAR;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 1;
|
|
break;
|
|
case 'c':
|
|
dir = PACK_DIR_CHAR;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 1;
|
|
flags |= PACK_FLAG_SIGNED;
|
|
break;
|
|
case 'E':
|
|
dir = PACK_DIR_DOUBLE;
|
|
type = PACK_TYPE_FLOAT;
|
|
size = 8;
|
|
flags |= PACK_FLAG_SIGNED;
|
|
break;
|
|
case 'H':
|
|
dir = PACK_DIR_HEX;
|
|
type = PACK_TYPE_STRING;
|
|
flags |= PACK_FLAG_COUNT2;
|
|
break;
|
|
case 'h':
|
|
dir = PACK_DIR_HEX;
|
|
type = PACK_TYPE_STRING;
|
|
flags |= PACK_FLAG_COUNT2 | PACK_FLAG_LSB;
|
|
break;
|
|
case 'L':
|
|
dir = PACK_DIR_LONG;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 4;
|
|
break;
|
|
case 'l':
|
|
dir = PACK_DIR_LONG;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 4;
|
|
flags |= PACK_FLAG_SIGNED;
|
|
break;
|
|
case 'm':
|
|
dir = PACK_DIR_BASE64;
|
|
type = PACK_TYPE_STRING;
|
|
flags |= PACK_FLAG_WIDTH;
|
|
break;
|
|
case 'N': /* = "L>" */
|
|
dir = PACK_DIR_LONG;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 4;
|
|
flags |= PACK_FLAG_GT;
|
|
break;
|
|
case 'n': /* = "S>" */
|
|
dir = PACK_DIR_SHORT;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 2;
|
|
flags |= PACK_FLAG_GT;
|
|
break;
|
|
case 'S':
|
|
dir = PACK_DIR_SHORT;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 2;
|
|
break;
|
|
case 's':
|
|
dir = PACK_DIR_SHORT;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 2;
|
|
flags |= PACK_FLAG_SIGNED;
|
|
break;
|
|
case 'V': /* = "L<" */
|
|
dir = PACK_DIR_LONG;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 4;
|
|
flags |= PACK_FLAG_LT;
|
|
break;
|
|
case 'v': /* = "S<" */
|
|
dir = PACK_DIR_SHORT;
|
|
type = PACK_TYPE_INTEGER;
|
|
size = 2;
|
|
flags |= PACK_FLAG_LT;
|
|
break;
|
|
case 'Z':
|
|
dir = PACK_DIR_STR;
|
|
type = PACK_TYPE_STRING;
|
|
flags |= PACK_FLAG_WIDTH | PACK_FLAG_COUNT2 | PACK_FLAG_Z;
|
|
break;
|
|
default:
|
|
dir = PACK_DIR_INVALID;
|
|
break;
|
|
}
|
|
|
|
/* read suffix [0-9*_!<>] */
|
|
while (tmpl->idx < tlen) {
|
|
ch = tptr[tmpl->idx++];
|
|
if (isdigit(ch)) {
|
|
count = ch - '0';
|
|
while (tmpl->idx < tlen && isdigit(tptr[tmpl->idx])) {
|
|
count = count * 10 + (tptr[tmpl->idx++] - '0');
|
|
}
|
|
continue; /* special case */
|
|
} else if (ch == '*') {
|
|
count = -1;
|
|
} else if (ch == '_' || ch == '!' || ch == '<' || ch == '>') {
|
|
if (strchr("sSiIlL", t) == NULL) {
|
|
mrb_raisef(mrb, E_ARGUMENT_ERROR, "'%c' allowed only after types sSiIlL", ch);
|
|
}
|
|
if (ch == '_' || ch == '!') {
|
|
flags |= PACK_FLAG_s;
|
|
} else if (ch == '<') {
|
|
flags |= PACK_FLAG_LT;
|
|
} else if (ch == '>') {
|
|
flags |= PACK_FLAG_GT;
|
|
}
|
|
} else {
|
|
tmpl->idx--;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if ((flags & PACK_FLAG_LT) || (!(flags & PACK_FLAG_GT) && littleendian)) {
|
|
flags |= PACK_FLAG_LITTLEENDIAN;
|
|
}
|
|
|
|
*dirp = dir;
|
|
*typep = type;
|
|
*sizep = size;
|
|
*countp = count;
|
|
*flagsp = flags;
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_pack_pack(mrb_state *mrb, mrb_value ary)
|
|
{
|
|
mrb_value o, result;
|
|
mrb_int aidx;
|
|
struct tmpl tmpl;
|
|
long count;
|
|
unsigned int flags;
|
|
int dir, ridx, size, type;
|
|
|
|
prepare_tmpl(mrb, &tmpl);
|
|
|
|
result = mrb_str_new(mrb, NULL, 128); /* allocate initial buffer */
|
|
aidx = 0;
|
|
ridx = 0;
|
|
while (has_tmpl(&tmpl)) {
|
|
read_tmpl(mrb, &tmpl, &dir, &type, &size, &count, &flags);
|
|
|
|
if (dir == PACK_DIR_INVALID)
|
|
continue;
|
|
|
|
for (; aidx < RARRAY_LEN(ary); aidx++) {
|
|
if (count == 0 && !(flags & PACK_FLAG_WIDTH))
|
|
break;
|
|
|
|
o = mrb_ary_ref(mrb, ary, aidx);
|
|
if (type == PACK_TYPE_INTEGER) {
|
|
if (mrb_float_p(o)) {
|
|
o = mrb_funcall(mrb, o, "to_i", 0);
|
|
} else if (!mrb_fixnum_p(o)) {
|
|
mrb_raisef(mrb, E_TYPE_ERROR, "can't convert %s into Integer", mrb_obj_classname(mrb, o));
|
|
}
|
|
} else if (type == PACK_TYPE_FLOAT) {
|
|
if (!mrb_float_p(o)) {
|
|
o = mrb_funcall(mrb, o, "to_f", 0);
|
|
}
|
|
} else if (type == PACK_TYPE_STRING) {
|
|
if (!mrb_string_p(o)) {
|
|
mrb_raisef(mrb, E_TYPE_ERROR, "can't convert %s into String", mrb_obj_classname(mrb, o));
|
|
}
|
|
}
|
|
|
|
switch (dir) {
|
|
case PACK_DIR_CHAR:
|
|
ridx += pack_c(mrb, o, result, ridx, flags);
|
|
break;
|
|
case PACK_DIR_SHORT:
|
|
ridx += pack_s(mrb, o, result, ridx, flags);
|
|
break;
|
|
case PACK_DIR_LONG:
|
|
ridx += pack_l(mrb, o, result, ridx, flags);
|
|
break;
|
|
case PACK_DIR_BASE64:
|
|
ridx += pack_m(mrb, o, result, ridx, count, flags);
|
|
break;
|
|
case PACK_DIR_HEX:
|
|
ridx += pack_h(mrb, o, result, ridx, count, flags);
|
|
break;
|
|
case PACK_DIR_STR:
|
|
ridx += pack_a(mrb, o, result, ridx, count, flags);
|
|
break;
|
|
case PACK_DIR_DOUBLE:
|
|
ridx += pack_double(mrb, o, result, ridx, flags);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
if (count > 0) {
|
|
count--;
|
|
}
|
|
}
|
|
}
|
|
|
|
mrb_str_resize(mrb, result, ridx);
|
|
return result;
|
|
}
|
|
|
|
static mrb_value
|
|
mrb_pack_unpack(mrb_state *mrb, mrb_value str)
|
|
{
|
|
mrb_value result;
|
|
struct tmpl tmpl;
|
|
long count;
|
|
unsigned int flags;
|
|
int dir, sidx, size, srclen, type;
|
|
const char *sptr;
|
|
|
|
prepare_tmpl(mrb, &tmpl);
|
|
|
|
sidx = 0;
|
|
srclen = RSTRING_LEN(str);
|
|
|
|
result = mrb_ary_new(mrb);
|
|
while (has_tmpl(&tmpl)) {
|
|
read_tmpl(mrb, &tmpl, &dir, &type, &size, &count, &flags);
|
|
|
|
if (dir == PACK_DIR_INVALID)
|
|
continue;
|
|
|
|
if (flags & PACK_FLAG_COUNT2) {
|
|
sptr = RSTRING_PTR(str) + sidx;
|
|
switch (dir) {
|
|
case PACK_DIR_HEX:
|
|
sidx += unpack_h(mrb, sptr, srclen - sidx, result, count, flags);
|
|
break;
|
|
case PACK_DIR_STR:
|
|
sidx += unpack_a(mrb, sptr, srclen - sidx, result, count, flags);
|
|
break;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
while (count != 0) {
|
|
if (srclen - sidx < size) {
|
|
while (count-- > 0) {
|
|
mrb_ary_push(mrb, result, mrb_nil_value());
|
|
}
|
|
break;
|
|
}
|
|
|
|
sptr = RSTRING_PTR(str) + sidx;
|
|
switch (dir) {
|
|
case PACK_DIR_CHAR:
|
|
sidx += unpack_c(mrb, sptr, srclen - sidx, result, flags);
|
|
break;
|
|
case PACK_DIR_SHORT:
|
|
sidx += unpack_s(mrb, sptr, srclen - sidx, result, flags);
|
|
break;
|
|
case PACK_DIR_LONG:
|
|
sidx += unpack_l(mrb, sptr, srclen - sidx, result, flags);
|
|
break;
|
|
case PACK_DIR_BASE64:
|
|
sidx += unpack_m(mrb, sptr, srclen - sidx, result, flags);
|
|
break;
|
|
}
|
|
if (count > 0) {
|
|
count--;
|
|
}
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
void
|
|
mrb_mruby_pack_gem_init(mrb_state *mrb)
|
|
{
|
|
struct RClass *cArray, *cString;
|
|
|
|
littleendian = check_little_endian();
|
|
make_base64_dec_tab();
|
|
|
|
cArray = mrb->array_class;
|
|
cString = mrb->string_class;
|
|
|
|
mrb_define_method(mrb, cArray, "pack", mrb_pack_pack, MRB_ARGS_REQ(1));
|
|
mrb_define_method(mrb, cString, "unpack", mrb_pack_unpack, MRB_ARGS_REQ(1));
|
|
}
|
|
|
|
void
|
|
mrb_mruby_pack_gem_final(mrb_state *mrb)
|
|
{
|
|
}
|