change lamedb format.. more compatible with old enigma format... please use the new...
[vuplus_dvbapp] / lib / dvb / db.cpp
1 #include <errno.h>
2 #include <lib/dvb/db.h>
3 #include <lib/dvb/frontend.h>
4 #include <lib/base/eerror.h>
5 #include <lib/base/estring.h>
6 #include <dvbsi++/service_description_section.h>
7 #include <dvbsi++/descriptor_tag.h>
8 #include <dvbsi++/service_descriptor.h>
9 #include <dvbsi++/satellite_delivery_system_descriptor.h>
10
11 DEFINE_REF(eDVBService);
12
13 eDVBService::eDVBService()
14 {
15 }
16
17 eDVBService::~eDVBService()
18 {
19 }
20
21 eDVBService &eDVBService::operator=(const eDVBService &s)
22 {
23         m_service_name = s.m_service_name;
24         m_service_name_sort = s.m_service_name_sort;
25         m_provider_name = s.m_provider_name;
26         m_flags = s.m_flags;
27         m_ca = s.m_ca;
28         m_cache = s.m_cache;
29         return *this;
30 }
31
32 RESULT eDVBService::getName(const eServiceReference &ref, std::string &name)
33 {
34         name = m_service_name;
35         return 0;
36 }
37
38 int eDVBService::getLength(const eServiceReference &ref)
39 {
40         return -1;
41 }
42
43 int eDVBService::checkFilter(const eServiceReferenceDVB &ref, const eDVBChannelQuery &query)
44 {
45         int res = 0;
46         switch (query.m_type)
47         {
48         case eDVBChannelQuery::tName:
49                 res = m_service_name_sort.find(query.m_string) != std::string::npos;
50                 break;
51         case eDVBChannelQuery::tProvider:
52                 res = m_provider_name.find(query.m_string) != std::string::npos;
53                 break;
54         case eDVBChannelQuery::tType:
55                 res = ref.getServiceType() == query.m_int;
56                 break;
57         case eDVBChannelQuery::tBouquet:
58                 res = 0;
59                 break;
60         case eDVBChannelQuery::tSatellitePosition:
61                 res = (ref.getDVBNamespace().get() >> 16) == query.m_int;
62                 break;
63         case eDVBChannelQuery::tChannelID:
64         {
65                 eDVBChannelID chid;
66                 ref.getChannelID(chid);
67                 res = chid == query.m_channelid;
68                 break;
69         }
70         case eDVBChannelQuery::tAND:
71                 res = checkFilter(ref, *query.m_p1) && checkFilter(ref, *query.m_p2);
72                 break;
73         case eDVBChannelQuery::tOR:
74                 res = checkFilter(ref, *query.m_p1) || checkFilter(ref, *query.m_p2);
75                 break;
76         }
77
78         if (query.m_inverse)
79                 return !res;
80         else
81                 return res;
82 }
83
84 int eDVBService::getCachePID(cacheID id)
85 {
86         std::map<int, int>::iterator it = m_cache.find(id);
87         if ( it != m_cache.end() )
88                 return it->second;
89         return -1;
90 }
91
92 void eDVBService::setCachePID(cacheID id, int pid)
93 {
94         m_cache[id] = pid;
95 }
96
97 DEFINE_REF(eDVBDB);
98
99         /* THIS CODE IS BAD. it should be replaced by somethine better. */
100 void eDVBDB::load()
101 {
102         eDebug("---- opening lame channel db");
103         FILE *f=fopen("lamedb", "rt");
104         if (!f)
105                 return;
106         char line[256];
107         if ((!fgets(line, 256, f)) || strncmp(line, "eDVB services", 13))
108         {
109                 eDebug("not a servicefile");
110                 fclose(f);
111                 return;
112         }
113         eDebug("reading services");
114         if ((!fgets(line, 256, f)) || strcmp(line, "transponders\n"))
115         {
116                 eDebug("services invalid, no transponders");
117                 fclose(f);
118                 return;
119         }
120
121         // clear all transponders
122
123         while (!feof(f))
124         {
125                 if (!fgets(line, 256, f))
126                         break;
127                 if (!strcmp(line, "end\n"))
128                         break;
129                 int dvb_namespace=-1, transport_stream_id=-1, original_network_id=-1;
130                 sscanf(line, "%x:%x:%x", &dvb_namespace, &transport_stream_id, &original_network_id);
131                 if (original_network_id == -1)
132                         continue;
133                 eDVBChannelID channelid = eDVBChannelID(
134                         eDVBNamespace(dvb_namespace),
135                         eTransportStreamID(transport_stream_id),
136                         eOriginalNetworkID(original_network_id));
137
138                 ePtr<eDVBFrontendParameters> feparm = new eDVBFrontendParameters;
139                 while (!feof(f))
140                 {
141                         fgets(line, 256, f);
142                         if (!strcmp(line, "/\n"))
143                                 break;
144                         if (line[1]=='s')
145                         {
146                                 eDVBFrontendParametersSatellite sat;
147                                 int frequency, symbol_rate, polarisation, fec, orbital_position, inversion;
148                                 sscanf(line+2, "%d:%d:%d:%d:%d:%d", &frequency, &symbol_rate, &polarisation, &fec, &orbital_position, &inversion);
149                                 sat.frequency = frequency;
150                                 sat.symbol_rate = symbol_rate;
151                                 sat.polarisation = polarisation;
152                                 sat.fec = fec;
153                                 sat.orbital_position = orbital_position;
154                                 sat.inversion = inversion;
155                                 // ...
156 //                              t.setSatellite(frequency, symbol_rate, polarisation, fec, sat, inversion);
157                                 feparm->setDVBS(sat);
158                         } else if (line[1]=='t')
159                         {
160                                 eDVBFrontendParametersTerrestrial ter;
161                                 int frequency, bandwidth, code_rate_HP, code_rate_LP, modulation, transmission_mode, guard_interval, hierarchy, inversion;
162                                 sscanf(line+2, "%d:%d:%d:%d:%d:%d:%d:%d:%d", &frequency, &bandwidth, &code_rate_HP, &code_rate_LP, &modulation, &transmission_mode, &guard_interval, &hierarchy, &inversion);
163                                 ter.frequency = frequency;
164                                 ter.bandwidth = bandwidth;
165                                 ter.code_rate_HP = code_rate_HP;
166                                 ter.code_rate_LP = code_rate_LP;
167                                 ter.modulation = modulation;
168                                 ter.transmission_mode = transmission_mode;
169                                 ter.guard_interval = guard_interval;
170                                 ter.hierarchy = hierarchy;
171                                 ter.inversion = inversion;
172                                 
173                                 feparm->setDVBT(ter);
174                         } else if (line[1]=='c')
175                         {
176                                 int frequency, symbol_rate, inversion=0, modulation=3;
177                                 sscanf(line+2, "%d:%d:%d:%d", &frequency, &symbol_rate, &inversion, &modulation);
178 //                              t.setCable(frequency, symbol_rate, inversion, modulation);
179                         }
180                 }
181                 addChannelToList(channelid, feparm);
182         }
183
184         if ((!fgets(line, 256, f)) || strcmp(line, "services\n"))
185         {
186                 eDebug("services invalid, no services");
187                 return;
188         }
189
190         // clear all services
191
192         int count=0;
193
194         while (!feof(f))
195         {
196                 if (!fgets(line, 256, f))
197                         break;
198                 if (!strcmp(line, "end\n"))
199                         break;
200
201                 int service_id=-1, dvb_namespace, transport_stream_id=-1, original_network_id=-1, service_type=-1, service_number=-1;
202                 sscanf(line, "%x:%x:%x:%x:%d:%d", &service_id, &dvb_namespace, &transport_stream_id, &original_network_id, &service_type, &service_number);
203                 if (service_number == -1)
204                         continue;
205                 ePtr<eDVBService> s = new eDVBService;
206                 eServiceReferenceDVB ref =
207                                                 eServiceReferenceDVB(
208                                                 eDVBNamespace(dvb_namespace),
209                                                 eTransportStreamID(transport_stream_id),
210                                                 eOriginalNetworkID(original_network_id),
211                                                 eServiceID(service_id),
212                                                 service_type);
213                 count++;
214                 fgets(line, 256, f);
215                 if (strlen(line))
216                         line[strlen(line)-1]=0;
217
218                 s->m_service_name = line;
219                 s->m_service_name_sort = removeDVBChars(line);
220                 makeUpper(s->m_service_name_sort);
221
222                 fgets(line, 256, f);
223                 if (strlen(line))
224                         line[strlen(line)-1]=0;
225
226                 std::string str=line;
227
228                 if (str[1]!=':')        // old ... (only service_provider)
229                 {
230                         s->m_provider_name=line;
231                 } else
232                         while ((!str.empty()) && str[1]==':') // new: p:, f:, c:%02d...
233                         {
234                                 unsigned int c=str.find(',');
235                                 char p=str[0];
236                                 std::string v;
237                                 if (c == std::string::npos)
238                                 {
239                                         v=str.substr(2);
240                                         str="";
241                                 } else
242                                 {
243                                         v=str.substr(2, c-2);
244                                         str=str.substr(c+1);
245                                 }
246 //                              eDebug("%c ... %s", p, v.c_str());
247                                 if (p == 'p')
248                                         s->m_provider_name=v;
249                                 else if (p == 'f')
250                                 {
251                                         sscanf(v.c_str(), "%x", &s->m_flags);
252                                 } else if (p == 'c')
253                                 {
254                                         int cid, val;
255                                         sscanf(v.c_str(), "%02d%04x", &cid, &val);
256                                         s->m_cache[cid]=val;
257                                 } else if (p == 'C')
258                                 {
259                                         int val;
260                                         sscanf(v.c_str(), "%04x", &val);
261                                         s->m_ca.insert(val);
262                                 }
263                         }
264                 addService(ref, s);
265         }
266
267         eDebug("loaded %d services", count);
268
269         fclose(f);
270 }
271
272 void eDVBDB::save()
273 {
274         eDebug("---- saving lame channel db");
275         FILE *f=fopen("lamedb", "wt");
276         int channels=0, services=0;
277         if (!f)
278                 eFatal("couldn't save lame channel db!");
279         fprintf(f, "eDVB services /3/\n");
280         fprintf(f, "transponders\n");
281         for (std::map<eDVBChannelID, channel>::const_iterator i(m_channels.begin());
282                         i != m_channels.end(); ++i)
283         {
284                 const eDVBChannelID &chid = i->first;
285                 const channel &ch = i->second;
286
287                 fprintf(f, "%08x:%04x:%04x\n", chid.dvbnamespace.get(),
288                                 chid.transport_stream_id.get(), chid.original_network_id.get());
289                 eDVBFrontendParametersSatellite sat;
290                 eDVBFrontendParametersTerrestrial ter;
291                 if (!ch.m_frontendParameters->getDVBS(sat))
292                 {
293                         fprintf(f, "\ts %d:%d:%d:%d:%d:%d\n",
294                                 sat.frequency, sat.symbol_rate,
295                                 sat.polarisation, sat.fec, sat.orbital_position,
296                                 sat.inversion);
297                 }
298                 if (!ch.m_frontendParameters->getDVBT(ter))
299                 {
300                         fprintf(f, "\tt %d:%d:%d:%d:%d:%d:%d:%d:%d\n",
301                                 ter.frequency, ter.bandwidth, ter.code_rate_HP,
302                                 ter.code_rate_LP, ter.modulation, ter.transmission_mode,
303                                 ter.guard_interval, ter.hierarchy, ter.inversion);
304                 }
305                 fprintf(f, "/\n");
306                 channels++;
307         }
308         fprintf(f, "end\nservices\n");
309
310         for (std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator i(m_services.begin());
311                 i != m_services.end(); ++i)
312         {
313                 const eServiceReferenceDVB &s = i->first;
314                 fprintf(f, "%04x:%08x:%04x:%04x:%d:%d\n",
315                                 s.getServiceID().get(), s.getDVBNamespace().get(),
316                                 s.getTransportStreamID().get(),s.getOriginalNetworkID().get(),
317                                 s.getServiceType(),
318                                 0);
319
320                 fprintf(f, "%s\n", i->second->m_service_name.c_str());
321                 fprintf(f, "p:%s", i->second->m_provider_name.c_str());
322
323                 // write cached pids
324                 for (std::map<int,int>::const_iterator ca(i->second->m_cache.begin());
325                         ca != i->second->m_cache.end(); ++ca)
326                         fprintf(f, ",c:%02d%04x", ca->first, ca->second);
327
328                 // write cached ca pids
329                 for (std::set<int>::const_iterator ca(i->second->m_ca.begin());
330                         ca != i->second->m_ca.end(); ++ca)
331                         fprintf(f, ",C:%04x", *ca);
332
333                 fprintf(f, "\n");
334                 services++;
335         }
336         fprintf(f, "end\nHave a lot of bugs!\n");
337         eDebug("saved %d channels and %d services!", channels, services);
338         fclose(f);
339 }
340
341 eDVBDB::eDVBDB()
342 {
343         load();
344 }
345
346 eDVBDB::~eDVBDB()
347 {
348 //      save();
349 }
350
351 RESULT eDVBDB::addChannelToList(const eDVBChannelID &id, iDVBFrontendParameters *feparm)
352 {
353         channel ch;
354         assert(feparm);
355         ch.m_frontendParameters = feparm;
356         m_channels.insert(std::pair<eDVBChannelID, channel>(id, ch));
357         return 0;
358 }
359
360 RESULT eDVBDB::removeChannel(const eDVBChannelID &id)
361 {
362         m_channels.erase(id);
363         return 0;
364 }
365
366 RESULT eDVBDB::getChannelFrontendData(const eDVBChannelID &id, ePtr<iDVBFrontendParameters> &parm)
367 {
368         std::map<eDVBChannelID, channel>::iterator i = m_channels.find(id);
369         if (i == m_channels.end())
370         {
371                 parm = 0;
372                 return -ENOENT;
373         }
374         parm = i->second.m_frontendParameters;
375         return 0;
376 }
377
378 RESULT eDVBDB::addService(const eServiceReferenceDVB &serviceref, eDVBService *service)
379 {
380         m_services.insert(std::pair<eServiceReferenceDVB, ePtr<eDVBService> >(serviceref, service));
381         return 0;
382 }
383
384 RESULT eDVBDB::getService(const eServiceReferenceDVB &reference, ePtr<eDVBService> &service)
385 {
386         std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator i;
387         i = m_services.find(reference);
388         if (i == m_services.end())
389         {
390                 service = 0;
391                 return -ENOENT;
392         }
393         service = i->second;
394         return 0;
395 }
396
397 RESULT eDVBDB::startQuery(ePtr<iDVBChannelListQuery> &query, eDVBChannelQuery *q)
398 {
399         query = new eDVBDBQuery(this, eServiceReference(), q);
400         return 0;
401 }
402
403 DEFINE_REF(eDVBDBQuery);
404
405 eDVBDBQuery::eDVBDBQuery(eDVBDB *db, const eServiceReference &source, eDVBChannelQuery *query): m_db(db), m_query(query)
406 {
407                 // TODO: use SOURCE ...
408         m_cursor = m_db->m_services.begin();
409 }
410
411 RESULT eDVBDBQuery::getNextResult(eServiceReferenceDVB &ref)
412 {
413         while (m_cursor != m_db->m_services.end())
414         {
415                 ref = m_cursor->first;
416                 
417                 int res = (!m_query) || m_cursor->second->checkFilter(ref, *m_query);
418
419                 ++m_cursor;
420                 
421                 if (res)
422                         return 0;
423         }
424         
425         ref = eServiceReferenceDVB();
426         return 1;
427 }
428
429 int eDVBDBQuery::compareLessEqual(const eServiceReferenceDVB &a, const eServiceReferenceDVB &b)
430 {
431         ePtr<eDVBService> a_service, b_service;
432         
433         int sortmode = m_query ? m_query->m_sort : eDVBChannelQuery::tName;
434         
435         if ((sortmode == eDVBChannelQuery::tName) || (sortmode == eDVBChannelQuery::tProvider))
436         {
437                 if (m_db->getService(a, a_service))
438                         return 1;
439                 if (m_db->getService(b, b_service))
440                         return 1;
441         }
442         
443         switch (sortmode)
444         {
445         case eDVBChannelQuery::tName:
446                 return a_service->m_service_name_sort < b_service->m_service_name_sort;
447         case eDVBChannelQuery::tProvider:
448                 return a_service->m_provider_name < b_service->m_provider_name;
449         case eDVBChannelQuery::tType:
450                 return a.getServiceType() < b.getServiceType();
451         case eDVBChannelQuery::tBouquet:
452                 return 1;
453         case eDVBChannelQuery::tSatellitePosition:
454                 return (a.getDVBNamespace().get() >> 16) < (b.getDVBNamespace().get() >> 16);
455         default:
456                 return 1;
457         }
458         return 0;
459 }
460
461 /* (<name|provider|type|bouquet|satpos|chid> <==|...> <"string"|int>)[||,&& (..)] */
462
463 static int decodeType(const std::string &type)
464 {
465         if (type == "name")
466                 return eDVBChannelQuery::tName;
467         else if (type == "provider")
468                 return eDVBChannelQuery::tProvider;
469         else if (type == "type")
470                 return eDVBChannelQuery::tType;
471         else if (type == "bouquet")
472                 return eDVBChannelQuery::tBouquet;
473         else if (type == "satellitePosition")
474                 return eDVBChannelQuery::tSatellitePosition;
475         else if (type == "channelID")
476                 return eDVBChannelQuery::tChannelID;
477         else
478                 return -1;
479 }
480
481         /* never, NEVER write a parser in C++! */
482 RESULT parseExpression(ePtr<eDVBChannelQuery> &res, std::list<std::string>::const_iterator begin, std::list<std::string>::const_iterator end)
483 {
484         std::list<std::string>::const_iterator end_of_exp;
485         if (*begin == "(")
486         {
487                 end_of_exp = begin;
488                 while (end_of_exp != end)
489                         if (*end_of_exp == ")")
490                                 break;
491                         else
492                                 ++end_of_exp;
493         
494                 if (end_of_exp == end)
495                 {
496                         eDebug("expression parse: end of expression while searching for closing brace");
497                         return -1;
498                 }
499                 
500                 ++begin;
501                 // begin..end_of_exp
502                 int r = parseExpression(res, begin, end_of_exp);
503                 if (r)
504                         return r;
505                 ++end_of_exp;
506                 
507                         /* we had only one sub expression */
508                 if (end_of_exp == end)
509                 {
510                         eDebug("only one sub expression");
511                         return 0;
512                 }
513                 
514                         /* otherwise we have an operator here.. */
515                 
516                 ePtr<eDVBChannelQuery> r2 = res;
517                 res = new eDVBChannelQuery();
518                 res->m_sort = 0;
519                 res->m_p1 = r2;
520                 res->m_inverse = 0;
521                 r2 = 0;
522                 
523                 if (*end_of_exp == "||")
524                         res->m_type = eDVBChannelQuery::tOR;
525                 else if (*end_of_exp == "&&")
526                         res->m_type = eDVBChannelQuery::tAND;
527                 else
528                 {
529                         eDebug("found operator %s, but only && and || are allowed!", end_of_exp->c_str());
530                         res = 0;
531                         return 1;
532                 }
533                 
534                 ++end_of_exp;
535                 
536                 return parseExpression(res->m_p2, end_of_exp, end);
537         }
538         
539         // "begin" <op> "end"
540         std::string type, op, val;
541         
542         res = new eDVBChannelQuery();
543         res->m_sort = 0;
544         
545         int cnt = 0;
546         while (begin != end)
547         {
548                 switch (cnt)
549                 {
550                 case 0:
551                         type = *begin;
552                         break;
553                 case 1:
554                         op = *begin;
555                         break;
556                 case 2:
557                         val = *begin;
558                         break;
559                 case 3:
560                         eDebug("malformed query: got '%s', but expected only <type> <op> <val>", begin->c_str());
561                         return 1;
562                 }
563                 ++begin;
564                 ++cnt;
565         }
566         
567         if (cnt != 3)
568         {
569                 eDebug("malformed query: missing stuff");
570                 res = 0;
571                 return 1;
572         }
573         
574         res->m_type = decodeType(type);
575         
576         if (res->m_type == -1)
577         {
578                 eDebug("malformed query: invalid type %s", type.c_str());
579                 res = 0;
580                 return 1;
581         }
582         
583         if (op == "==")
584                 res->m_inverse = 0;
585         else if (op == "!=")
586                 res->m_inverse = 1;
587         else
588         {
589                 eDebug("invalid operator %s", op.c_str());
590                 res = 0;
591                 return 1;
592         }
593         
594         res->m_string = val;
595         res->m_int = atoi(val.c_str());
596 //      res->m_channelid = eDVBChannelID(val);
597         
598         return 0;
599 }
600
601 RESULT eDVBChannelQuery::compile(ePtr<eDVBChannelQuery> &res, std::string query)
602 {
603         std::list<std::string> tokens;
604         
605         std::string current_token;
606         
607         eDebug("splitting %s....", query.c_str());
608         unsigned int i = 0;
609         const char *splitchars="()";
610         int quotemode = 0, lastsplit = 0, lastalnum = 0;
611         while (i <= query.size())
612         {
613                 int c = (i < query.size()) ? query[i] : ' ';
614                 ++i;
615                 
616                 int issplit = !!strchr(splitchars, c);
617                 int isaln = isalnum(c);
618                 int iswhite = c == ' ';
619                 int isquot = c == '\"';
620                 
621                 if (quotemode)
622                 {
623                         iswhite = issplit = 0;
624                         isaln = lastalnum;
625                 }
626                 
627                 if (issplit || iswhite || isquot || lastsplit || (lastalnum != isaln))
628                 {
629                         if (current_token.size())
630                                 tokens.push_back(current_token);
631                         current_token.clear();
632                 }
633                 
634                 if (!(iswhite || isquot))
635                         current_token += c;
636                 
637                 if (isquot)
638                         quotemode = !quotemode;
639                 lastsplit = issplit;
640                 lastalnum = isaln;
641         }
642         
643 //      for (std::list<std::string>::const_iterator a(tokens.begin()); a != tokens.end(); ++a)
644 //      {
645 //              printf("%s\n", a->c_str());
646 //      }
647
648         int sort = eDVBChannelQuery::tName;
649                 /* check for "ORDER BY ..." */
650         if (tokens.size() > 2)
651         {
652                 std::list<std::string>::iterator i = tokens.end();
653                 --i; --i; --i;
654                 if (*i == "ORDER")
655                 {
656                         ++i;
657                         if (*i == "BY")
658                         {
659                                 ++i;
660                                 sort = decodeType(*i);
661                                 tokens.pop_back(); // ...
662                                 tokens.pop_back(); // BY
663                                 tokens.pop_back(); // ORDER
664                         } else
665                                 sort = -1;
666                 }
667         }
668         
669         if (sort == -1)
670         {       
671                 eWarning("ORDER BY .. string invalid.");
672                 res = 0;
673                 return -1;
674         }
675         
676         eDebug("sort by %d", sort);
677         
678                 /* now we recursivly parse that. */
679         int r = parseExpression(res, tokens.begin(), tokens.end());
680         
681         if (res)
682                 res->m_sort = sort;
683
684         eDebug("return: %d", r);
685         return r;
686 }
687
688 DEFINE_REF(eDVBChannelQuery);