1//=========================================================================
2// FILENAME	: tagutils-wav.c
3// DESCRIPTION	: WAV metadata reader
4//=========================================================================
5// Copyright (c) 2009 NETGEAR, Inc. All Rights Reserved.
6// based on software from Ron Pedde's FireFly Media Server project
7//=========================================================================
8
9/*
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
14 *
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18 * GNU General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License
21 * along with this program. If not, see <http://www.gnu.org/licenses/>.
22 */
23
24#define GET_WAV_INT32(p) ((((uint8_t)((p)[3])) << 24) |   \
25			  (((uint8_t)((p)[2])) << 16) |   \
26			  (((uint8_t)((p)[1])) << 8) |	   \
27			  (((uint8_t)((p)[0]))))
28
29#define GET_WAV_INT16(p) ((((uint8_t)((p)[1])) << 8) |	   \
30			  (((uint8_t)((p)[0]))))
31
32static int
33_get_wavtags(char *filename, struct song_metadata *psong)
34{
35	int fd;
36	uint32_t len;
37	unsigned char hdr[12];
38	unsigned char fmt[16];
39	//uint32_t chunk_data_length;
40	uint32_t format_data_length = 0;
41	uint32_t compression_code = 0;
42	uint32_t channel_count = 0;
43	uint32_t sample_rate = 0;
44	uint32_t sample_bit_length = 0;
45	uint32_t bit_rate;
46	uint32_t data_length = 0;
47	uint32_t sec, ms;
48
49	uint32_t current_offset;
50	uint32_t block_len;
51
52	//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Getting WAV file info\n");
53
54	if(!(fd = open(filename, O_RDONLY)))
55	{
56		DPRINTF(E_WARN, L_SCANNER, "Could not create file handle\n");
57		return -1;
58	}
59
60	len = 12;
61	if(!(len = read(fd, hdr, len)) || (len != 12))
62	{
63		DPRINTF(E_WARN, L_SCANNER, "Could not read wav header from %s\n", filename);
64		close(fd);
65		return -1;
66	}
67
68	/* I've found some wav files that have INFO tags
69	 * in them... */
70	if(strncmp((char*)hdr + 0, "RIFF", 4) ||
71	   strncmp((char*)hdr + 8, "WAVE", 4))
72	{
73		DPRINTF(E_WARN, L_SCANNER, "Invalid wav header in %s\n", filename);
74		close(fd);
75		return -1;
76	}
77
78	//chunk_data_length = GET_WAV_INT32(hdr + 4);
79
80	/* now, walk through the chunks */
81	current_offset = 12;
82	while(current_offset + 8 < psong->file_size)
83	{
84		len = 8;
85		if(!(len = read(fd, hdr, len)) || (len != 8))
86		{
87			close(fd);
88			DPRINTF(E_WARN, L_SCANNER, "Error reading block: %s\n", filename);
89			return -1;
90		}
91
92		current_offset += 8;
93		block_len = GET_WAV_INT32(hdr + 4);
94
95		//DEBUG DPRINTF(E_DEBUG, L_SCANNER, "Read block %02x%02x%02x%02x (%c%c%c%c) of "
96		//        "size 0x%08x\n",hdr[0],hdr[1],hdr[2],hdr[3],
97		//        isprint(hdr[0]) ? hdr[0] : '?',
98		//        isprint(hdr[1]) ? hdr[1] : '?',
99		//        isprint(hdr[2]) ? hdr[2] : '?',
100		//        isprint(hdr[3]) ? hdr[3] : '?',
101		//        block_len);
102
103		if(block_len > psong->file_size)
104		{
105			close(fd);
106			DPRINTF(E_WARN, L_SCANNER, "Bad block len: %s\n", filename);
107			return -1;
108		}
109
110		if(strncmp((char*)&hdr, "fmt ", 4) == 0)
111		{
112			//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Found 'fmt ' header\n");
113			len = 16;
114			if(read(fd, fmt, len) != len)
115			{
116				close(fd);
117				DPRINTF(E_WARN, L_SCANNER, "Bad .wav file: can't read fmt: %s\n",
118					filename);
119				return -1;
120			}
121
122			format_data_length = block_len;
123			compression_code = GET_WAV_INT16(fmt);
124			channel_count = GET_WAV_INT16(fmt + 2);
125			sample_rate = GET_WAV_INT32(fmt + 4);
126			sample_bit_length = GET_WAV_INT16(fmt + 14);
127			//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Compression code: %d\n",compression_code);
128			//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Channel count:    %d\n",channel_count);
129			//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Sample Rate:      %d\n",sample_rate);
130			//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Sample bit length %d\n",sample_bit_length);
131
132		}
133		else if(strncmp((char*)&hdr, "data", 4) == 0)
134		{
135			//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Found 'data' header\n");
136			data_length = block_len;
137			goto next_block;
138		}
139		else if(strncmp((char*)&hdr, "LIST", 4) == 0)
140		{
141			char *tags;
142			char *p;
143			int off;
144			uint32_t taglen;
145			char **m;
146
147			len = GET_WAV_INT32(hdr + 4);
148			if(len > 65536 || len < 9)
149				goto next_block;
150
151			tags = malloc(len+1);
152			if(!tags)
153				goto next_block;
154
155			if(read(fd, tags, len) < len ||
156			   strncmp(tags, "INFO", 4) != 0)
157			{
158				free(tags);
159				goto next_block;
160			}
161			tags[len] = '\0';
162
163			off = 4;
164			p = tags + off;
165			while(off < len - 8)
166			{
167				taglen = GET_WAV_INT32(p + 4);
168
169				//DEBUG DPRINTF(E_DEBUG, L_SCANNER, "%.*s: %.*s (%d)\n", 4, p, taglen, p + 8, taglen);
170				m = NULL;
171				if (taglen > 2048) {
172					DPRINTF(E_WARN, L_SCANNER, "Ignoring long tag [%.*s] in %s\n",
173				                4, p+8, filename);
174				}
175				else if(strncmp(p, "INAM", 4) == 0)
176					m = &(psong->title);
177				else if(strncmp(p, "IALB", 4) == 0 ||
178				        strncmp(p, "IPRD", 4) == 0)
179					m = &(psong->album);
180				else if(strncmp(p, "IGRE", 4) == 0 ||
181				        strncmp(p, "IGNR", 4) == 0)
182					m = &(psong->genre);
183				else if(strncmp(p, "ICMT", 4) == 0)
184					m = &(psong->comment);
185				else if(strncmp(p, "IART", 4) == 0)
186					m = &(psong->contributor[ROLE_TRACKARTIST]);
187				else if(strncmp(p, "IAAR", 4) == 0)
188					m = &(psong->contributor[ROLE_ALBUMARTIST]);
189				else if(strncmp(p, "ICOM", 4) == 0 ||
190				        strncmp(p, "IMUS", 4) == 0)
191					m = &(psong->contributor[ROLE_COMPOSER]);
192				else if(strncasecmp(p, "ITRK", 4) == 0)
193					psong->track = atoi(p + 8);
194				else if(strncmp(p, "ICRD", 4) == 0 ||
195				        strncmp(p, "IYER", 4) == 0)
196					psong->year = atoi(p + 8);
197				if(m)
198				{
199					*m = malloc(taglen + 1);
200					strncpy(*m, p + 8, taglen);
201					(*m)[taglen] = '\0';
202				}
203
204				p += taglen + 8;
205				off += taglen + 8;
206				/* Handle some common WAV file malformations */
207				while (*p == '\0' && off < len) {
208					p++;
209					off++;
210				}
211			}
212			free(tags);
213		}
214next_block:
215		lseek(fd, current_offset + block_len, SEEK_SET);
216		current_offset += block_len;
217	}
218	close(fd);
219
220	if(((format_data_length != 16) && (format_data_length != 18)) ||
221	   (compression_code != 1) ||
222	   (channel_count < 1))
223	{
224		DPRINTF(E_WARN, L_SCANNER, "Invalid wav header in %s\n", filename);
225		return -1;
226	}
227
228	if( !data_length )
229		data_length = psong->file_size - 44;
230
231	bit_rate = sample_rate * channel_count * ((sample_bit_length + 7) / 8) * 8;
232	psong->bitrate = bit_rate;
233	psong->samplerate = sample_rate;
234	psong->channels = channel_count;
235	//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Data length: %d\n", data_length);
236	sec = data_length / (bit_rate / 8);
237	ms = ((data_length % (bit_rate / 8)) * 1000) / (bit_rate / 8);
238	psong->song_length = (sec * 1000) + ms;
239	//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Song length: %d\n", psong->song_length);
240	//DEBUG DPRINTF(E_DEBUG,L_SCANNER,"Bit rate: %d\n", psong->bitrate);
241
242	/* Upon further review, WAV files should be little-endian, and DLNA requires the LPCM profile to be big-endian.
243	asprintf(&(psong->mime), "audio/L16;rate=%d;channels=%d", psong->samplerate, psong->channels);
244	*/
245
246	return 0;
247}
248
249static int
250_get_wavfileinfo(char *filename, struct song_metadata *psong)
251{
252	psong->lossless = 1;
253	/* Upon further review, WAV files should be little-endian, and DLNA requires the LPCM profile to be big-endian.
254	asprintf(&(psong->dlna_pn), "LPCM");
255	*/
256
257	return 0;
258}
259