src/lib/nrv2b.c

   1 // This GETBIT is supposed to work on little endian
   2 // 32bit systems. The algorithm will definitely need
   3 // some fixing on other systems, but it might not be
   4 // a problem since the nrv2b binary behaves the same..
   5
   6 #ifndef ENDIAN
   7 #define ENDIAN   0
   8 #endif
   9 #ifndef BITSIZE
  10 #define BITSIZE 32
  11 #endif
  12
  13 #define GETBIT_8(bb, src, ilen) \
  14     (((bb = bb & 0x7f ? bb*2 : ((unsigned)src[ilen++]*2+1)) >> 8) & 1)
  15
  16 #define GETBIT_LE16(bb, src, ilen) \
  17     (bb*=2,bb&0xffff ? (bb>>16)&1 : (ilen+=2,((bb=(src[ilen-2]+src[ilen-1]*256u)*2+1)>>16)&1))
  18 #define GETBIT_LE32(bb, src, ilen) \
  19     (bc > 0 ? ((bb>>--bc)&1) : (bc=31,\
  20     bb=*(const uint32_t *)((src)+ilen),ilen+=4,(bb>>31)&1))
  21
  22 #if ENDIAN == 0 && BITSIZE == 8
  23 #define GETBIT(bb, src, ilen) GETBIT_8(bb, src, ilen)
  24 #endif
  25 #if ENDIAN == 0 && BITSIZE == 16
  26 #define GETBIT(bb, src, ilen) GETBIT_LE16(bb, src, ilen)
  27 #endif
  28 #if ENDIAN == 0 && BITSIZE == 32
  29 #define GETBIT(bb, src, ilen) GETBIT_LE32(bb, src, ilen)
  30 #endif
  31
  32 static unsigned long unrv2b(uint8_t * src, uint8_t * dst, unsigned long *ilen_p)
  33 {
  34         unsigned long ilen = 0, olen = 0, last_m_off = 1;
  35         uint32_t bb = 0;
  36         unsigned bc = 0;
  37         const uint8_t *m_pos;
  38
  39         // skip length
  40         src += 4;
  41
  42         /* FIXME: check olen with len on first 4 bytes */
  43         for (;;) {
  44                 unsigned int m_off, m_len;
  45                 while (GETBIT(bb, src, ilen)) {
  46                         dst[olen++] = src[ilen++];
  47                 }
  48
  49                 m_off = 1;
  50                 do {
  51                         m_off = m_off * 2 + GETBIT(bb, src, ilen);
  52                 } while (!GETBIT(bb, src, ilen));
  53                 if (m_off == 2) {
  54                         m_off = last_m_off;
  55                 } else {
  56                         m_off = (m_off - 3) * 256 + src[ilen++];
  57                         if (m_off == 0xffffffffU)
  58                                 break;
  59                         last_m_off = ++m_off;
  60                 }
  61
  62                 m_len = GETBIT(bb, src, ilen);
  63                 m_len = m_len * 2 + GETBIT(bb, src, ilen);
  64                 if (m_len == 0) {
  65                         m_len++;
  66                         do {
  67                                 m_len = m_len * 2 + GETBIT(bb, src, ilen);
  68                         } while (!GETBIT(bb, src, ilen));
  69                         m_len += 2;
  70                 }
  71                 m_len += (m_off > 0xd00);
  72
  73                 m_pos = dst + olen - m_off;
  74                 dst[olen++] = *m_pos++;
  75                 do {
  76                         dst[olen++] = *m_pos++;
  77                 } while (--m_len > 0);
  78         }
  79
  80         *ilen_p = ilen;
  81
  82         return olen;
  83
  84 }
  85