wma_common.c

   1 /*
   2  * Copyright (C) 2009 Andre Noll <maan@systemlinux.org>
   3  *
   4  * Licensed under the GPL v2. For licencing details see COPYING.
   5  */
   6
   7 /** \file wma_common.c Functions used by both the WMA afh and decoder. */
   8
   9 #include <inttypes.h>
  10 #include <sys/types.h>
  11 #include <sys/stat.h>
  12 #include <fcntl.h>
  13 #include <stdio.h>
  14 #include <unistd.h>
  15 #include <string.h>
  16
  17 #include "para.h"
  18 #include "error.h"
  19 #include "afh.h"
  20 #include "portable_io.h"
  21 #include "imdct.h"
  22 #include "wma.h"
  23
  24 const char *search_pattern(const char *pattern, int pattern_len, const char *buf, int buf_size)
  25 {
  26         const char *p, *end = buf + buf_size;
  27
  28         if (pattern_len > buf_size)
  29                 return NULL;
  30         for (p = buf; p + pattern_len < end; p++) {
  31                 if (memcmp(p, pattern, pattern_len))
  32                         continue;
  33                 PARA_DEBUG_LOG("found %d byte pattern@%d\n",
  34                         pattern_len, p - buf);
  35                 return p;
  36         }
  37         PARA_NOTICE_LOG("%d byte pattern not found\n", pattern_len);
  38         return NULL;
  39 }
  40
  41 /*
  42    40 9e 69 f8 4d 5b cf 11  a8 fd 00 80 5f 5c 44 2b
  43  */
  44 static int find_audio_stream_info(const char *buf, int len)
  45 {
  46         const char pattern[] = {0x40, 0x9e, 0x69, 0xf8};
  47         const char *p = search_pattern(pattern, sizeof(pattern), buf, len);
  48
  49         if (!p)
  50                 return -E_WMA_NO_GUID;
  51         PARA_DEBUG_LOG("found audio stream guid@%0zx\n", p - buf);
  52         return p - buf;
  53 }
  54
  55 static int read_header_len(char *buf, int len)
  56 {
  57         uint16_t header_len;
  58
  59         header_len = read_u16(buf + 16) + 46;
  60         if (header_len > len)
  61                 return -E_WMA_BAD_ASF_HEADER;
  62         PARA_DEBUG_LOG("header_len: %d\n", header_len);
  63         return header_len;
  64 }
  65
  66 int read_asf_header(char *buf, int loaded, struct asf_header_info *ahi)
  67 {
  68         int ret;
  69         char *start;
  70
  71         ret = read_header_len(buf, loaded);
  72         if (ret < 0)
  73                 return ret;
  74         ahi->header_len = ret;
  75
  76         ret = find_audio_stream_info(buf, ahi->header_len);
  77         if (ret < 0)
  78                 return ret;
  79         ahi->audio_stream_info_start = ret + 16;
  80         start = buf + ahi->audio_stream_info_start;
  81
  82         ahi->channels = ((uint8_t *)start)[40];
  83         ahi->sample_rate = read_u16(start + 42);
  84         PARA_NOTICE_LOG("%d channels, sample rate: %d\n", ahi->channels,
  85                 ahi->sample_rate);
  86
  87         ahi->bit_rate = 8 * read_u16(start + 46);
  88         PARA_INFO_LOG("bit rate: %d\n", ahi->bit_rate);
  89
  90         ahi->block_align = read_u16(start + 50);
  91         PARA_INFO_LOG("block_align: %d\n", ahi->block_align);
  92
  93         ahi->flags1 = read_u32(start + 56);
  94         ahi->flags2 = read_u16(start + 60);
  95         PARA_INFO_LOG("read_asf_header: flags1: %d, flag2: %d\n",
  96                 ahi->flags1, ahi->flags2);
  97         return 42;
  98 }
  99
 100 const uint8_t log2_tab[256] = {
 101         0, 0, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4,
 102         4, 4, 4, 4, 4, 4, 4, 4,
 103         5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
 104         5, 5, 5, 5, 5, 5, 5, 5,
 105         6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
 106         6, 6, 6, 6, 6, 6, 6, 6,
 107         6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
 108         6, 6, 6, 6, 6, 6, 6, 6,
 109         7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
 110         7, 7, 7, 7, 7, 7, 7, 7,
 111         7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
 112         7, 7, 7, 7, 7, 7, 7, 7,
 113         7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
 114         7, 7, 7, 7, 7, 7, 7, 7,
 115         7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7,
 116         7, 7, 7, 7, 7, 7, 7, 7
 117 };
 118
 119 int wma_log2(unsigned int v)
 120 {
 121         int n = 0;
 122         if (v & 0xffff0000) {
 123                 v >>= 16;
 124                 n += 16;
 125         }
 126         if (v & 0xff00) {
 127                 v >>= 8;
 128                 n += 8;
 129         }
 130         n += log2_tab[v];
 131
 132         return n;
 133 }