add startmmi
[enigma2.git] / 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 RESULT eBouquet::addService(const eServiceReference &ref)
14 {
15         list::iterator it =
16                 std::find(m_services.begin(), m_services.end(), ref);
17         if ( it != m_services.end() )
18                 return -1;
19         m_services.push_back(ref);
20         return 0;
21 }
22
23 RESULT eBouquet::removeService(const eServiceReference &ref)
24 {
25         list::iterator it =
26                 std::find(m_services.begin(), m_services.end(), ref);
27         if ( it == m_services.end() )
28                 return -1;
29         m_services.erase(it);
30         return 0;
31 }
32
33 RESULT eBouquet::moveService(const eServiceReference &ref, unsigned int pos)
34 {
35         if ( pos < 0 || pos >= m_services.size() )
36                 return -1;
37         list::iterator source=m_services.end();
38         list::iterator dest=m_services.end();
39         for (list::iterator it(m_services.begin()); it != m_services.end(); ++it)
40         {
41                 if (dest == m_services.end() && !pos--)
42                         dest = it;
43                 if (*it == ref)
44                         source = it;
45                 if (dest != m_services.end() && source != m_services.end())
46                         break;
47         }
48         if (dest == m_services.end() || source == m_services.end() || source == dest)
49                 return -1;
50         std::iter_swap(source,dest);
51         return 0;
52 }
53
54 RESULT eBouquet::flushChanges()
55 {
56         FILE *f=fopen(m_path.c_str(), "wt");
57         if (!f)
58                 return -1;
59         if ( fprintf(f, "#NAME %s\r\n", m_bouquet_name.c_str()) < 0 )
60                 goto err;
61         for (list::iterator i(m_services.begin()); i != m_services.end(); ++i)
62         {
63                 eServiceReference tmp = *i;
64                 std::string str = tmp.path;
65                 if ( (i->flags&eServiceReference::flagDirectory) == eServiceReference::flagDirectory )
66                 {
67                         unsigned int p1 = str.find("FROM BOUQUET \"");
68                         if (p1 == std::string::npos)
69                         {
70                                 eDebug("doof... kaputt");
71                                 continue;
72                         }
73                         str.erase(0, p1+14);
74                         p1 = str.find("\"");
75                         if (p1 == std::string::npos)
76                         {
77                                 eDebug("doof2... kaputt");
78                                 continue;
79                         }
80                         str.erase(p1);
81                         tmp.path=str;
82                 }
83                 if ( fprintf(f, "#SERVICE %s\r\n", tmp.toString().c_str()) < 0 )
84                         goto err;
85                 if ( i->name.length() )
86                         if ( fprintf(f, "#DESCRIPTION %s\r\n", i->name.c_str()) < 0 )
87                                 goto err;
88         }
89         fclose(f);
90         return 0;
91 err:
92         fclose(f);
93         eDebug("couldn't write file %s", m_path.c_str());
94         return -1;
95 }
96
97 eDVBService::eDVBService()
98 {
99 }
100
101 eDVBService::~eDVBService()
102 {
103 }
104
105 eDVBService &eDVBService::operator=(const eDVBService &s)
106 {
107         m_service_name = s.m_service_name;
108         m_service_name_sort = s.m_service_name_sort;
109         m_provider_name = s.m_provider_name;
110         m_flags = s.m_flags;
111         m_ca = s.m_ca;
112         m_cache = s.m_cache;
113         return *this;
114 }
115
116 void eDVBService::genSortName()
117 {
118         m_service_name_sort = removeDVBChars(m_service_name);
119         makeUpper(m_service_name_sort);
120         while ((!m_service_name_sort.empty()) && m_service_name_sort[0] == ' ')
121                 m_service_name_sort.erase(0, 1);
122         
123                 /* put unnamed services at the end, not at the beginning. */
124         if (m_service_name_sort.empty())
125                 m_service_name_sort = "\xFF";
126 }
127
128 RESULT eDVBService::getName(const eServiceReference &ref, std::string &name)
129 {
130         if (!ref.name.empty())
131                 name = ref.name;
132         else if (!m_service_name.empty())
133                 name = m_service_name;
134         else
135                 name = "(...)";
136         return 0;
137 }
138
139 int eDVBService::getLength(const eServiceReference &ref)
140 {
141         return -1;
142 }
143
144 int eDVBService::checkFilter(const eServiceReferenceDVB &ref, const eDVBChannelQuery &query)
145 {
146         int res = 0;
147         switch (query.m_type)
148         {
149         case eDVBChannelQuery::tName:
150                 res = m_service_name_sort.find(query.m_string) != std::string::npos;
151                 break;
152         case eDVBChannelQuery::tProvider:
153                 res = m_provider_name.find(query.m_string) != std::string::npos;
154                 break;
155         case eDVBChannelQuery::tType:
156                 res = ref.getServiceType() == query.m_int;
157                 break;
158         case eDVBChannelQuery::tBouquet:
159                 res = 0;
160                 break;
161         case eDVBChannelQuery::tSatellitePosition:
162                 res = (ref.getDVBNamespace().get() >> 16) == query.m_int;
163                 break;
164         case eDVBChannelQuery::tChannelID:
165         {
166                 eDVBChannelID chid;
167                 ref.getChannelID(chid);
168                 res = chid == query.m_channelid;
169                 break;
170         }
171         case eDVBChannelQuery::tAND:
172                 res = checkFilter(ref, *query.m_p1) && checkFilter(ref, *query.m_p2);
173                 break;
174         case eDVBChannelQuery::tOR:
175                 res = checkFilter(ref, *query.m_p1) || checkFilter(ref, *query.m_p2);
176                 break;
177         }
178
179         if (query.m_inverse)
180                 return !res;
181         else
182                 return res;
183 }
184
185 int eDVBService::getCachePID(cacheID id)
186 {
187         std::map<int, int>::iterator it = m_cache.find(id);
188         if ( it != m_cache.end() )
189                 return it->second;
190         return -1;
191 }
192
193 void eDVBService::setCachePID(cacheID id, int pid)
194 {
195         m_cache[id] = pid;
196 }
197
198 DEFINE_REF(eDVBDB);
199
200         /* THIS CODE IS BAD. it should be replaced by somethine better. */
201 void eDVBDB::load()
202 {
203         eDebug("---- opening lame channel db");
204         FILE *f=fopen("lamedb", "rt");
205         if (!f)
206                 return;
207         char line[256];
208         if ((!fgets(line, 256, f)) || strncmp(line, "eDVB services", 13))
209         {
210                 eDebug("not a servicefile");
211                 fclose(f);
212                 return;
213         }
214         eDebug("reading services");
215         if ((!fgets(line, 256, f)) || strcmp(line, "transponders\n"))
216         {
217                 eDebug("services invalid, no transponders");
218                 fclose(f);
219                 return;
220         }
221
222         // clear all transponders
223
224         while (!feof(f))
225         {
226                 if (!fgets(line, 256, f))
227                         break;
228                 if (!strcmp(line, "end\n"))
229                         break;
230                 int dvb_namespace=-1, transport_stream_id=-1, original_network_id=-1;
231                 sscanf(line, "%x:%x:%x", &dvb_namespace, &transport_stream_id, &original_network_id);
232                 if (original_network_id == -1)
233                         continue;
234                 eDVBChannelID channelid = eDVBChannelID(
235                         eDVBNamespace(dvb_namespace),
236                         eTransportStreamID(transport_stream_id),
237                         eOriginalNetworkID(original_network_id));
238
239                 ePtr<eDVBFrontendParameters> feparm = new eDVBFrontendParameters;
240                 while (!feof(f))
241                 {
242                         fgets(line, 256, f);
243                         if (!strcmp(line, "/\n"))
244                                 break;
245                         if (line[1]=='s')
246                         {
247                                 eDVBFrontendParametersSatellite sat;
248                                 int frequency, symbol_rate, polarisation, fec, orbital_position, inversion;
249                                 sscanf(line+2, "%d:%d:%d:%d:%d:%d", &frequency, &symbol_rate, &polarisation, &fec, &orbital_position, &inversion);
250                                 sat.frequency = frequency;
251                                 sat.symbol_rate = symbol_rate;
252                                 sat.polarisation = polarisation;
253                                 sat.fec = fec;
254                                 sat.orbital_position = orbital_position;
255                                 sat.inversion = inversion;
256                                 feparm->setDVBS(sat);
257                         } else if (line[1]=='t')
258                         {
259                                 eDVBFrontendParametersTerrestrial ter;
260                                 int frequency, bandwidth, code_rate_HP, code_rate_LP, modulation, transmission_mode, guard_interval, hierarchy, inversion;
261                                 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);
262                                 ter.frequency = frequency;
263                                 ter.bandwidth = bandwidth;
264                                 ter.code_rate_HP = code_rate_HP;
265                                 ter.code_rate_LP = code_rate_LP;
266                                 ter.modulation = modulation;
267                                 ter.transmission_mode = transmission_mode;
268                                 ter.guard_interval = guard_interval;
269                                 ter.hierarchy = hierarchy;
270                                 ter.inversion = inversion;
271                                 
272                                 feparm->setDVBT(ter);
273                         } else if (line[1]=='c')
274                         {
275                                 int frequency, symbol_rate, inversion=0, modulation=3;
276                                 sscanf(line+2, "%d:%d:%d:%d", &frequency, &symbol_rate, &inversion, &modulation);
277 //                              t.setCable(frequency, symbol_rate, inversion, modulation);
278                         }
279                 }
280                 addChannelToList(channelid, feparm);
281         }
282
283         if ((!fgets(line, 256, f)) || strcmp(line, "services\n"))
284         {
285                 eDebug("services invalid, no services");
286                 return;
287         }
288
289         // clear all services
290
291         int count=0;
292
293         while (!feof(f))
294         {
295                 if (!fgets(line, 256, f))
296                         break;
297                 if (!strcmp(line, "end\n"))
298                         break;
299
300                 int service_id=-1, dvb_namespace, transport_stream_id=-1, original_network_id=-1, service_type=-1, service_number=-1;
301                 sscanf(line, "%x:%x:%x:%x:%d:%d", &service_id, &dvb_namespace, &transport_stream_id, &original_network_id, &service_type, &service_number);
302                 if (service_number == -1)
303                         continue;
304                 ePtr<eDVBService> s = new eDVBService;
305                 eServiceReferenceDVB ref =
306                                                 eServiceReferenceDVB(
307                                                 eDVBNamespace(dvb_namespace),
308                                                 eTransportStreamID(transport_stream_id),
309                                                 eOriginalNetworkID(original_network_id),
310                                                 eServiceID(service_id),
311                                                 service_type);
312                 count++;
313                 fgets(line, 256, f);
314                 if (strlen(line))
315                         line[strlen(line)-1]=0;
316
317                 s->m_service_name = line;
318                 s->genSortName();
319                  
320                 fgets(line, 256, f);
321                 if (strlen(line))
322                         line[strlen(line)-1]=0;
323
324                 std::string str=line;
325
326                 if (str[1]!=':')        // old ... (only service_provider)
327                 {
328                         s->m_provider_name=line;
329                 } else
330                         while ((!str.empty()) && str[1]==':') // new: p:, f:, c:%02d...
331                         {
332                                 unsigned int c=str.find(',');
333                                 char p=str[0];
334                                 std::string v;
335                                 if (c == std::string::npos)
336                                 {
337                                         v=str.substr(2);
338                                         str="";
339                                 } else
340                                 {
341                                         v=str.substr(2, c-2);
342                                         str=str.substr(c+1);
343                                 }
344 //                              eDebug("%c ... %s", p, v.c_str());
345                                 if (p == 'p')
346                                         s->m_provider_name=v;
347                                 else if (p == 'f')
348                                 {
349                                         sscanf(v.c_str(), "%x", &s->m_flags);
350                                 } else if (p == 'c')
351                                 {
352                                         int cid, val;
353                                         sscanf(v.c_str(), "%02d%04x", &cid, &val);
354                                         s->m_cache[cid]=val;
355                                 } else if (p == 'C')
356                                 {
357                                         int val;
358                                         sscanf(v.c_str(), "%04x", &val);
359                                         s->m_ca.insert(val);
360                                 }
361                         }
362                 addService(ref, s);
363         }
364
365         eDebug("loaded %d services", count);
366
367         fclose(f);
368 }
369
370 void eDVBDB::save()
371 {
372         eDebug("---- saving lame channel db");
373         FILE *f=fopen("lamedb", "wt");
374         int channels=0, services=0;
375         if (!f)
376                 eFatal("couldn't save lame channel db!");
377         fprintf(f, "eDVB services /3/\n");
378         fprintf(f, "transponders\n");
379         for (std::map<eDVBChannelID, channel>::const_iterator i(m_channels.begin());
380                         i != m_channels.end(); ++i)
381         {
382                 const eDVBChannelID &chid = i->first;
383                 const channel &ch = i->second;
384
385                 fprintf(f, "%08x:%04x:%04x\n", chid.dvbnamespace.get(),
386                                 chid.transport_stream_id.get(), chid.original_network_id.get());
387                 eDVBFrontendParametersSatellite sat;
388                 eDVBFrontendParametersTerrestrial ter;
389                 if (!ch.m_frontendParameters->getDVBS(sat))
390                 {
391                         fprintf(f, "\ts %d:%d:%d:%d:%d:%d\n",
392                                 sat.frequency, sat.symbol_rate,
393                                 sat.polarisation, sat.fec, sat.orbital_position,
394                                 sat.inversion);
395                 }
396                 if (!ch.m_frontendParameters->getDVBT(ter))
397                 {
398                         fprintf(f, "\tt %d:%d:%d:%d:%d:%d:%d:%d:%d\n",
399                                 ter.frequency, ter.bandwidth, ter.code_rate_HP,
400                                 ter.code_rate_LP, ter.modulation, ter.transmission_mode,
401                                 ter.guard_interval, ter.hierarchy, ter.inversion);
402                 }
403                 fprintf(f, "/\n");
404                 channels++;
405         }
406         fprintf(f, "end\nservices\n");
407
408         for (std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator i(m_services.begin());
409                 i != m_services.end(); ++i)
410         {
411                 const eServiceReferenceDVB &s = i->first;
412                 fprintf(f, "%04x:%08x:%04x:%04x:%d:%d\n",
413                                 s.getServiceID().get(), s.getDVBNamespace().get(),
414                                 s.getTransportStreamID().get(),s.getOriginalNetworkID().get(),
415                                 s.getServiceType(),
416                                 0);
417
418                 fprintf(f, "%s\n", i->second->m_service_name.c_str());
419                 fprintf(f, "p:%s", i->second->m_provider_name.c_str());
420
421                 // write cached pids
422                 for (std::map<int,int>::const_iterator ca(i->second->m_cache.begin());
423                         ca != i->second->m_cache.end(); ++ca)
424                         fprintf(f, ",c:%02d%04x", ca->first, ca->second);
425
426                 // write cached ca pids
427                 for (std::set<int>::const_iterator ca(i->second->m_ca.begin());
428                         ca != i->second->m_ca.end(); ++ca)
429                         fprintf(f, ",C:%04x", *ca);
430
431                 fprintf(f, "\n");
432                 services++;
433         }
434         fprintf(f, "end\nHave a lot of bugs!\n");
435         eDebug("saved %d channels and %d services!", channels, services);
436         fclose(f);
437 }
438
439 void eDVBDB::loadBouquet(const char *path)
440 {
441         std::string bouquet_name = path;
442         if (!bouquet_name.length())
443         {
444                 eDebug("Bouquet load failed.. no path given..");
445                 return;
446         }
447         unsigned int pos = bouquet_name.rfind('/');
448         if ( pos != std::string::npos )
449                 bouquet_name.erase(0, pos+1);
450         if (bouquet_name.empty())
451         {
452                 eDebug("Bouquet load failed.. no filename given..");
453                 return;
454         }
455         eBouquet &bouquet = m_bouquets[bouquet_name];
456         bouquet.m_path = path;
457         std::list<eServiceReference> &list = bouquet.m_services;
458         list.clear();
459
460         eDebug("loading bouquet... %s", path);
461         FILE *fp=fopen(path, "rt");
462         int entries=0;
463         if (!fp)
464         {
465                 eDebug("failed to open.");
466                 if ( strstr(path, "bouquets.tv") )
467                 {
468                         eDebug("recreate bouquets.tv");
469                         bouquet.m_bouquet_name="Bouquets (TV)";
470                         bouquet.flushChanges();
471                 }
472                 else if ( strstr(path, "bouquets.radio") )
473                 {
474                         eDebug("recreate bouquets.radio");
475                         bouquet.m_bouquet_name="Bouquets (Radio)";
476                         bouquet.flushChanges();
477                 }
478                 return;
479         }
480         char line[256];
481         bool read_descr=false;
482         eServiceReference *e = NULL;
483         while (1)
484         {
485                 if (!fgets(line, 256, fp))
486                         break;
487                 line[strlen(line)-1]=0;
488                 if (strlen(line) && line[strlen(line)-1]=='\r')
489                         line[strlen(line)-1]=0;
490                 if (!line[0])
491                         break;
492                 if (line[0]=='#')
493                 {
494                         if (!strncmp(line, "#SERVICE ", 9) || !strncmp(line, "#SERVICE: ", 10))
495                         {
496                                 int offs = line[8] == ':' ? 10 : 9;
497                                 eServiceReference tmp(line+offs);
498                                 if (tmp.type != eServiceReference::idDVB)
499                                 {
500                                         eDebug("only DVB Bouquets supported");
501                                         continue;
502                                 }
503                                 if ( (tmp.flags&eServiceReference::flagDirectory) == eServiceReference::flagDirectory )
504                                 {
505                                         std::string str = tmp.path;
506                                         unsigned int pos = str.rfind('/');
507                                         if ( pos != std::string::npos )
508                                                 str.erase(0, pos+1);
509                                         if (str.empty())
510                                         {
511                                                 eDebug("Bouquet load failed.. no filename given..");
512                                                 continue;
513                                         }
514                                         loadBouquet(tmp.path.c_str());
515                                         char buf[256];
516                                         snprintf(buf, 256, "(type == %d) FROM BOUQUET \"%s\" ORDER BY bouquet", tmp.data[0], str.c_str());
517                                         tmp.path = buf;
518                                 }
519                                 list.push_back(tmp);
520                                 e = &list.back();
521                                 read_descr=true;
522                                 ++entries;
523                         }
524                         else if (read_descr && !strncmp(line, "#DESCRIPTION ", 13))
525                         {
526                                 e->name = line+13;
527                                 read_descr=false;
528                         }
529                         else if (!strncmp(line, "#NAME ", 6))
530                                 bouquet.m_bouquet_name=line+6;
531                         continue;
532                 }
533         }
534         fclose(fp);
535         eDebug("%d entries in Bouquet %s", entries, bouquet_name.c_str());
536 }
537
538 void eDVBDB::loadBouquets()
539 {
540         loadBouquet("bouquets.tv");
541         loadBouquet("bouquets.radio");
542 // create default bouquets when missing
543         if ( m_bouquets.find("userbouquet.favourites.tv") == m_bouquets.end() )
544         {
545                 eBouquet &b = m_bouquets["userbouquet.favourites.tv"];
546                 b.m_path = "userbouquet.favourites.tv";
547                 b.m_bouquet_name = "Favourites (TV)";
548                 b.flushChanges();
549                 eServiceReference ref;
550                 memset(ref.data, 0, sizeof(ref.data));
551                 ref.type=1;
552                 ref.flags=7;
553                 ref.data[0]=1;
554                 ref.path="(type == 1) FROM BOUQUET \"userbouquet.favourites.tv\" ORDER BY bouquet";
555                 eBouquet &parent = m_bouquets["bouquets.tv"];
556                 parent.m_services.push_back(ref);
557                 parent.flushChanges();
558         }
559         if ( m_bouquets.find("userbouquet.favourites.radio") == m_bouquets.end() )
560         {
561                 eBouquet &b = m_bouquets["userbouquet.favourites.radio"];
562                 b.m_path = "userbouquet.favourites.radio";
563                 b.m_bouquet_name = "Favourites (Radio)";
564                 b.flushChanges();
565                 eServiceReference ref;
566                 memset(ref.data, 0, sizeof(ref.data));
567                 ref.type=1;
568                 ref.flags=7;
569                 ref.data[0]=1;
570                 ref.path="(type == 2) FROM BOUQUET \"userbouquet.favourites.radio\" ORDER BY bouquet";
571                 eBouquet &parent = m_bouquets["bouquets.radio"];
572                 parent.m_services.push_back(ref);
573                 parent.flushChanges();
574         }
575 }
576
577 eDVBDB::eDVBDB()
578 {
579         load();
580         loadBouquets();
581 }
582
583 eDVBDB::~eDVBDB()
584 {
585 //      save();
586 }
587
588 RESULT eDVBDB::addChannelToList(const eDVBChannelID &id, iDVBFrontendParameters *feparm)
589 {
590         channel ch;
591         assert(feparm);
592         ch.m_frontendParameters = feparm;
593         m_channels.insert(std::pair<eDVBChannelID, channel>(id, ch));
594         return 0;
595 }
596
597 RESULT eDVBDB::removeChannel(const eDVBChannelID &id)
598 {
599         m_channels.erase(id);
600         return 0;
601 }
602
603 RESULT eDVBDB::getChannelFrontendData(const eDVBChannelID &id, ePtr<iDVBFrontendParameters> &parm)
604 {
605         std::map<eDVBChannelID, channel>::iterator i = m_channels.find(id);
606         if (i == m_channels.end())
607         {
608                 parm = 0;
609                 return -ENOENT;
610         }
611         parm = i->second.m_frontendParameters;
612         return 0;
613 }
614
615 RESULT eDVBDB::addService(const eServiceReferenceDVB &serviceref, eDVBService *service)
616 {
617         m_services.insert(std::pair<eServiceReferenceDVB, ePtr<eDVBService> >(serviceref, service));
618         return 0;
619 }
620
621 RESULT eDVBDB::getService(const eServiceReferenceDVB &reference, ePtr<eDVBService> &service)
622 {
623         std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator i;
624         i = m_services.find(reference);
625         if (i == m_services.end())
626         {
627                 service = 0;
628                 return -ENOENT;
629         }
630         service = i->second;
631         return 0;
632 }
633
634 RESULT eDVBDB::flush()
635 {
636         save();
637         return 0;
638 }
639
640 RESULT eDVBDB::getBouquet(const eServiceReference &ref, eBouquet* &bouquet)
641 {
642         std::string str = ref.path;
643         if (str.empty())
644         {
645                 eDebug("getBouquet failed.. no path given!");
646                 return -1;
647         }
648         unsigned int pos = str.find("FROM BOUQUET \"");
649         if ( pos != std::string::npos )
650         {
651                 str.erase(0, pos+14);
652                 pos = str.find('"');
653                 if ( pos != std::string::npos )
654                         str.erase(pos);
655                 else
656                         str.clear();
657         }
658         if (str.empty())
659         {
660                 eDebug("getBouquet failed.. couldn't parse bouquet name");
661                 return -1;
662         }
663         std::map<std::string, eBouquet>::iterator i =
664                 m_bouquets.find(str);
665         if (i == m_bouquets.end())
666         {
667                 bouquet = 0;
668                 return -ENOENT;
669         }
670         bouquet = &i->second;
671         return 0;
672 }
673
674 RESULT eDVBDB::startQuery(ePtr<iDVBChannelListQuery> &query, eDVBChannelQuery *q, const eServiceReference &source)
675 {
676         if ( q && q->m_bouquet_name.length() )
677                 query = new eDVBDBBouquetQuery(this, source, q);
678         else
679                 query = new eDVBDBQuery(this, source, q);
680         return 0;
681 }
682
683 DEFINE_REF(eDVBDBQueryBase);
684
685 eDVBDBQueryBase::eDVBDBQueryBase(eDVBDB *db, const eServiceReference &source, eDVBChannelQuery *query)
686         :m_db(db), m_query(query), m_source(source)
687 {
688 }
689
690 int eDVBDBQueryBase::compareLessEqual(const eServiceReferenceDVB &a, const eServiceReferenceDVB &b)
691 {
692         ePtr<eDVBService> a_service, b_service;
693         
694         int sortmode = m_query ? m_query->m_sort : eDVBChannelQuery::tName;
695         
696         if ((sortmode == eDVBChannelQuery::tName) || (sortmode == eDVBChannelQuery::tProvider))
697         {
698                 if (m_db->getService(a, a_service))
699                         return 1;
700                 if (m_db->getService(b, b_service))
701                         return 1;
702         }
703         
704         switch (sortmode)
705         {
706         case eDVBChannelQuery::tName:
707                 return a_service->m_service_name_sort < b_service->m_service_name_sort;
708         case eDVBChannelQuery::tProvider:
709                 return a_service->m_provider_name < b_service->m_provider_name;
710         case eDVBChannelQuery::tType:
711                 return a.getServiceType() < b.getServiceType();
712         case eDVBChannelQuery::tBouquet:
713                 return 0;
714         case eDVBChannelQuery::tSatellitePosition:
715                 return (a.getDVBNamespace().get() >> 16) < (b.getDVBNamespace().get() >> 16);
716         default:
717                 return 1;
718         }
719         return 0;
720 }
721
722 eDVBDBQuery::eDVBDBQuery(eDVBDB *db, const eServiceReference &source, eDVBChannelQuery *query)
723         :eDVBDBQueryBase(db, source, query)
724 {
725         m_cursor = m_db->m_services.begin();
726 }
727
728 RESULT eDVBDBQuery::getNextResult(eServiceReferenceDVB &ref)
729 {
730         while (m_cursor != m_db->m_services.end())
731         {
732                 ref = m_cursor->first;
733
734                 int res = (!m_query) || m_cursor->second->checkFilter(ref, *m_query);
735
736                 ++m_cursor;
737
738                 if (res)
739                         return 0;
740         }
741
742         ref.type = eServiceReference::idInvalid;
743
744         return 1;
745 }
746
747 eDVBDBBouquetQuery::eDVBDBBouquetQuery(eDVBDB *db, const eServiceReference &source, eDVBChannelQuery *query)
748         :eDVBDBQueryBase(db, source, query), m_cursor(db->m_bouquets[query->m_bouquet_name].m_services.begin())
749 {
750 }
751
752 RESULT eDVBDBBouquetQuery::getNextResult(eServiceReferenceDVB &ref)
753 {
754         eBouquet &bouquet = m_db->m_bouquets[m_query->m_bouquet_name];
755         std::list<eServiceReference> &list = bouquet.m_services;
756         while (m_cursor != list.end())
757         {
758                 ref = *((eServiceReferenceDVB*)&(*m_cursor));
759
760                 std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator it =
761                         m_db->m_services.find(ref);
762
763                 int res = (!m_query) || it == m_db->m_services.end() || it->second->checkFilter(ref, *m_query);
764
765                 ++m_cursor;
766
767                 if (res)
768                         return 0;
769         }
770
771         ref.type = eServiceReference::idInvalid;
772
773         return 1;
774 }
775
776 /* (<name|provider|type|bouquet|satpos|chid> <==|...> <"string"|int>)[||,&& (..)] */
777
778 static int decodeType(const std::string &type)
779 {
780         if (type == "name")
781                 return eDVBChannelQuery::tName;
782         else if (type == "provider")
783                 return eDVBChannelQuery::tProvider;
784         else if (type == "type")
785                 return eDVBChannelQuery::tType;
786         else if (type == "bouquet")
787                 return eDVBChannelQuery::tBouquet;
788         else if (type == "satellitePosition")
789                 return eDVBChannelQuery::tSatellitePosition;
790         else if (type == "channelID")
791                 return eDVBChannelQuery::tChannelID;
792         else
793                 return -1;
794 }
795
796         /* never, NEVER write a parser in C++! */
797 RESULT parseExpression(ePtr<eDVBChannelQuery> &res, std::list<std::string>::const_iterator begin, std::list<std::string>::const_iterator end)
798 {
799         std::list<std::string>::const_iterator end_of_exp;
800         if (*begin == "(")
801         {
802                 end_of_exp = begin;
803                 while (end_of_exp != end)
804                         if (*end_of_exp == ")")
805                                 break;
806                         else
807                                 ++end_of_exp;
808         
809                 if (end_of_exp == end)
810                 {
811                         eDebug("expression parse: end of expression while searching for closing brace");
812                         return -1;
813                 }
814                 
815                 ++begin;
816                 // begin..end_of_exp
817                 int r = parseExpression(res, begin, end_of_exp);
818                 if (r)
819                         return r;
820                 ++end_of_exp;
821                 
822                         /* we had only one sub expression */
823                 if (end_of_exp == end)
824                 {
825 //                      eDebug("only one sub expression");
826                         return 0;
827                 }
828                 
829                         /* otherwise we have an operator here.. */
830                 
831                 ePtr<eDVBChannelQuery> r2 = res;
832                 res = new eDVBChannelQuery();
833                 res->m_sort = 0;
834                 res->m_p1 = r2;
835                 res->m_inverse = 0;
836                 r2 = 0;
837                 
838                 if (*end_of_exp == "||")
839                         res->m_type = eDVBChannelQuery::tOR;
840                 else if (*end_of_exp == "&&")
841                         res->m_type = eDVBChannelQuery::tAND;
842                 else
843                 {
844                         eDebug("found operator %s, but only && and || are allowed!", end_of_exp->c_str());
845                         res = 0;
846                         return 1;
847                 }
848                 
849                 ++end_of_exp;
850                 
851                 return parseExpression(res->m_p2, end_of_exp, end);
852         }
853         
854         // "begin" <op> "end"
855         std::string type, op, val;
856         
857         res = new eDVBChannelQuery();
858         res->m_sort = 0;
859         
860         int cnt = 0;
861         while (begin != end)
862         {
863                 switch (cnt)
864                 {
865                 case 0:
866                         type = *begin;
867                         break;
868                 case 1:
869                         op = *begin;
870                         break;
871                 case 2:
872                         val = *begin;
873                         break;
874                 case 3:
875                         eDebug("malformed query: got '%s', but expected only <type> <op> <val>", begin->c_str());
876                         return 1;
877                 }
878                 ++begin;
879                 ++cnt;
880         }
881         
882         if (cnt != 3)
883         {
884                 eDebug("malformed query: missing stuff");
885                 res = 0;
886                 return 1;
887         }
888         
889         res->m_type = decodeType(type);
890         
891         if (res->m_type == -1)
892         {
893                 eDebug("malformed query: invalid type %s", type.c_str());
894                 res = 0;
895                 return 1;
896         }
897         
898         if (op == "==")
899                 res->m_inverse = 0;
900         else if (op == "!=")
901                 res->m_inverse = 1;
902         else
903         {
904                 eDebug("invalid operator %s", op.c_str());
905                 res = 0;
906                 return 1;
907         }
908         
909         res->m_string = val;
910         res->m_int = atoi(val.c_str());
911 //      res->m_channelid = eDVBChannelID(val);
912         
913         return 0;
914 }
915
916 RESULT eDVBChannelQuery::compile(ePtr<eDVBChannelQuery> &res, std::string query)
917 {
918         std::list<std::string> tokens;
919         
920         std::string current_token;
921         std::string bouquet_name;
922
923 //      eDebug("splitting %s....", query.c_str());
924         unsigned int i = 0;
925         const char *splitchars="()";
926         int quotemode = 0, lastsplit = 0, lastalnum = 0;
927         while (i <= query.size())
928         {
929                 int c = (i < query.size()) ? query[i] : ' ';
930                 ++i;
931                 
932                 int issplit = !!strchr(splitchars, c);
933                 int isaln = isalnum(c);
934                 int iswhite = c == ' ';
935                 int isquot = c == '\"';
936                 
937                 if (quotemode)
938                 {
939                         iswhite = issplit = 0;
940                         isaln = lastalnum;
941                 }
942                 
943                 if (issplit || iswhite || isquot || lastsplit || (lastalnum != isaln))
944                 {
945                         if (current_token.size())
946                                 tokens.push_back(current_token);
947                         current_token.clear();
948                 }
949                 
950                 if (!(iswhite || isquot))
951                         current_token += c;
952                 
953                 if (isquot)
954                         quotemode = !quotemode;
955                 lastsplit = issplit;
956                 lastalnum = isaln;
957         }
958         
959 //      for (std::list<std::string>::const_iterator a(tokens.begin()); a != tokens.end(); ++a)
960 //      {
961 //              printf("%s\n", a->c_str());
962 //      }
963
964         int sort = eDVBChannelQuery::tName;
965                 /* check for "ORDER BY ..." */
966
967         while (tokens.size() > 2)
968         {
969                 std::list<std::string>::iterator it = tokens.end();
970                 --it; --it; --it;
971                 if (*it == "ORDER")
972                 {
973                         ++it;
974                         if (*it == "BY")
975                         {
976                                 ++it;
977                                 sort = decodeType(*it);
978                                 tokens.pop_back(); // ...
979                                 tokens.pop_back(); // BY
980                                 tokens.pop_back(); // ORDER
981                         } else
982                                 sort = -1;
983                 }
984                 else if (*it == "FROM")
985                 {
986                         ++it;
987                         if (*it == "BOUQUET")
988                         {
989                                 ++it;
990                                 bouquet_name = *it;
991                                 tokens.pop_back(); // ...
992                                 tokens.pop_back(); // FROM
993                                 tokens.pop_back(); // BOUQUET
994                         }
995                 }
996                 else
997                         break;
998         }
999
1000         if (sort == -1)
1001         {
1002                 eWarning("ORDER BY .. string invalid.");
1003                 res = 0;
1004                 return -1;
1005         }
1006         
1007 //      eDebug("sort by %d", sort);
1008         
1009                 /* now we recursivly parse that. */
1010         int r = parseExpression(res, tokens.begin(), tokens.end());
1011         
1012         if (res)
1013         {
1014                 res->m_sort = sort;
1015                 res->m_bouquet_name = bouquet_name;
1016         }
1017
1018 //      eDebug("return: %d", r);
1019         return r;
1020 }
1021
1022 DEFINE_REF(eDVBChannelQuery);