X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/4ea3e16caea18f079d66b387d000d9ba2db8fd1b..24ac423a92cf63ed6a8a09fa1ef48bef2c9382a6:/lib/dvb/epgcache.cpp diff --git a/lib/dvb/epgcache.cpp b/lib/dvb/epgcache.cpp index e8394c40..550561e2 100644 --- a/lib/dvb/epgcache.cpp +++ b/lib/dvb/epgcache.cpp @@ -12,15 +12,40 @@ #include // for statfs // #include #include +#include #include #include -#include +#include +#include int eventData::CacheSize=0; descriptorMap eventData::descriptors; __u8 eventData::data[4108]; extern const uint32_t crc32_table[256]; +const eServiceReference &handleGroup(const eServiceReference &ref) +{ + if (ref.flags & eServiceReference::isGroup) + { + ePtr res; + if (!eDVBResourceManager::getInstance(res)) + { + ePtr db; + if (!res->getChannelList(db)) + { + eBouquet *bouquet=0; + if (!db->getBouquet(ref, bouquet)) + { + std::list::iterator it(bouquet->m_services.begin()); + if (it != bouquet->m_services.end()) + return *it; + } + } + } + } + return ref; +} + eventData::eventData(const eit_event_struct* e, int size, int type) :ByteSize(size&0xFF), type(type&0xFF) { @@ -31,48 +56,67 @@ eventData::eventData(const eit_event_struct* e, int size, int type) __u32 *pdescr=descr; __u8 *data = (__u8*)e; - int ptr=10; - int descriptors_length = (data[ptr++]&0x0F) << 8; - descriptors_length |= data[ptr++]; - while ( descriptors_length > 0 ) + int ptr=12; + size -= 12; + + while(size > 1) { __u8 *descr = data+ptr; - int descr_len = descr[1]+2; - - __u32 crc = 0; - int cnt=0; - while(cnt++ < descr_len) - crc = (crc << 8) ^ crc32_table[((crc >> 24) ^ data[ptr++]) & 0xFF]; - - descriptorMap::iterator it = - descriptors.find(crc); - if ( it == descriptors.end() ) + int descr_len = descr[1]; + descr_len += 2; + if (size >= descr_len) { - CacheSize+=descr_len; - __u8 *d = new __u8[descr_len]; - memcpy(d, descr, descr_len); - descriptors[crc] = descriptorPair(1, d); + switch (descr[0]) + { + case EXTENDED_EVENT_DESCRIPTOR: + case SHORT_EVENT_DESCRIPTOR: + case LINKAGE_DESCRIPTOR: + case COMPONENT_DESCRIPTOR: + { + __u32 crc = 0; + int cnt=0; + while(cnt++ < descr_len) + crc = (crc << 8) ^ crc32_table[((crc >> 24) ^ data[ptr++]) & 0xFF]; + + descriptorMap::iterator it = + descriptors.find(crc); + if ( it == descriptors.end() ) + { + CacheSize+=descr_len; + __u8 *d = new __u8[descr_len]; + memcpy(d, descr, descr_len); + descriptors[crc] = descriptorPair(1, d); + } + else + ++it->second.first; + *pdescr++=crc; + break; + } + default: // do not cache all other descriptors + ptr += descr_len; + break; + } + size -= descr_len; } else - ++it->second.first; - - *pdescr++=crc; - descriptors_length -= descr_len; + break; } - ByteSize = 12+((pdescr-descr)*4); + ASSERT(pdescr <= &descr[65]); + ByteSize = 10+((pdescr-descr)*4); EITdata = new __u8[ByteSize]; CacheSize+=ByteSize; - memcpy(EITdata, (__u8*) e, 12); - memcpy(EITdata+12, descr, ByteSize-12); + memcpy(EITdata, (__u8*) e, 10); + memcpy(EITdata+10, descr, ByteSize-10); } const eit_event_struct* eventData::get() const { int pos = 12; - int tmp = ByteSize-12; - memcpy(data, EITdata, 12); - __u32 *p = (__u32*)(EITdata+12); - while(tmp>0) + int tmp = ByteSize-10; + memcpy(data, EITdata, 10); + int descriptors_length=0; + __u32 *p = (__u32*)(EITdata+10); + while(tmp>3) { descriptorMap::iterator it = descriptors.find(*p++); @@ -81,21 +125,26 @@ const eit_event_struct* eventData::get() const int b = it->second.second[1]+2; memcpy(data+pos, it->second.second, b ); pos += b; + descriptors_length += b; } + else + eFatal("LINE %d descriptor not found in descriptor cache %08x!!!!!!", __LINE__, *(p-1)); tmp-=4; } - - return (const eit_event_struct*)data; + ASSERT(pos <= 4108); + data[10] = (descriptors_length >> 8) & 0x0F; + data[11] = descriptors_length & 0xFF; + return (eit_event_struct*)data; } eventData::~eventData() { if ( ByteSize ) { - CacheSize-=ByteSize; - ByteSize-=12; - __u32 *d = (__u32*)(EITdata+12); - while(ByteSize) + CacheSize -= ByteSize; + __u32 *d = (__u32*)(EITdata+10); + ByteSize -= 10; + while(ByteSize>3) { descriptorMap::iterator it = descriptors.find(*d++); @@ -109,7 +158,9 @@ eventData::~eventData() descriptors.erase(it); // remove entry from descriptor map } } - ByteSize-=4; + else + eFatal("LINE %d descriptor not found in descriptor cache %08x!!!!!!", __LINE__, *(d-1)); + ByteSize -= 4; } delete [] EITdata; } @@ -175,7 +226,11 @@ eEPGCache::eEPGCache() if (!res_mgr) eDebug("[eEPGCache] no resource manager !!!!!!!"); else + { res_mgr->connectChannelAdded(slot(*this,&eEPGCache::DVBChannelAdded), m_chanAddedConn); + if (eDVBLocalTimeHandler::getInstance()->ready()) + timeUpdated(); + } instance=this; } @@ -465,10 +520,11 @@ void eEPGCache::sectionRead(const __u8 *data, int source, channel_data *channel) if ( tm_it_tmp->first == TM ) // just update eventdata { // exempt memory - delete ev_it->second; + eventData *tmp = ev_it->second; ev_it->second = tm_it_tmp->second = new eventData(eit_event, eit_event_size, source); FixOverlapping(servicemap, TM, duration, tm_it_tmp, service); + delete tmp; goto next; } else // event has new event begin time @@ -505,7 +561,6 @@ void eEPGCache::sectionRead(const __u8 *data, int source, channel_data *channel) prevEventIt=servicemap.first.end(); } } - evt = new eventData(eit_event, eit_event_size, source); #ifdef EPG_DEBUG bool consistencyCheck=true; @@ -687,7 +742,7 @@ void eEPGCache::cleanLoop() content_time_tables.find( DBIt->first ); if ( x != content_time_tables.end() ) { - timeMap &tmMap = eventDB[DBIt->first].second; + timeMap &tmMap = DBIt->second.second; for ( contentMap::iterator i = x->second.begin(); i != x->second.end(); ) { for ( contentTimeMap::iterator it(i->second.begin()); @@ -754,6 +809,7 @@ void eEPGCache::gotMessage( const Message &msg ) #ifdef ENABLE_PRIVATE_EPG case Message::got_private_pid: { + singleLock s(channel_map_lock); for (channelMapIterator it(m_knownChannels.begin()); it != m_knownChannels.end(); ++it) { eDVBChannel *channel = (eDVBChannel*) it->first; @@ -805,10 +861,10 @@ void eEPGCache::thread() void eEPGCache::load() { - singleLock s(cache_lock); FILE *f = fopen("/hdd/epg.dat", "r"); if (f) { + unlink("/hdd/epg.dat"); int size=0; int cnt=0; #if 0 @@ -840,8 +896,9 @@ void eEPGCache::load() } char text1[13]; fread( text1, 13, 1, f); - if ( !strncmp( text1, "ENIGMA_EPG_V5", 13) ) + if ( !strncmp( text1, "ENIGMA_EPG_V7", 13) ) { + singleLock s(cache_lock); fread( &size, sizeof(int), 1, f); while(size--) { @@ -943,7 +1000,7 @@ void eEPGCache::save() { unsigned int magic = 0x98765432; fwrite( &magic, sizeof(int), 1, f); - const char *text = "ENIGMA_EPG_V5"; + const char *text = "ENIGMA_EPG_V7"; fwrite( text, 13, 1, f ); int size = eventDB.size(); fwrite( &size, sizeof(int), 1, f ); @@ -1317,7 +1374,7 @@ RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, co // if t == -1 we search the current event... { singleLock s(cache_lock); - uniqueEPGKey key(service); + uniqueEPGKey key(handleGroup(service)); // check if EPG for this service is ready... eventCache::iterator It = eventDB.find( key ); @@ -1329,13 +1386,13 @@ RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, co It->second.second.upper_bound(t); // just > if ( i != It->second.second.end() ) { - if ( direction < 0 || (direction == 0 && i->second->getStartTime() > t) ) + if ( direction < 0 || (direction == 0 && i->first > t) ) { timeMap::iterator x = i; --x; if ( x != It->second.second.end() ) { - time_t start_time = x->second->getStartTime(); + time_t start_time = x->first; if (direction >= 0) { if (t < start_time) @@ -1393,7 +1450,7 @@ RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, eP RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, const eventData *&result ) { singleLock s(cache_lock); - uniqueEPGKey key( service ); + uniqueEPGKey key(handleGroup(service)); eventCache::iterator It = eventDB.find( key ); if ( It != eventDB.end() && !It->second.first.empty() ) // entrys cached? @@ -1450,22 +1507,23 @@ RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, RESULT eEPGCache::startTimeQuery(const eServiceReference &service, time_t begin, int minutes) { - eventCache::iterator It = eventDB.find( service ); + const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)handleGroup(service); + Lock(); + eventCache::iterator It = eventDB.find(ref); if ( It != eventDB.end() && It->second.second.size() ) { - m_timemap_end = minutes != -1 ? It->second.second.upper_bound(begin+minutes*60) : It->second.second.end(); if ( begin != -1 ) { m_timemap_cursor = It->second.second.lower_bound(begin); if ( m_timemap_cursor != It->second.second.end() ) { - if ( m_timemap_cursor->second->getStartTime() != begin ) + if ( m_timemap_cursor->first != begin ) { timeMap::iterator x = m_timemap_cursor; --x; if ( x != It->second.second.end() ) { - time_t start_time = x->second->getStartTime(); + time_t start_time = x->first; if ( begin > start_time && begin < (start_time+x->second->getDuration())) m_timemap_cursor = x; } @@ -1474,10 +1532,17 @@ RESULT eEPGCache::startTimeQuery(const eServiceReference &service, time_t begin, } else m_timemap_cursor = It->second.second.begin(); - const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service; + + if (minutes != -1 && m_timemap_cursor != It->second.second.end()) + m_timemap_end = It->second.second.upper_bound(m_timemap_cursor->first+minutes*60); + else + m_timemap_end = It->second.second.end(); + currentQueryTsidOnid = (ref.getTransportStreamID().get()<<16) | ref.getOriginalNetworkID().get(); + Unlock(); return 0; } + Unlock(); return -1; } @@ -1522,9 +1587,9 @@ RESULT eEPGCache::getNextTimeEntry(ePtr &result) return -1; } -void fillTuple(PyObject *tuple, char *argstring, int argcount, PyObject *service, ePtr &ptr, PyObject *nowTime, PyObject *service_name ) +void fillTuple(ePyObject tuple, char *argstring, int argcount, ePyObject service, ePtr &ptr, ePyObject nowTime, ePyObject service_name ) { - PyObject *tmp=NULL; + ePyObject tmp; int pos=0; while(pos < argcount) { @@ -1535,22 +1600,22 @@ void fillTuple(PyObject *tuple, char *argstring, int argcount, PyObject *service tmp = PyLong_FromLong(0); break; case 'I': // Event Id - tmp = ptr ? PyLong_FromLong(ptr->getEventId()) : NULL; + tmp = ptr ? PyLong_FromLong(ptr->getEventId()) : ePyObject(); break; case 'B': // Event Begin Time - tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : NULL; + tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : ePyObject(); break; case 'D': // Event Duration - tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : NULL; + tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : ePyObject(); break; case 'T': // Event Title - tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : NULL; + tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : ePyObject(); break; case 'S': // Event Short Description - tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : NULL; + tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : ePyObject(); break; case 'E': // Event Extended Description - tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : NULL; + tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : ePyObject(); break; case 'C': // Current Time tmp = nowTime; @@ -1560,6 +1625,7 @@ void fillTuple(PyObject *tuple, char *argstring, int argcount, PyObject *service tmp = service; inc_refcount = true; break; + case 'n': // short service name case 'N': // service name tmp = service_name; inc_refcount = true; @@ -1575,13 +1641,13 @@ void fillTuple(PyObject *tuple, char *argstring, int argcount, PyObject *service } } -int handleEvent(ePtr &ptr, PyObject *dest_list, char* argstring, int argcount, PyObject *service, PyObject *nowTime, PyObject *service_name, PyObject *convertFunc, PyObject *convertFuncArgs) +int handleEvent(ePtr &ptr, ePyObject dest_list, char* argstring, int argcount, ePyObject service, ePyObject nowTime, ePyObject service_name, ePyObject convertFunc, ePyObject convertFuncArgs) { if (convertFunc) { fillTuple(convertFuncArgs, argstring, argcount, service, ptr, nowTime, service_name); - PyObject *result = PyObject_CallObject(convertFunc, convertFuncArgs); - if (result == NULL) + ePyObject result = PyObject_CallObject(convertFunc, convertFuncArgs); + if (result) { if (service_name) Py_DECREF(service_name); @@ -1599,7 +1665,7 @@ int handleEvent(ePtr &ptr, PyObject *dest_list, char* argstring, } else { - PyObject *tuple = PyTuple_New(argcount); + ePyObject tuple = PyTuple_New(argcount); fillTuple(tuple, argstring, argcount, service, ptr, nowTime, service_name); PyList_Append(dest_list, tuple); Py_DECREF(tuple); @@ -1619,6 +1685,7 @@ int handleEvent(ePtr &ptr, PyObject *dest_list, char* argstring, // C = Current Time // R = Service Reference // N = Service Name +// n = Short Service Name // then for each service follows a tuple // first tuple entry is the servicereference (as string... use the ref.toString() function) // the second is the type of query @@ -1628,12 +1695,12 @@ int handleEvent(ePtr &ptr, PyObject *dest_list, char* argstring, // +1 = event after given start_time // the third // when type is eventid it is the event_id -// when type is time then it is the start_time ( 0 for now_time ) +// when type is time then it is the start_time ( -1 for now_time ) // the fourth is the end_time .. ( optional .. for query all events in time range) -PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc) +PyObject *eEPGCache::lookupEvent(ePyObject list, ePyObject convertFunc) { - PyObject *convertFuncArgs=NULL; + ePyObject convertFuncArgs; int argcount=0; char *argstring=NULL; if (!PyList_Check(list)) @@ -1654,7 +1721,7 @@ PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc) } else { - PyObject *argv=PyList_GET_ITEM(list, 0); // borrowed reference! + ePyObject argv=PyList_GET_ITEM(list, 0); // borrowed reference! if (PyString_Check(argv)) { argstring = PyString_AS_STRING(argv); @@ -1677,17 +1744,17 @@ PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc) convertFuncArgs = PyTuple_New(argcount); } - PyObject *nowTime = strchr(argstring, 'C') ? + ePyObject nowTime = strchr(argstring, 'C') ? PyLong_FromLong(eDVBLocalTimeHandler::getInstance()->nowTime()) : - NULL; + ePyObject(); - bool must_get_service_name = strchr(argstring, 'N') ? true : false; + int must_get_service_name = strchr(argstring, 'N') ? 1 : strchr(argstring, 'n') ? 2 : 0; // create dest list - PyObject *dest_list=PyList_New(0); + ePyObject dest_list=PyList_New(0); while(listSize > listIt) { - PyObject *item=PyList_GET_ITEM(list, listIt++); // borrowed reference! + ePyObject item=PyList_GET_ITEM(list, listIt++); // borrowed reference! if (PyTuple_Check(item)) { bool service_changed=false; @@ -1697,10 +1764,10 @@ PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc) int minutes=0; int tupleSize=PyTuple_Size(item); int tupleIt=0; - PyObject *service=NULL; + ePyObject service; while(tupleSize > tupleIt) // parse query args { - PyObject *entry=PyTuple_GET_ITEM(item, tupleIt); // borrowed reference! + ePyObject entry=PyTuple_GET_ITEM(item, tupleIt); // borrowed reference! switch(tupleIt++) { case 0: @@ -1732,7 +1799,7 @@ PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc) break; } } - eServiceReference ref(PyString_AS_STRING(service)); + eServiceReference ref(handleGroup(eServiceReference(PyString_AS_STRING(service)))); if (ref.type != eServiceReference::idDVB) { eDebug("service reference for epg query is not valid"); @@ -1756,7 +1823,7 @@ PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc) } } - PyObject *service_name=NULL; + ePyObject service_name; if (must_get_service_name) { ePtr sptr; @@ -1769,6 +1836,19 @@ PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc) { std::string name; sptr->getName(ref, name); + + if (must_get_service_name == 1) + { + unsigned int pos; + // filter short name brakets + while((pos = name.find("\xc2\x86")) != std::string::npos) + name.erase(pos,2); + while((pos = name.find("\xc2\x87")) != std::string::npos) + name.erase(pos,2); + } + else + name = buildShortName(name); + if (name.length()) service_name = PyString_FromString(name.c_str()); } @@ -1821,9 +1901,9 @@ skip_entry: return dest_list; } -void fillTuple2(PyObject *tuple, const char *argstring, int argcount, eventData *evData, ePtr &ptr, PyObject *service_name, PyObject *service_reference) +void fillTuple2(ePyObject tuple, const char *argstring, int argcount, eventData *evData, ePtr &ptr, ePyObject service_name, ePyObject service_reference) { - PyObject *tmp=NULL; + ePyObject tmp; int pos=0; while(pos < argcount) { @@ -1838,29 +1918,30 @@ void fillTuple2(PyObject *tuple, const char *argstring, int argcount, eventData break; case 'B': // Event Begin Time if (ptr) - tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : NULL; + tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : ePyObject(); else tmp = PyLong_FromLong(evData->getStartTime()); break; case 'D': // Event Duration if (ptr) - tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : NULL; + tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : ePyObject(); else tmp = PyLong_FromLong(evData->getDuration()); break; case 'T': // Event Title - tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : NULL; + tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : ePyObject(); break; case 'S': // Event Short Description - tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : NULL; + tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : ePyObject(); break; case 'E': // Event Extended Description - tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : NULL; + tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : ePyObject(); break; case 'R': // service reference string tmp = service_reference; inc_refcount = true; break; + case 'n': // short service name case 'N': // service name tmp = service_name; inc_refcount = true; @@ -1887,6 +1968,7 @@ void fillTuple2(PyObject *tuple, const char *argstring, int argcount, eventData // E = Event Extended Description // R = Service Reference // N = Service Name +// n = Short Service Name // the second tuple entry is the MAX matches value // the third tuple entry is the type of query // 0 = search for similar broadcastings (SIMILAR_BROADCASTINGS_SEARCH) @@ -1901,9 +1983,9 @@ void fillTuple2(PyObject *tuple, const char *argstring, int argcount, eventData // 0 = case sensitive (CASE_CHECK) // 1 = case insensitive (NO_CASECHECK) -PyObject *eEPGCache::search(PyObject *arg) +PyObject *eEPGCache::search(ePyObject arg) { - PyObject *ret = 0; + ePyObject ret; int descridx = -1; __u32 descr[512]; int eventid = -1; @@ -1919,7 +2001,7 @@ PyObject *eEPGCache::search(PyObject *arg) int tuplesize=PyTuple_Size(arg); if (tuplesize > 0) { - PyObject *obj = PyTuple_GET_ITEM(arg,0); + ePyObject obj = PyTuple_GET_ITEM(arg,0); if (PyString_Check(obj)) { argcount = PyString_GET_SIZE(obj); @@ -1950,7 +2032,7 @@ PyObject *eEPGCache::search(PyObject *arg) querytype = PyLong_AsLong(PyTuple_GET_ITEM(arg, 2)); if (tuplesize > 4 && querytype == 0) { - PyObject *obj = PyTuple_GET_ITEM(arg, 3); + ePyObject obj = PyTuple_GET_ITEM(arg, 3); if (PyString_Check(obj)) { refstr = PyString_AS_STRING(obj); @@ -1964,10 +2046,10 @@ PyObject *eEPGCache::search(PyObject *arg) if (evData) { __u8 *data = evData->EITdata; - int tmp = evData->ByteSize-12; - __u32 *p = (__u32*)(data+12); + int tmp = evData->ByteSize-10; + __u32 *p = (__u32*)(data+10); // search short and extended event descriptors - while(tmp>0) + while(tmp>3) { __u32 crc = *p++; descriptorMap::iterator it = @@ -2007,7 +2089,7 @@ PyObject *eEPGCache::search(PyObject *arg) } else if (tuplesize > 4 && (querytype == 1 || querytype == 2) ) { - PyObject *obj = PyTuple_GET_ITEM(arg, 3); + ePyObject obj = PyTuple_GET_ITEM(arg, 3); if (PyString_Check(obj)) { int casetype = PyLong_AsLong(PyTuple_GET_ITEM(arg, 4)); @@ -2110,7 +2192,7 @@ PyObject *eEPGCache::search(PyObject *arg) if (descridx > -1) { int maxcount=maxmatches; - eServiceReferenceDVB ref(refstr?refstr:""); + eServiceReferenceDVB ref(refstr?(const eServiceReferenceDVB&)handleGroup(eServiceReference(refstr)):eServiceReferenceDVB("")); // ref is only valid in SIMILAR_BROADCASTING_SEARCH // in this case we start searching with the base service bool first = ref.valid() ? true : false; @@ -2124,8 +2206,8 @@ PyObject *eEPGCache::search(PyObject *arg) ++cit; continue; } - PyObject *service_name=0; - PyObject *service_reference=0; + ePyObject service_name; + ePyObject service_reference; timeMap &evmap = cit->second.second; // check all events for (timeMap::iterator evit(evmap.begin()); evit != evmap.end() && maxcount; ++evit) @@ -2134,11 +2216,11 @@ PyObject *eEPGCache::search(PyObject *arg) if ( evid == eventid) continue; __u8 *data = evit->second->EITdata; - int tmp = evit->second->ByteSize-12; - __u32 *p = (__u32*)(data+12); + int tmp = evit->second->ByteSize-10; + __u32 *p = (__u32*)(data+10); // check if any of our descriptor used by this event int cnt=-1; - while(tmp>0) + while(tmp>3) { __u32 crc32 = *p++; for ( int i=0; i <= descridx; ++i) @@ -2165,24 +2247,41 @@ PyObject *eEPGCache::search(PyObject *arg) eDebug("event not found !!!!!!!!!!!"); } // create service name - if (!service_name && strchr(argstring,'N')) + if (!service_name) { - ePtr sptr; - eServiceCenterPtr service_center; - eServiceCenter::getPrivInstance(service_center); - if (service_center) + int must_get_service_name = strchr(argstring, 'N') ? 1 : strchr(argstring, 'n') ? 2 : 0; + if (must_get_service_name) { - service_center->info(ref, sptr); - if (sptr) + ePtr sptr; + eServiceCenterPtr service_center; + eServiceCenter::getPrivInstance(service_center); + if (service_center) { - std::string name; - sptr->getName(ref, name); - if (name.length()) - service_name = PyString_FromString(name.c_str()); + service_center->info(ref, sptr); + if (sptr) + { + std::string name; + sptr->getName(ref, name); + + if (must_get_service_name == 1) + { + unsigned int pos; + // filter short name brakets + while((pos = name.find("\xc2\x86")) != std::string::npos) + name.erase(pos,2); + while((pos = name.find("\xc2\x87")) != std::string::npos) + name.erase(pos,2); + } + else + name = buildShortName(name); + + if (name.length()) + service_name = PyString_FromString(name.c_str()); + } } + if (!service_name) + service_name = PyString_FromString(""); } - if (!service_name) - service_name = PyString_FromString(""); } // create servicereference string if (!service_reference && strchr(argstring,'R')) @@ -2191,7 +2290,7 @@ PyObject *eEPGCache::search(PyObject *arg) if (!ret) ret = PyList_New(0); // create tuple - PyObject *tuple = PyTuple_New(argcount); + ePyObject tuple = PyTuple_New(argcount); // fill tuple fillTuple2(tuple, argstring, argcount, evit->second, ptr, service_name, service_reference); PyList_Append(ret, tuple); @@ -2216,10 +2315,7 @@ PyObject *eEPGCache::search(PyObject *arg) } if (!ret) - { - Py_INCREF(Py_None); - ret=Py_None; - } + Py_RETURN_NONE; return ret; } @@ -2362,62 +2458,70 @@ void eEPGCache::privateSectionRead(const uniqueEPGKey ¤t_service, const __ int descriptors_length = (data[ptr++]&0x0F) << 8; descriptors_length |= data[ptr++]; - while ( descriptors_length > 0 ) + while ( descriptors_length > 1 ) { int descr_type = data[ptr]; int descr_len = data[ptr+1]; - descriptors_length -= (descr_len+2); - if ( descr_type == 0xf2 ) + descriptors_length -= 2; + if (descriptors_length >= descr_len) { - ptr+=2; - int tsid = data[ptr++] << 8; - tsid |= data[ptr++]; - int onid = data[ptr++] << 8; - onid |= data[ptr++]; - int sid = data[ptr++] << 8; - sid |= data[ptr++]; + descriptors_length -= descr_len; + if ( descr_type == 0xf2 && descr_len > 5) + { + ptr+=2; + int tsid = data[ptr++] << 8; + tsid |= data[ptr++]; + int onid = data[ptr++] << 8; + onid |= data[ptr++]; + int sid = data[ptr++] << 8; + sid |= data[ptr++]; // WORKAROUND for wrong transmitted epg data (01.08.2006) - if ( onid == 0x85 ) - { - switch( (tsid << 16) | sid ) + if ( onid == 0x85 ) { - case 0x01030b: sid = 0x1b; tsid = 4; break; // Premiere Win - case 0x0300f0: sid = 0xe0; tsid = 2; break; - case 0x0300f1: sid = 0xe1; tsid = 2; break; - case 0x0300f5: sid = 0xdc; break; - case 0x0400d2: sid = 0xe2; tsid = 0x11; break; - case 0x1100d3: sid = 0xe3; break; + switch( (tsid << 16) | sid ) + { + case 0x01030b: sid = 0x1b; tsid = 4; break; // Premiere Win + case 0x0300f0: sid = 0xe0; tsid = 2; break; + case 0x0300f1: sid = 0xe1; tsid = 2; break; + case 0x0300f5: sid = 0xdc; break; + case 0x0400d2: sid = 0xe2; tsid = 0x11; break; + case 0x1100d3: sid = 0xe3; break; + } } - } //////////////////////////////////////////// - uniqueEPGKey service( sid, onid, tsid ); - descr_len -= 6; - while( descr_len > 0 ) - { - __u8 datetime[5]; - datetime[0] = data[ptr++]; - datetime[1] = data[ptr++]; - int tmp_len = data[ptr++]; - descr_len -= 3; - while( tmp_len > 0 ) + uniqueEPGKey service( sid, onid, tsid ); + descr_len -= 6; + while( descr_len > 2 ) { - memcpy(datetime+2, data+ptr, 3); - ptr+=3; + __u8 datetime[5]; + datetime[0] = data[ptr++]; + datetime[1] = data[ptr++]; + int tmp_len = data[ptr++]; descr_len -= 3; - tmp_len -= 3; - start_times[datetime].push_back(service); + if (descr_len >= tmp_len) + { + descr_len -= tmp_len; + while( tmp_len > 2 ) + { + memcpy(datetime+2, data+ptr, 3); + ptr += 3; + tmp_len -= 3; + start_times[datetime].push_back(service); + } + } } } - } - else - { - *pdescr++=data+ptr; - ptr += 2; - ptr += descr_len; + else + { + *pdescr++=data+ptr; + ptr += 2; + ptr += descr_len; + } } } + ASSERT(pdescr <= &descriptors[65]) __u8 event[4098]; eit_event_struct *ev_struct = (eit_event_struct*) event; ev_struct->running_status = 0; @@ -2431,6 +2535,7 @@ void eEPGCache::privateSectionRead(const uniqueEPGKey ¤t_service, const __ ptr+=(*d++)[1]; ptr+=2; } + ASSERT(ptr <= 4098); for ( std::map< date_time, std::list >::iterator it(start_times.begin()); it != start_times.end(); ++it ) { time_t now = eDVBLocalTimeHandler::getInstance()->nowTime(); @@ -2470,6 +2575,7 @@ void eEPGCache::privateSectionRead(const uniqueEPGKey ¤t_service, const __ eventData *d = new eventData( ev_struct, bptr, PRIVATE ); evMap[event_id] = d; tmMap[stime] = d; + ASSERT(bptr <= 4098); } } @@ -2496,7 +2602,7 @@ void eEPGCache::channel_data::startPrivateReader() void eEPGCache::channel_data::readPrivateData( const __u8 *data) { - if ( seenPrivateSections.find( data[6] ) == seenPrivateSections.end() ) + if ( seenPrivateSections.find(data[6]) == seenPrivateSections.end() ) { cache->privateSectionRead(m_PrivateService, data); seenPrivateSections.insert(data[6]); @@ -2573,19 +2679,21 @@ void eEPGCache::channel_data::timeMHW2DVB( u_char day, u_char hours, u_char minu putenv("TZ=CET-1CEST,M3.5.0/2,M10.5.0/3"); tzset(); - tm *localnow = localtime( &dt ); + tm localnow; + localtime_r(&dt, &localnow); if (day == 7) day = 0; - if ( day + 1 < localnow->tm_wday ) // day + 1 to prevent old events to show for next week. + if ( day + 1 < localnow.tm_wday ) // day + 1 to prevent old events to show for next week. day += 7; if (local_hours <= 5) day++; - dt += 3600*24*(day - localnow->tm_wday); // Shift dt to the recording date (local time zone). - dt += 3600*(local_hours - localnow->tm_hour); // Shift dt to the recording hour. + dt += 3600*24*(day - localnow.tm_wday); // Shift dt to the recording date (local time zone). + dt += 3600*(local_hours - localnow.tm_hour); // Shift dt to the recording hour. - tm *recdate = gmtime( &dt ); // This will also take care of DST. + tm recdate; + gmtime_r( &dt, &recdate ); // This will also take care of DST. if ( old_tz == NULL ) unsetenv( "TZ" ); @@ -2595,21 +2703,22 @@ void eEPGCache::channel_data::timeMHW2DVB( u_char day, u_char hours, u_char minu // Calculate MJD according to annex in ETSI EN 300 468 int l=0; - if ( recdate->tm_mon <= 1 ) // Jan or Feb + if ( recdate.tm_mon <= 1 ) // Jan or Feb l=1; - int mjd = 14956 + recdate->tm_mday + int( (recdate->tm_year - l) * 365.25) + - int( (recdate->tm_mon + 2 + l * 12) * 30.6001); + int mjd = 14956 + recdate.tm_mday + int( (recdate.tm_year - l) * 365.25) + + int( (recdate.tm_mon + 2 + l * 12) * 30.6001); return_time[0] = (mjd & 0xFF00)>>8; return_time[1] = mjd & 0xFF; - timeMHW2DVB( recdate->tm_hour, minutes, return_time+2 ); + timeMHW2DVB( recdate.tm_hour, minutes, return_time+2 ); } void eEPGCache::channel_data::storeTitle(std::map<__u32, mhw_title_t>::iterator itTitle, std::string sumText, const __u8 *data) // data is borrowed from calling proc to save memory space. { __u8 name[34]; + // For each title a separate EIT packet will be sent to eEPGCache::sectionRead() bool isMHW2 = itTitle->second.mhw2_mjd_hi || itTitle->second.mhw2_mjd_lo || itTitle->second.mhw2_duration_hi || itTitle->second.mhw2_duration_lo; @@ -2617,6 +2726,7 @@ void eEPGCache::channel_data::storeTitle(std::map<__u32, mhw_title_t>::iterator eit_t *packet = (eit_t *) data; packet->table_id = 0x50; packet->section_syntax_indicator = 1; + packet->service_id_hi = m_channels[ itTitle->second.channel_id - 1 ].channel_id_hi; packet->service_id_lo = m_channels[ itTitle->second.channel_id - 1 ].channel_id_lo; packet->version_number = 0; // eEPGCache::sectionRead() will dig this for the moment @@ -2801,10 +2911,11 @@ void eEPGCache::channel_data::readMHWData(const __u8 *data) int record_size = sizeof( mhw_channel_name_t ); int nbr_records = int (len/record_size); + m_channels.resize(nbr_records); for ( int i = 0; i < nbr_records; i++ ) { mhw_channel_name_t *channel = (mhw_channel_name_t*) &data[4 + i*record_size]; - m_channels.push_back( *channel ); + m_channels[i]=*channel; } haveData |= MHW; @@ -2870,7 +2981,7 @@ void eEPGCache::channel_data::readMHWData(const __u8 *data) m_titles[ title_id ] = *title; if ( (title->ms.summary_available) && (m_program_ids.find(program_id) == m_program_ids.end()) ) // program_ids will be used to gather summaries. - m_program_ids[ program_id ] = title_id; + m_program_ids.insert(std::pair<__u32,__u32>(program_id,title_id)); return; // Continue reading of the current table. } else if (!checkTimeout()) @@ -2903,7 +3014,7 @@ void eEPGCache::channel_data::readMHWData(const __u8 *data) memcpy(&tmp, &data, sizeof(void*)); tmp[len+3] = 0; // Terminate as a string. - std::map<__u32, __u32>::iterator itProgid( m_program_ids.find( program_id ) ); + std::multimap<__u32, __u32>::iterator itProgid( m_program_ids.find( program_id ) ); if ( itProgid == m_program_ids.end() ) { /* This part is to prevent to looping forever if some summaries are not received yet. There is a timeout of 4 sec. after the last successfully read summary. */ @@ -2962,10 +3073,11 @@ void eEPGCache::channel_data::readMHWData2(const __u8 *data) else if (m_MHWFilterMask2.pid == 0x231 && m_MHWFilterMask2.data[0] == 0xC8 && m_MHWFilterMask2.data[1] == 0) // Channels table { - int num_channels = data[120]; - if(dataLen > 120) + int num_channels = data[119]; + m_channels.resize(num_channels); + if(dataLen > 119) { - int ptr = 121 + 6 * num_channels; + int ptr = 120 + 8 * num_channels; if( dataLen > ptr ) { for( int chid = 0; chid < num_channels; ++chid ) @@ -2981,18 +3093,17 @@ void eEPGCache::channel_data::readMHWData2(const __u8 *data) else goto abort; // data seems consistent... - const __u8 *tmp = data+121; + const __u8 *tmp = data+120; for (int i=0; i < num_channels; ++i) { mhw_channel_name_t channel; + channel.network_id_hi = *(tmp++); + channel.network_id_lo = *(tmp++); channel.transport_stream_id_hi = *(tmp++); channel.transport_stream_id_lo = *(tmp++); channel.channel_id_hi = *(tmp++); channel.channel_id_lo = *(tmp++); -#warning FIXME hardcoded network_id in mhw2 epg - channel.network_id_hi = 0; // hardcoded astra 19.2 - channel.network_id_lo = 1; - m_channels.push_back(channel); + m_channels[i]=channel; tmp+=2; } for (int i=0; i < num_channels; ++i) @@ -3071,7 +3182,7 @@ void eEPGCache::channel_data::readMHWData2(const __u8 *data) __u8 slen = data[pos+10] & 0x3f; __u8 *dest = ((__u8*)title.title)-4; memcpy(dest, &data[pos+11], slen>33 ? 33 : slen); - memset(dest+slen, 0x20, 33-slen); + memset(dest+slen, 0, 33-slen); pos += 11 + slen; // not used theme id (data[7] & 0x3f) + (data[pos] & 0x3f); __u32 summary_id = (data[pos+1] << 8) | data[pos+2]; @@ -3079,19 +3190,27 @@ void eEPGCache::channel_data::readMHWData2(const __u8 *data) // Create unique key per title __u32 title_id = (title.channel_id<<16) | (title.program_id_ml<<8) | title.program_id_lo; -// eDebug("program_id: %08x, %s", program_id, -// std::string((const char *)title.title, (int)(slen > 23 ? 23 : slen)).c_str()); - pos += 4; - if ( m_titles.find( title_id ) == m_titles.end() ) + std::map<__u32, mhw_title_t>::iterator it = m_titles.find( title_id ); + if ( it == m_titles.end() ) { - startTimeout(4000); + startTimeout(5000); m_titles[ title_id ] = title; - if (summary_id != 0xFFFF && // no summary avail - m_program_ids.find(summary_id) == m_program_ids.end()) + if (summary_id != 0xFFFF) { - m_program_ids[ summary_id ] = title_id; + bool add=true; + std::multimap<__u32, __u32>::iterator it(m_program_ids.lower_bound(summary_id)); + while (it != m_program_ids.end() && it->first == summary_id) + { + if (it->second == title_id) { + add=false; + break; + } + ++it; + } + if (add) + m_program_ids.insert(std::pair<__u32,__u32>(summary_id,title_id)); } } else @@ -3111,7 +3230,7 @@ start_summary: // Titles table has been read, there are summaries to read. // Start reading summaries, store corresponding titles on the fly. startMHWReader2(0x236, 0x96); - startTimeout(4000); + startTimeout(15000); return; } } @@ -3121,84 +3240,91 @@ start_summary: else if (m_MHWFilterMask2.pid == 0x236 && m_MHWFilterMask2.data[0] == 0x96) // Summaries table { - int len, loop, pos, lenline; - bool valid; - valid = true; - if( dataLen > 18 ) + if (!checkTimeout()) { - loop = data[12]; - pos = 13 + loop; - if( dataLen > pos ) + int len, loop, pos, lenline; + bool valid; + valid = true; + if( dataLen > 15 ) { - loop = data[pos] & 0x0f; - pos += 1; + loop = data[14]; + pos = 15 + loop; if( dataLen > pos ) { - len = 0; - for( ; loop > 0; --loop ) + loop = data[pos] & 0x0f; + pos += 1; + if( dataLen > pos ) { - if( dataLen > (pos+len) ) + len = 0; + for( ; loop > 0; --loop ) { - lenline = data[pos+len]; - len += lenline + 1; + if( dataLen > (pos+len) ) + { + lenline = data[pos+len]; + len += lenline + 1; + } + else + valid=false; } - else - valid=false; } } } - } - else if (!checkTimeout()) - return; // continue reading - if (valid && !checkTimeout()) - { - // data seems consistent... - __u32 summary_id = (data[3]<<8)|data[4]; - - // ugly workaround to convert const __u8* to char* - char *tmp=0; - memcpy(&tmp, &data, sizeof(void*)); - - len = 0; - loop = data[12]; - pos = 13 + loop; - loop = tmp[pos] & 0x0f; - pos += 1; - for( ; loop > 0; loop -- ) - { - lenline = tmp[pos+len]; - tmp[pos+len] = ' '; - len += lenline + 1; - } - if( len > 0 ) - tmp[pos+len] = 0; - else - tmp[pos+1] = 0; - - std::map<__u32, __u32>::iterator itProgid( m_program_ids.find( summary_id ) ); - if ( itProgid == m_program_ids.end() ) - { /* This part is to prevent to looping forever if some summaries are not received yet. - There is a timeout of 4 sec. after the last successfully read summary. */ - - if ( !m_program_ids.empty() && !checkTimeout() ) - return; // Continue reading of the current table. - } else + return; // continue reading + if (valid) { - startTimeout(4000); - std::string the_text = (char *) (data + pos + 1); + // data seems consistent... + __u32 summary_id = (data[3]<<8)|data[4]; - // Find corresponding title, store title and summary in epgcache. - std::map<__u32, mhw_title_t>::iterator itTitle( m_titles.find( itProgid->second ) ); - if ( itTitle != m_titles.end() ) + // ugly workaround to convert const __u8* to char* + char *tmp=0; + memcpy(&tmp, &data, sizeof(void*)); + + len = 0; + loop = data[14]; + pos = 15 + loop; + loop = tmp[pos] & 0x0f; + pos += 1; + for( ; loop > 0; loop -- ) { - storeTitle( itTitle, the_text, data ); - m_titles.erase( itTitle ); + lenline = tmp[pos+len]; + tmp[pos+len] = ' '; + len += lenline + 1; + } + if( len > 0 ) + tmp[pos+len] = 0; + else + tmp[pos+1] = 0; + + std::multimap<__u32, __u32>::iterator itProgId( m_program_ids.lower_bound(summary_id) ); + if ( itProgId == m_program_ids.end() || itProgId->first != summary_id) + { /* This part is to prevent to looping forever if some summaries are not received yet. + There is a timeout of 4 sec. after the last successfully read summary. */ + if ( !m_program_ids.empty() ) + return; // Continue reading of the current table. + } + else + { + startTimeout(15000); + std::string the_text = (char *) (data + pos + 1); + + while( itProgId != m_program_ids.end() && itProgId->first == summary_id ) + { + // Find corresponding title, store title and summary in epgcache. + std::map<__u32, mhw_title_t>::iterator itTitle( m_titles.find( itProgId->second ) ); + if ( itTitle != m_titles.end() ) + { + storeTitle( itTitle, the_text, data ); + m_titles.erase( itTitle ); + } + m_program_ids.erase( itProgId++ ); + } + if ( !m_program_ids.empty() ) + return; // Continue reading of the current table. } - m_program_ids.erase( itProgid ); - if ( !m_program_ids.empty() ) - return; // Continue reading of the current table. } + else + return; // continue reading } } if (isRunning & eEPGCache::MHW)