ExtendLib/xLib/asn1_tool.cpp

Go to the documentation of this file.

#include "asn1_tool.h"


// ASN.1/DER
//

int  skip_DER_node(Buffer param, unsigned char node, int ls, int* lp) 
{
    int i, sz=-1;

    if ((unsigned char)param.buf[ls]==node) {
        if ((unsigned char)param.buf[ls+1]>=0x80) {
            sz = (int)param.buf[ls+1] - 128;
            *lp = 0;
            for (i=ls+2; i<ls+sz+2; i++) {
                *lp *= 256;
                *lp += (unsigned char)param.buf[i];
            }    
        }
        else {
            sz = 0;
            *lp = (int)param.buf[ls+1];
        }
        //if ((unsigned char)param.buf[ls]==JBXL_ASN1_BIT) sz++;
        sz = ls + sz + 2;
    }

    return sz;
}



Buffer  node2DER(Buffer pt, unsigned char node)
{
    int  sz, len, pp;
    unsigned char cnt;
    Buffer    buf;

    len = get_size_toDER(pt, node);  
    buf = make_Buffer(len);
    sz = pt.vldsz;
    pp = len - sz;                  // ノード値の格納場所の先頭
    memcpy(buf.buf+pp, pt.buf, sz);

    pp--;
    if (node==JBXL_ASN1_INT) {
/*
        int2bin_DER() で処理済み
        if (pt.buf[0]>=0x80) {
            sz++;
            buf.buf[pp--] = 0x00;
        }
*/
    }
    else if (node==JBXL_ASN1_BIT) {
        sz++;
        buf.buf[pp--] = 0x00;
    }

    buf.buf[0] = node;

    cnt = 0;
    if (sz>=128) {
        cnt++;
        while (sz>=256) {
            buf.buf[pp--] = sz % 256;
            sz >>= 8;
            cnt++;
        }
        buf.buf[pp] = (unsigned char)sz;
        buf.buf[1]  = 0x80 + cnt;
    }
    else {
        buf.buf[1] = (unsigned char)sz;
    }

    buf.vldsz = len;
    return buf;
}



Buffer  int2bin_DER(long int n)
{
    int  ii, cnt;
    unsigned long int div;
    Buffer bin;

    cnt = 1;
    div = n;
    while(div>=256) {
        div >>= 8;
        cnt++;
    }
    bin = make_Buffer(cnt+1);           // cnt : バイト数
    ii = cnt - 1;

    while(n>=256) {
        bin.buf[ii--] = (unsigned char)(n % 256);
        n >>= 8;
    }
    bin.buf[0] = (unsigned char)n;

    // 負数ではない場合 先頭に 0x00 を追加
    if (n>0 && bin.buf[0]>=0x80) {     
        for (ii=cnt; ii>0; ii--) bin.buf[ii] = bin.buf[ii-1];
        bin.buf[0] = 0x00;
        cnt++;
    }

    bin.vldsz  = cnt;

    return bin;
}



long int  bin2int_DER(Buffer buf)
{
    if (buf.buf==NULL || buf.vldsz==0) return 0;

    int i;
    int sz = buf.vldsz;
    unsigned char* pp = buf.buf;

    int minus = OFF;
    if (*pp & 0x80) {   // 負数
        minus = ON;
        for (i=0; i<sz; i++) pp[i] = pp[i] ^ 0xff;  // bit反転
    }

    long int ret = pp[sz-1];
    for (i=sz-2; i>=0; i--) {
        ret += pp[i] * 256;
    }
    if (minus) ret = - (ret + 1);

    return ret;
}




int  get_size_toDER(Buffer pt, unsigned char node)
{
    int  sz, div, cnt;

    sz = pt.vldsz;
    if (node==JBXL_ASN1_BIT) sz++;    // 未使用bit数を格納する先頭データ用

    cnt = 1;                    // ノード長のバイト数
    div = sz;
    if (div>=128) cnt++;        // ノード長のバイト数が2以上の場合
    while (div>=256) {
        div >>= 8;
        cnt++;
    }
    sz = sz + cnt + 1;          // 1=>node
    return sz;
}




