Git

1 /*\r
2  * FAPG 0.0 released under GPL\r
3  * http://royale.zerezo.com/fapg/\r
4  */\r
5 \r
6 #include <stdio.h>\r
7 #include <stdlib.h>\r
8 #include <getopt.h>\r
9 #include <dirent.h>\r
10 #include <sys/types.h>\r
11 #include <string.h>\r
12 #include <limits.h>\r
13 \r
14 #define MAX 10240\r
15 \r
16 int debug=0;\r
17 int format=0; /* 0 = m3u ; 1 = pls ; 2 = html */\r
18 unsigned char *prefix="";\r
19 int recursive=0;\r
20 int separator='/';\r
21 int skip=0;\r
22 int windows=0;\r
23 unsigned char buffer[MAX];\r
24 \r
25 int counter=0;\r
26 \r
27 unsigned char artist[1024];\r
28 unsigned char title[1024];\r
29 int time;\r
30 \r
31 unsigned char unix2dos[256]={0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,70,35,36,37,38,39,40,41,82,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,84,59,36,61,65,71,64,65,66,67,68,69,70,71,72,73,74,75,76,77,78,79,80,81,82,83,84,85,86,87,88,89,90,91,36,93,94,95,96,97,98,99,100,101,102,103,104,105,106,107,108,109,110,111,112,113,114,115,116,117,118,119,120,121,122,123,36,125,126,127,199,252,233,226,228,224,229,231,234,235,232,239,238,236,196,197,201,230,198,244,246,242,251,249,255,214,220,248,163,216,215,131,225,237,243,250,241,209,170,186,191,174,172,189,188,161,171,187,166,166,166,166,166,193,194,192,169,166,166,43,43,162,165,43,43,45,45,43,45,43,227,195,43,43,45,45,166,45,43,164,240,208,202,203,200,105,205,206,207,43,43,166,220,166,204,175,211,223,212,210,245,213,181,254,222,218,219,217,253,221,175,180,173,177,61,190,182,167,247,184,176,168,183,185,179,178,166,160};\r
32 \r
33 void usage()\r
34 {\r
35   fprintf(stderr,"Usage >> fapg [-backslash] [-debug] [-format=m3u|pls|html] [-output=/path/to/file.m3u] [-prefix=/the/prefix] [-recursive] [-windows] /path/to/mp3/dir1 [/path/to/mp3/dir2 ...]\n");\r
36   exit(1);\r
37 }\r
38 \r
39 void parse_options(int argc,char **argv)\r
40 {\r
41   static struct option long_options[]=\r
42   {\r
43     {"backslash",no_argument,&separator,'\\'},\r
44     {"debug",no_argument,&debug,1},\r
45     {"format",required_argument,0,'f'},\r
46     {"output",required_argument,0,'o'},\r
47     {"prefix",required_argument,0,'p'},\r
48     {"recursive",no_argument,&recursive,1},\r
49     {"windows",no_argument,&windows,1}\r
50   };\r
51   int c;\r
52   int option_index=0;\r
53   while ((c=getopt_long_only(argc,argv,"",long_options,&option_index))!=-1)\r
54   {\r
55     switch(c)\r
56     {\r
57       case 0:\r
58         break;\r
59       case 'f':\r
60         if (strcmp(optarg,"m3u")==0)  format=0; else\r
61         if (strcmp(optarg,"pls")==0)  format=1; else\r
62         if (strcmp(optarg,"html")==0) format=2; else\r
63         usage();\r
64         break;\r
65       case 'o':\r
66         close(1);\r
67         if (fopen(optarg,"w")==NULL) { fprintf(stderr,"Error >> unable to open output file : %s\n",optarg); exit(2); }\r
68         break;\r
69       case 'p':\r
70         prefix=malloc(strlen(optarg)+1);\r
71         strcpy(prefix,optarg);\r
72         break;\r
73       default:\r
74         usage();\r
75     }\r
76   }\r
77 }\r
78 \r
79 void parse_mp3(unsigned char *file)\r
80 {\r
81   int bitrates[2][3][15]=\r
82     {{{0, 32, 64, 96, 128, 160, 192, 224, 256, 288, 320, 352, 384, 416, 448},\r
83       {0, 32, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320, 384},\r
84       {0, 32, 40, 48, 56, 64, 80, 96, 112, 128, 160, 192, 224, 256, 320}},\r
85      {{0, 32, 48, 56, 64, 80, 96, 112, 128, 144, 160, 176, 192, 224, 256},\r
86       {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160},\r
87       {0, 8, 16, 24, 32, 40, 48, 56, 64, 80, 96, 112, 128, 144, 160}}};\r
88   FILE *fic;\r
89   unsigned char *c;\r
90   int lus;\r
91 \r
92   if (debug) fprintf(stderr,"Debug >> parsing mp3 : %s\n",file);\r
93 \r
94   /* read header */\r
95   if ((fic=fopen(file,"r"))==NULL) { fprintf(stderr,"Warning >> can't open file : %s\n",file); return; }\r
96   lus=fread(buffer,1,MAX,fic);\r
97   c=buffer;  \r
98 \r
99   /* try ID3v2 */\r
100   if (buffer[0]=='I' && buffer[1]=='D' && buffer[2]=='3')\r
101   {\r
102     int size;\r
103     int version;\r
104     version=*(buffer+3);\r
105     if (version<2 || version>4)\r
106       fprintf(stderr,"Warning >> ID3 v2.%d not implemented ! trying anyway : %s\n",version,file);\r
107     if (*(buffer+5)!=0)\r
108       fprintf(stderr,"Warning >> specials headers not implemented (%d) ! trying anyway : %s\n",*(buffer+5),file);\r
109     c=buffer+6;\r
110     size=(*c<<24)+(*(c+1)<<16)+(*(c+2)<<8)+(*(c+3));\r
111     if (size>lus) size=lus;\r
112     c+=4;\r
113     if (version==2) while (c<buffer+size-5)\r
114     {\r
115       int size=(*(c+3)<<16)+(*(c+4)<<8)+(*(c+5));\r
116       if (*c==0) break;\r
117       if (strncmp(c,"TT2",3)==0)\r
118       {\r
119         strncpy(title,c+7,size-1);\r
120         title[size-1]='\0';\r
121       }\r
122       if (strncmp(c,"TP1",3)==0)\r
123       {\r
124         strncpy(artist,c+7,size-1);\r
125         artist[size-1]='\0';\r
126       }\r
127       c+=size+6;\r
128     }\r
129     if (version==3 || version==4) while (c<buffer+size-7)\r
130     {\r
131       int size=(*(c+4)<<24)+(*(c+5)<<16)+(*(c+6)<<8)+(*(c+7));\r
132       if (*c==0) break;\r
133       if (strncmp(c,"TIT2",4)==0)\r
134       {\r
135         strncpy(title,c+11,size-1);\r
136         title[size-1]='\0';\r
137       }\r
138       if (strncmp(c,"TPE1",4)==0)\r
139       {\r
140         strncpy(artist,c+11,size-1);\r
141         artist[size-1]='\0';\r
142       }\r
143       c+=size+10;\r
144     }\r
145   }\r
146   \r
147   while (c<buffer+lus-10)\r
148   {\r
149     if (*c==0xFF && (*(c+1)&0xF0)==0xF0)\r
150     {\r
151       int version;\r
152       int lay;\r
153       int bitrate_index;\r
154       int bitrate;\r
155       version=2-(*(c+1)>>3&1);\r
156       lay=4-(*(c+1)>>1&3);\r
157       bitrate_index=*(c+2)>>4&0xF;\r
158       if (version>=1 && version<=2 && lay-1>=0 && lay-1<=2 && bitrate_index>=0 && bitrate_index<=14)\r
159         bitrate=bitrates[version-1][lay-1][bitrate_index];\r
160       else\r
161         bitrate=0;\r
162       if (bitrate!=0)\r
163       {\r
164         fseek(fic,0,SEEK_END);\r
165         time=(ftell(fic)+buffer-c)/125/bitrate;\r
166       }\r
167       else\r
168         time=0;\r
169       break;\r
170     }\r
171     c++;\r
172   }\r
173 \r
174   /* try ID3v1 */\r
175   if (strlen(artist)==0 && strlen(title)==0)\r
176   {\r
177     fseek(fic,-128,SEEK_END);\r
178     lus=fread(buffer,1,128,fic);\r
179     if (lus==128 && buffer[0]=='T' && buffer[1]=='A' && buffer[2]=='G')\r
180     {\r
181       strncpy(title,buffer+3,30);\r
182       title[30]='\0';\r
183       c=title+29;\r
184       while (c>title && *c==' ') *(c--)='\0';\r
185       strncpy(artist,buffer+33,30);\r
186       artist[30]='\0';\r
187       c=artist+29;\r
188       while (c>artist && *c==' ') *(c--)='\0';\r
189     }\r
190   }\r
191 \r
192   fclose(fic);\r
193 }  \r
194 \r
195 void parse_ogg(unsigned char *file)\r
196 {\r
197   FILE *fic;\r
198   unsigned char *c;\r
199   int lus;\r
200   int sample_rate;\r
201   int samples;\r
202 \r
203   if (debug) fprintf(stderr,"Debug >> parsing ogg : %s\n",file);\r
204 \r
205   /* read header */\r
206   if ((fic=fopen(file,"r"))==NULL) { fprintf(stderr,"Warning >> can't open file : %s\n",file); return; }\r
207   lus=fread(buffer,1,MAX,fic);\r
208 \r
209   /* try Ogg */\r
210   if (buffer[0]!='O' && buffer[1]!='g' && buffer[2]!='g')\r
211   {\r
212     fprintf(stderr,"Warning >> not a Ogg header : %s\n",file);\r
213     return;\r
214   }\r
215   \r
216   c=buffer+0x28;\r
217   sample_rate=(*c)+(*(c+1)<<8)+(*(c+2)<<16)+(*(c+3)<<24);\r
218 \r
219   while (c<buffer+lus-10)\r
220   {\r
221     int size;\r
222     if (strncmp(c,"TITLE=",6)==0)\r
223     {\r
224       size=*(c-4)+(*(c-3)<<8)+(*(c-2)<<16)+(*(c-1)<<24);\r
225       strncpy(title,c+6,size-6);\r
226       title[size-6]='\0';\r
227       c+=size;\r
228     }\r
229     if (strncmp(c,"ARTIST=",7)==0)\r
230     {\r
231       size=*(c-4)+(*(c-3)<<8)+(*(c-2)<<16)+(*(c-1)<<24);\r
232       strncpy(artist,c+7,size-7);\r
233       artist[size-7]='\0';\r
234       c+=size;\r
235     }\r
236     c++;\r
237   }\r
238   \r
239   fseek(fic,-MAX,SEEK_END);\r
240   lus=fread(buffer,1,MAX,fic);\r
241   c=buffer+lus-1;\r
242   while (strncmp(c,"OggS",4)!=0 && c>buffer) c--;\r
243   if (c!=buffer)\r
244   {\r
245     c+=6;\r
246     samples=(*c)+(*(c+1)<<8)+(*(c+2)<<16)+(*(c+3)<<24);\r
247     time=samples/sample_rate;\r
248   }\r
249   \r
250   fclose(fic);\r
251 }\r
252 \r
253 void parse_directory(unsigned char *path)\r
254 {\r
255   DIR *dirp;\r
256   struct dirent *dp;\r
257   unsigned char newpath[PATH_MAX];\r
258   unsigned char *c;\r
259 \r
260   void print_faketitle()\r
261   {\r
262     c=newpath+strlen(newpath);\r
263     while (c>newpath && *c!='/') c--;\r
264     while (c<newpath+strlen(newpath)-5)\r
265     {\r
266       c++;\r
267       if (*c=='_') putchar(' '); else if (windows) putchar(unix2dos[*c]); else putchar(*c);\r
268     }\r
269   }\r
270   \r
271   void print_path()\r
272   {\r
273     printf(prefix);\r
274     for (c=newpath+skip;*c!='\0';c++) if (*c=='/') putchar(separator); else if (windows) putchar(unix2dos[*c]); else putchar(*c);\r
275   }\r
276 \r
277   if (debug) fprintf(stderr,"Debug >> parsing directory : %s\n",path);\r
278   if ((dirp=opendir(path))==NULL) { fprintf(stderr,"Warning >> can't open directory : %s\n",path); return; }\r
279   while ((dp=readdir(dirp))!=NULL)\r
280   {\r
281     sprintf(newpath,"%s/%s",path,dp->d_name);\r
282     if (recursive && dp->d_type & DT_DIR && strcmp(dp->d_name,".")!=0 && strcmp(dp->d_name,"..")!=0)\r
283     {\r
284       parse_directory(newpath);\r
285     }\r
286     if (dp->d_type & DT_REG)\r
287     {\r
288       unsigned char ext[5];\r
289       int i;\r
290       for (i=0;i<5;i++) ext[i]=tolower(dp->d_name[strlen(dp->d_name)-4+i]);\r
291       artist[0]='\0';\r
292       title[0]='\0';\r
293       time=-2;\r
294       if (strcmp(".mp3",ext)==0) { time=-1; parse_mp3(newpath); }\r
295       if (strcmp(".ogg",ext)==0) { time=-1; parse_ogg(newpath); }\r
296       if (strcmp(".wav",ext)==0) { time=-1; /* parse_wav(newpath); */ }\r
297       \r
298       if (time!=-2) /* is it an audio file ? */\r
299       {\r
300         counter++;\r
301         switch (format)\r
302         {\r
303           case 0:\r
304             if (time!=-1)\r
305             {\r
306               printf("#EXTINF:%d,",time);\r
307               if (strlen(artist)==0 && strlen(title)==0) print_faketitle();\r
308               else printf("%s - %s",artist,title);\r
309               putchar('\n');\r
310             }\r
311             print_path();\r
312             putchar('\n');\r
313             break;\r
314           case 1:\r
315             printf("File%d=",counter);\r
316             print_path();\r
317             putchar('\n');\r
318             printf("Title%d=",counter);\r
319             if (strlen(artist)==0 && strlen(title)==0) print_faketitle();\r
320             else printf("%s - %s",artist,title);\r
321             putchar('\n');\r
322             if (time!=-1) printf("Length%d=%d\n",counter,time);\r
323             break;\r
324           case 2:\r
325             printf("<tr><td>%d</td><td>%s</td><td>%s</td><td>",counter,artist,title);\r
326             if (time==-1) printf("?"); else printf("%d",time);\r
327             printf("</td></tr>\n");\r
328             break;\r
329         }\r
330       }\r
331     }\r
332   }\r
333   closedir(dirp);\r
334 }\r
335 \r
336 int main(int argc,char **argv)\r
337 {\r
338   parse_options(argc,argv);\r
339   if (optind==argc) usage();\r
340   switch (format)\r
341   {\r
342     case 0:\r
343       printf("#EXTM3U\n");\r
344       break;\r
345     case 1:\r
346       printf("[playlist]\n");\r
347       break;\r
348     case 2:\r
349       printf("<!DOCTYPE html PUBLIC \"-//W3C//DTD XHTML 1.0 Strict//EN\" \"http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd\">\n\n<html>\n\n<head>\n<title>Playlist generated by FAPG 0.0</title>\n<meta http-equiv=\"Content-Type\" content=\"text/html; charset=iso-8859-1\" />\n<link rel=\"stylesheet\" type=\"text/css\" href=\"fapg.css\" />\n</head>\n\n<body>\n\n<h1>Playlist</h1>\n\n<table>\n<tr><th>Entry</th><th>Artist</th><th>Title</th><th>Length</th></tr>\n");\r
350       break;\r
351   }\r
352   for (;optind<argc;optind++)\r
353   {\r
354     if (strlen(prefix)!=0) skip=strlen(argv[optind]);\r
355     parse_directory(argv[optind]);\r
356   }\r
357   switch(format)\r
358   {\r
359     case 1:\r
360       printf("NumberOfEntries=%d\nVersion=2\n",counter);\r
361       break;\r
362     case 2:\r
363       printf("</table>\n\n<p>Playlist generated by <a href=\"http://royale.zerezo.com/fapg/\">FAPG 0.0</a></p>\n\n</body>\n\n</html>");\r
364       break;\r
365   }\r
366   exit(0);\r
367 }\r