lib/dvb/db.cpp

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