fix potential segfault in para_client
[paraslash.git] / wav.c
1 /*
2 * Copyright (C) 2005-2006 Andre Noll <maan@systemlinux.org>
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
17 */
18
19 /** \file wav.c a filter that inserts a wave header */
20
21 #include "para.h"
22
23 #include "list.h"
24 #include "filter.h"
25 #include "string.h"
26
27 #define WAV_OUTBUF_SIZE 81920
28 #define WAV_HEADER_LEN 44
29 #define BITS 16
30
31 /* derived from oggdec.c of the vorbis-tools-1.0.1 package */
32 #define WRITE_U32(buf, x) (buf)[0] = (unsigned char)((x) & 0xff);\
33 (buf)[1] = (unsigned char)(((x) >> 8) & 0xff);\
34 (buf)[2] = (unsigned char)(((x) >> 16) & 0xff);\
35 (buf)[3] = (unsigned char)(((x) >> 24) & 0xff);
36 #define WRITE_U16(buf, x) (buf)[0] = (unsigned char)((x) & 0xff);
37
38 static void make_wav_header(int channels, int samplerate, struct filter_node *fn)
39 {
40
41 char *headbuf = fn->buf;
42 unsigned int size = 0x7fffffff;
43 int bytespersec = channels * samplerate * BITS / 8;
44 int align = channels * BITS / 8;
45
46 PARA_DEBUG_LOG("writing wave header: %d channels, %d KHz\n", channels, samplerate);
47 memset(headbuf, 0, WAV_HEADER_LEN);
48 memcpy(headbuf, "RIFF", 4);
49 WRITE_U32(headbuf + 4, size - 8);
50 memcpy(headbuf + 8, "WAVE", 4);
51 memcpy(headbuf + 12, "fmt ", 4);
52 WRITE_U32(headbuf + 16, 16);
53 WRITE_U16(headbuf + 20, 1); /* format */
54 WRITE_U16(headbuf + 22, channels);
55 WRITE_U32(headbuf + 24, samplerate);
56 WRITE_U32(headbuf + 28, bytespersec);
57 WRITE_U16(headbuf + 32, align);
58 WRITE_U16(headbuf + 34, BITS);
59 memcpy(headbuf + 36, "data", 4);
60 WRITE_U32(headbuf + 40, size - 44);
61 }
62
63 static ssize_t wav_convert(char *inbuf, size_t len, struct filter_node *fn)
64 {
65 size_t copy;
66 int *bof = fn->private_data;
67
68 if (*bof) {
69 make_wav_header(fn->fci->channels, fn->fci->samplerate, fn);
70 fn->loaded = WAV_HEADER_LEN;
71 *bof = 0;
72 // return 0;
73 }
74 copy = MIN(len, fn->bufsize - fn->loaded);
75 memmove(fn->buf + fn->loaded, inbuf, copy);
76 fn->loaded += copy;
77 // PARA_DEBUG_LOG("len = %d, copy = %d\n", len, copy);
78 return copy;
79 }
80
81 static void wav_close(struct filter_node *fn)
82 {
83 free(fn->buf);
84 fn->buf = NULL;
85 free(fn->private_data);
86 fn->private_data = NULL;
87 }
88
89 static void wav_open(struct filter_node *fn)
90 {
91 int *bof;
92
93 fn->bufsize = WAV_OUTBUF_SIZE;
94 fn->buf = para_malloc(fn->bufsize);
95 fn->private_data = para_malloc(sizeof(int));
96 bof = fn->private_data;
97 *bof = 1;
98 PARA_DEBUG_LOG("wav filter node: %p, output buffer: %p, loaded: %zd\n",
99 fn, fn->buf, fn->loaded);
100 }
101
102 void wav_init(struct filter *f)
103 {
104 f->convert = wav_convert;
105 f->close = wav_close;
106 f->open = wav_open;
107 }