//
//

tDER*  DER_parse(tDER* der, Buffer* buf)
{
    if (buf==NULL || buf->buf==NULL) return NULL;

    if (der==NULL) {
        der = new_DER_node();
        der->ldat.id = JBXL_ASN1_ANCHOR;
        der->ldat.lv = buf->vldsz;
        der->state   = JBXL_STATE_ANCHOR;
    }
    
    int sz = get_DER_size(buf->buf, NULL);
    if (sz > buf->vldsz) return NULL;     // データが短い

    _DER_parse_children(der, buf);
    
    return der;
}



void  _DER_parse_children(tDER* der, Buffer* buf)
{
    if (der==NULL) return;
    if (buf==NULL) buf = &(der->ldat.val);
    if (buf->buf==NULL) return;

    unsigned char* pp = buf->buf;

    int pos = 0;
    do {
        int sz = get_DER_size(pp+pos, NULL);
        if (sz > buf->vldsz - pos) return; 

        tDER* tmp = new_DER_node();
        int len = set_DER_node(tmp, pp+pos); 
        if (len==0) {
            del_DER_node(&tmp);
            return;
        }
        //
        add_DER_node(der, tmp);
        if (tmp->ldat.id & JBXL_ASN1_CNSTRCTD) {
            _DER_parse_children(tmp, NULL);
            free_Buffer(&(tmp->ldat.val));      // メモリが勿体ないので解放
        }
        pos += len;

    } while (pos<buf->vldsz);
    
    return;
}



int  set_DER_node(tDER* der, unsigned char* buf)
{
    if (der==NULL || buf==NULL) return 0;

    int len = 0;
    int cnt = get_DER_size(buf, &len);

    der->ldat.id  = (int)(*buf);         // buf[0] Tag
    der->ldat.lv  = cnt;
    der->ldat.val = set_Buffer((void*)(buf+cnt-len), len);

    return cnt;
}



int  get_DER_size(unsigned char* buf, int* valsz)
{
    if (buf==NULL) return 0;

    int cnt = 0;
    int len = 0;
    cnt++;
    buf++;
    if (*buf>=0x80) {
        int i;
        int sz = (int)(*buf - 0x80);
        cnt++;
        buf++;
        for (i=0; i<sz; i++) {
            len = len*256 + (int)(*buf);
            cnt++;
            buf++;
        } 
    }
    else {
        len  = (int)(*buf);
        cnt++;
        buf++;
    }
    
    if (valsz!=NULL) *valsz = len;
    cnt += len;

    return cnt;
}



void  print_tDER(FILE* fp, tDER* pp)
{
    if (fp==NULL) fp = stderr;

    if (pp!=NULL) {
        while(pp->esis!=NULL) pp = pp->esis;
        do {
            int i;
            tList_data ld = pp->ldat;

            for(i=0; i<pp->depth; i++) fprintf(fp, "    ");
            if (pp->depth>0) fprintf(fp, " -> ");
            fprintf(fp, "%d: ", pp->depth);
            asn1_print_node(fp, ld.id);
            fprintf(fp, "%d,%d ", ld.lv, ld.val.vldsz);
            asn1_print_node_value(fp, ld.id, ld.val);
            fprintf(fp, "\n");

            if (pp->next!=NULL) print_tDER(fp, pp->next);

            pp = pp->ysis;
            //if (pp!=NULL) {
            //    for(i=1; i<pp->depth-1; i++) fprintf(fp, "           ");      // for " -> "
            //}
        } while(pp!=NULL);
    }
    else {
        fprintf(fp, "(Tree is NULL)\n");
    }
    fflush(fp);

    return;
}



/*
Buffer* DER_inverse_parse(tDER* pp);
{
    Buffer* buf = new_Buffer(0);

    return buf;
}
*/

---