code.vuplus.com Git - vuplus_dvbapp/blob - 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/dvb_si/sdt.h>
   6 #include <lib/dvb_si/descriptor_tag.h>
   7 #include <lib/dvb_si/service_descriptor.h>
   8 #include <lib/dvb_si/satellite_delivery_system_descriptor.h>
   9
  10 DEFINE_REF(eDVBService);
  11
  12 eDVBService::eDVBService()
  13 {
  14 }
  15
  16 eDVBService::~eDVBService()
  17 {
  18 }
  19
  20 eDVBService &eDVBService::operator=(const eDVBService &s)
  21 {
  22         m_service_name = s.m_service_name;
  23         m_provider_name = s.m_provider_name;
  24         m_flags = s.m_flags;
  25         m_ca = s.m_ca;
  26         m_cache = s.m_cache;
  27         return *this;
  28 }
  29
  30 RESULT eDVBService::getName(const eServiceReference &ref, std::string &name)
  31 {
  32         name = m_service_name;
  33 }
  34
  35 int eDVBService::checkFilter(const eServiceReferenceDVB &ref, const eDVBChannelQuery &query)
  36 {
  37         int res = 0;
  38         switch (query.m_type)
  39         {
  40         case eDVBChannelQuery::tName:
  41                 res = m_service_name.find(query.m_string) != std::string::npos;
  42                 break;
  43         case eDVBChannelQuery::tProvider:
  44                 res = m_provider_name.find(query.m_string) != std::string::npos;
  45                 break;
  46         case eDVBChannelQuery::tType:
  47                 res = ref.getServiceType() == query.m_int;
  48                 break;
  49         case eDVBChannelQuery::tBouquet:
  50                 res = 0;
  51                 break;
  52         case eDVBChannelQuery::tSatellitePosition:
  53                 res = (ref.getDVBNamespace().get() >> 16) == query.m_int;
  54                 break;
  55         case eDVBChannelQuery::tChannelID:
  56         {
  57                 eDVBChannelID chid;
  58                 ref.getChannelID(chid);
  59                 res = chid == query.m_channelid;
  60                 break;
  61         }
  62         case eDVBChannelQuery::tAND:
  63                 res = checkFilter(ref, *query.m_p1) && checkFilter(ref, *query.m_p2);
  64                 break;
  65         case eDVBChannelQuery::tOR:
  66                 res = checkFilter(ref, *query.m_p1) || checkFilter(ref, *query.m_p2);
  67                 break;
  68         }
  69
  70         if (query.m_inverse)
  71                 return !res;
  72         else
  73                 return res;
  74 }
  75
  76 DEFINE_REF(eDVBDB);
  77
  78 eDVBDB::eDVBDB()
  79 {
  80         eDebug("---- opening lame channel db");
  81         FILE *f=fopen("lamedb", "rt");
  82         if (!f)
  83                 return;
  84         char line[256];
  85         if ((!fgets(line, 256, f)) || strncmp(line, "eDVB services", 13))
  86         {
  87                 eDebug("not a servicefile");
  88                 fclose(f);
  89                 return;
  90         }
  91         eDebug("reading services");
  92         if ((!fgets(line, 256, f)) || strcmp(line, "transponders\n"))
  93         {
  94                 eDebug("services invalid, no transponders");
  95                 fclose(f);
  96                 return;
  97         }
  98
  99         // clear all transponders
 100
 101         while (!feof(f))
 102         {
 103                 if (!fgets(line, 256, f))
 104                         break;
 105                 if (!strcmp(line, "end\n"))
 106                         break;
 107                 int dvb_namespace=-1, transport_stream_id=-1, original_network_id=-1;
 108                 sscanf(line, "%x:%x:%x", &dvb_namespace, &transport_stream_id, &original_network_id);
 109                 if (original_network_id == -1)
 110                         continue;
 111                 eDVBChannelID channelid = eDVBChannelID(
 112                         eDVBNamespace(dvb_namespace),
 113                         eTransportStreamID(transport_stream_id),
 114                         eOriginalNetworkID(original_network_id));
 115
 116                 ePtr<eDVBFrontendParameters> feparm = new eDVBFrontendParameters;
 117                 while (!feof(f))
 118                 {
 119                         fgets(line, 256, f);
 120                         if (!strcmp(line, "/\n"))
 121                                 break;
 122                         if (line[1]=='s')
 123                         {
 124                                 eDVBFrontendParametersSatellite sat;
 125                                 int frequency, symbol_rate, polarisation, fec, orbital_position, inversion;
 126                                 sscanf(line+2, "%d:%d:%d:%d:%d:%d", &frequency, &symbol_rate, &polarisation, &fec, &inversion, &orbital_position);
 127                                 sat.frequency = frequency;
 128                                 sat.symbol_rate = symbol_rate;
 129                                 sat.polarisation = polarisation;
 130                                 sat.fec = fec;
 131                                 sat.orbital_position = orbital_position;
 132                                 sat.inversion = inversion;
 133                                 // ...
 134 //                              t.setSatellite(frequency, symbol_rate, polarisation, fec, sat, inversion);
 135                                 feparm->setDVBS(sat);
 136                         }
 137                         if (line[1]=='c')
 138                         {
 139                                 int frequency, symbol_rate, inversion=0, modulation=3;
 140                                 sscanf(line+2, "%d:%d:%d:%d", &frequency, &symbol_rate, &inversion, &modulation);
 141 //                              t.setCable(frequency, symbol_rate, inversion, modulation);
 142                         }
 143                 }
 144                 addChannelToList(channelid, feparm);
 145         }
 146
 147         if ((!fgets(line, 256, f)) || strcmp(line, "services\n"))
 148         {
 149                 eDebug("services invalid, no services");
 150                 return;
 151         }
 152
 153         // clear all services
 154
 155         int count=0;
 156
 157         while (!feof(f))
 158         {
 159                 if (!fgets(line, 256, f))
 160                         break;
 161                 if (!strcmp(line, "end\n"))
 162                         break;
 163
 164                 int service_id=-1, dvb_namespace, transport_stream_id=-1, original_network_id=-1, service_type=-1, service_number=-1;
 165                 sscanf(line, "%x:%x:%x:%x:%d:%d", &service_id, &dvb_namespace, &transport_stream_id, &original_network_id, &service_type, &service_number);
 166                 if (service_number == -1)
 167                         continue;
 168                 ePtr<eDVBService> s = new eDVBService;
 169                 eServiceReferenceDVB ref =
 170                                                 eServiceReferenceDVB(
 171                                                 eDVBNamespace(dvb_namespace),
 172                                                 eTransportStreamID(transport_stream_id),
 173                                                 eOriginalNetworkID(original_network_id),
 174                                                 eServiceID(service_id),
 175                                                 service_type);
 176                 count++;
 177                 fgets(line, 256, f);
 178                 if (strlen(line))
 179                         line[strlen(line)-1]=0;
 180                 s->m_service_name=line;
 181                 fgets(line, 256, f);
 182                 if (strlen(line))
 183                         line[strlen(line)-1]=0;
 184
 185                 std::string str=line;
 186
 187                 if (str[1]!=':')        // old ... (only service_provider)
 188                 {
 189                         s->m_provider_name=line;
 190                 } else
 191                         while ((!str.empty()) && str[1]==':') // new: p:, f:, c:%02d...
 192                         {
 193                                 unsigned int c=str.find(',');
 194                                 char p=str[0];
 195                                 std::string v;
 196                                 if (c == std::string::npos)
 197                                 {
 198                                         v=str.substr(2);
 199                                         str="";
 200                                 } else
 201                                 {
 202                                         v=str.substr(2, c-2);
 203                                         str=str.substr(c+1);
 204                                 }
 205 //                              eDebug("%c ... %s", p, v.c_str());
 206                                 if (p == 'p')
 207                                         s->m_provider_name=v;
 208                                 else if (p == 'f')
 209                                 {
 210                                         sscanf(v.c_str(), "%x", &s->m_flags);
 211                                 } else if (p == 'c')
 212                                 {
 213                                         int cid, val;
 214                                         sscanf(v.c_str(), "%02d%04x", &cid, &val);
 215                                         s->m_cache[cid]=val;
 216                                 } else if (p == 'C')
 217                                 {
 218                                         int val;
 219                                         sscanf(v.c_str(), "%04x", &val);
 220                                         s->m_ca.insert(val);
 221                                 }
 222                         }
 223                 addService(ref, s);
 224         }
 225
 226         eDebug("loaded %d services", count);
 227
 228         fclose(f);
 229
 230 }
 231
 232 eDVBDB::~eDVBDB()
 233 {
 234         eDebug("---- saving lame channel db");
 235         FILE *f=fopen("lamedb", "wt");
 236         int channels=0, services=0;
 237         if (!f)
 238                 eFatal("couldn't save lame channel db!");
 239         fprintf(f, "eDVB services /3/\n");
 240         fprintf(f, "transponders\n");
 241         for (std::map<eDVBChannelID, channel>::const_iterator i(m_channels.begin());
 242                         i != m_channels.end(); ++i)
 243         {
 244                 const eDVBChannelID &chid = i->first;
 245                 const channel &ch = i->second;
 246
 247                 fprintf(f, "%08x:%04x:%04x\n", chid.dvbnamespace.get(),
 248                                 chid.transport_stream_id.get(), chid.original_network_id.get());
 249                 eDVBFrontendParametersSatellite sat;
 250                 if (!ch.m_frontendParameters->getDVBS(sat))
 251                 {
 252                         fprintf(f, "\ts %d:%d:%d:%d:%d:%d\n",
 253                                 sat.frequency, sat.symbol_rate,
 254                                 sat.polarisation, sat.fec, sat.inversion,
 255                                 sat.orbital_position);
 256                 }
 257                 fprintf(f, "/\n");
 258                 channels++;
 259         }
 260         fprintf(f, "end\nservices\n");
 261
 262         for (std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator i(m_services.begin());
 263                 i != m_services.end(); ++i)
 264         {
 265                 const eServiceReferenceDVB &s = i->first;
 266                 fprintf(f, "%04x:%08x:%04x:%04x:%d:%d\n",
 267                                 s.getServiceID().get(), s.getDVBNamespace().get(),
 268                                 s.getTransportStreamID().get(),s.getOriginalNetworkID().get(),
 269                                 s.getServiceType(),
 270                                 0);
 271
 272                 fprintf(f, "%s\n", i->second->m_service_name.c_str());
 273                 fprintf(f, "p:%s", i->second->m_provider_name.c_str());
 274                 for (std::set<int>::const_iterator ca(i->second->m_ca.begin());
 275                         ca != i->second->m_ca.end(); ++ca)
 276                         fprintf(f, ",C:%04x", *ca);
 277                 fprintf(f, "\n");
 278                 services++;
 279         }
 280         fprintf(f, "end\nHave a lot of bugs!\n");
 281         eDebug("saved %d channels and %d services!", channels, services);
 282         fclose(f);
 283 }
 284
 285 RESULT eDVBDB::addChannelToList(const eDVBChannelID &id, iDVBFrontendParameters *feparm)
 286 {
 287         channel ch;
 288         assert(feparm);
 289         ch.m_frontendParameters = feparm;
 290         m_channels.insert(std::pair<eDVBChannelID, channel>(id, ch));
 291         return 0;
 292 }
 293
 294 RESULT eDVBDB::removeChannel(const eDVBChannelID &id)
 295 {
 296         m_channels.erase(id);
 297         return 0;
 298 }
 299
 300 RESULT eDVBDB::getChannelFrontendData(const eDVBChannelID &id, ePtr<iDVBFrontendParameters> &parm)
 301 {
 302         std::map<eDVBChannelID, channel>::iterator i = m_channels.find(id);
 303         if (i == m_channels.end())
 304         {
 305                 parm = 0;
 306                 return -ENOENT;
 307         }
 308         parm = i->second.m_frontendParameters;
 309         return 0;
 310 }
 311
 312 RESULT eDVBDB::addService(const eServiceReferenceDVB &serviceref, eDVBService *service)
 313 {
 314         m_services.insert(std::pair<eServiceReferenceDVB, ePtr<eDVBService> >(serviceref, service));
 315         return 0;
 316 }
 317
 318 RESULT eDVBDB::getService(const eServiceReferenceDVB &reference, ePtr<eDVBService> &service)
 319 {
 320         std::map<eServiceReferenceDVB, ePtr<eDVBService> >::iterator i;
 321         i = m_services.find(reference);
 322         if (i == m_services.end())
 323         {
 324                 service = 0;
 325                 return -ENOENT;
 326         }
 327         service = i->second;
 328         return 0;
 329 }
 330
 331 RESULT eDVBDB::startQuery(ePtr<iDVBChannelListQuery> &query, eDVBChannelQuery *q)
 332 {
 333         query = new eDVBDBQuery(this, eServiceReference(), q);
 334         return 0;
 335 }
 336
 337 DEFINE_REF(eDVBDBQuery);
 338
 339 eDVBDBQuery::eDVBDBQuery(eDVBDB *db, const eServiceReference &source, eDVBChannelQuery *query): m_db(db), m_query(query)
 340 {
 341                 // TODO: use SOURCE ...
 342         m_cursor = m_db->m_services.begin();
 343 }
 344
 345 RESULT eDVBDBQuery::getNextResult(eServiceReferenceDVB &ref)
 346 {
 347         while (m_cursor != m_db->m_services.end())
 348         {
 349                 ref = m_cursor->first;
 350
 351                 int res = (!m_query) || m_cursor->second->checkFilter(ref, *m_query);
 352
 353                 ++m_cursor;
 354
 355                 if (res)
 356                         return 0;
 357         }
 358         return 1;
 359 }
 360
 361 /* (<name|provider|type|bouquet|satpos|chid> <==|...> <"string"|int>)[AND (..)] */
 362
 363         /* never, NEVER write a parser in C++! */
 364 RESULT parseExpression(ePtr<eDVBChannelQuery> &res, std::list<std::string>::const_iterator begin, std::list<std::string>::const_iterator end)
 365 {
 366         std::list<std::string>::const_iterator end_of_exp;
 367         if (*begin == "(")
 368         {
 369                 end_of_exp = begin;
 370                 while (end_of_exp != end)
 371                         if (*end_of_exp == ")")
 372                                 break;
 373                         else
 374                                 ++end_of_exp;
 375
 376                 if (end_of_exp == end)
 377                 {
 378                         eDebug("expression parse: end of expression while searching for closing brace");
 379                         return -1;
 380                 }
 381
 382                 ++begin;
 383                 // begin..end_of_exp
 384                 int r = parseExpression(res, begin, end_of_exp);
 385                 if (r)
 386                         return r;
 387                 ++end_of_exp;
 388
 389                         /* we had only one sub expression */
 390                 if (end_of_exp == end)
 391                         return 1;
 392
 393                         /* otherwise we have an operator here.. */
 394
 395                 ePtr<eDVBChannelQuery> r2 = res;
 396                 res = new eDVBChannelQuery();
 397                 res->m_p1 = r2;
 398                 res->m_inverse = 0;
 399                 r2 = 0;
 400
 401                 if (*end_of_exp == "||")
 402                         res->m_type = eDVBChannelQuery::tOR;
 403                 else if (*end_of_exp == "&&")
 404                         res->m_type = eDVBChannelQuery::tAND;
 405                 else
 406                 {
 407                         eDebug("found operator %s, but only && and || are allowed!", end_of_exp->c_str());
 408                         res = 0;
 409                         return 1;
 410                 }
 411
 412                 ++end_of_exp;
 413
 414                 return parseExpression(res->m_p2, end_of_exp, end);
 415         }
 416
 417         // "begin" <op> "end"
 418         std::string type, op, val;
 419
 420         res = new eDVBChannelQuery();
 421
 422         int cnt = 0;
 423         while (begin != end)
 424         {
 425                 switch (cnt)
 426                 {
 427                 case 0:
 428                         type = *begin;
 429                         break;
 430                 case 1:
 431                         op = *begin;
 432                         break;
 433                 case 2:
 434                         val = *begin;
 435                         break;
 436                 case 3:
 437                         eDebug("malformed query: got '%s', but expected only <type> <op> <val>", begin->c_str());
 438                         return 1;
 439                 }
 440                 ++begin;
 441                 ++cnt;
 442         }
 443
 444         if (cnt != 3)
 445         {
 446                 eDebug("malformed query: missing stuff");
 447                 res = 0;
 448                 return 1;
 449         }
 450
 451         if (type == "name")
 452                 res->m_type = eDVBChannelQuery::tName;
 453         else if (type == "provider")
 454                 res->m_type = eDVBChannelQuery::tProvider;
 455         else if (type == "type")
 456                 res->m_type = eDVBChannelQuery::tType;
 457         else if (type == "bouquet")
 458                 res->m_type = eDVBChannelQuery::tBouquet;
 459         else if (type == "satellitePosition")
 460                 res->m_type = eDVBChannelQuery::tSatellitePosition;
 461         else if (type == "channelID")
 462                 res->m_type = eDVBChannelQuery::tChannelID;
 463         else
 464         {
 465                 eDebug("malformed query: invalid type %s", type.c_str());
 466                 res = 0;
 467                 return 1;
 468         }
 469
 470         eDebug("type is %d, nice!", res->m_type);
 471
 472         if (op == "==")
 473                 res->m_inverse = 0;
 474         else if (op == "!=")
 475                 res->m_inverse = 1;
 476         else
 477         {
 478                 eDebug("invalid operator %s", op.c_str());
 479                 res = 0;
 480                 return 1;
 481         }
 482
 483         res->m_string = val;
 484         res->m_int = atoi(val.c_str());
 485 //      res->m_channelid = eDVBChannelID(val);
 486
 487         return 0;
 488 }
 489
 490 RESULT eDVBChannelQuery::compile(ePtr<eDVBChannelQuery> &res, std::string query)
 491 {
 492         std::list<std::string> tokens;
 493
 494         std::string current_token;
 495
 496 //      eDebug("splitting %s....", query.c_str());
 497         unsigned int i = 0;
 498         const char *splitchars="()";
 499         int quotemode = 0, lastsplit = 0, lastalnum = 0;
 500         while (i <= query.size())
 501         {
 502                 int c = (i < query.size()) ? query[i] : ' ';
 503                 ++i;
 504
 505                 int issplit = !!strchr(splitchars, c);
 506                 int isaln = isalnum(c);
 507                 int iswhite = c == ' ';
 508                 int isquot = c == '\"';
 509
 510                 if (quotemode)
 511                 {
 512                         iswhite = issplit = 0;
 513                         isaln = lastalnum;
 514                 }
 515
 516                 if (issplit || iswhite || isquot || lastsplit || (lastalnum != isaln))
 517                 {
 518                         if (current_token.size())
 519                                 tokens.push_back(current_token);
 520                         current_token.clear();
 521                 }
 522
 523                 if (!(iswhite || isquot))
 524                         current_token += c;
 525
 526                 if (isquot)
 527                         quotemode = !quotemode;
 528                 lastsplit = issplit;
 529                 lastalnum = isaln;
 530         }
 531
 532 //      for (std::list<std::string>::const_iterator a(tokens.begin()); a != tokens.end(); ++a)
 533 //      {
 534 //              printf("%s\n", a->c_str());
 535 //      }
 536
 537                 /* now we recursivly parse that. */
 538         return  parseExpression(res, tokens.begin(), tokens.end());
 539 /*
 540         res = new eDVBChannelQuery();
 541         res->m_type = eDVBChannelQuery::tName;
 542         res->m_inverse = 0;
 543         res->m_string = query;
 544         return 0; */
 545 }
 546
 547 DEFINE_REF(eDVBChannelQuery);