Initial commit

1 files changed, 598 insertions, 0 deletions

diff --git a/process.cpp b/process.cpp
new file mode 100644
index 0000000..47d5563
--- /dev/null
+++ b/process.cpp
@@ -0,0 +1,598 @@
+/*
+ * process.cpp: A plugin for the Video Disk Recorder
+ *
+ * See the README file for copyright information and how to reach the author.
+ *
+ * $Id$
+ */
+
+#include <stdio.h>
+#include <vdr/epg.h>
+#include <vdr/tools.h>
+#include <vdr/skins.h>
+#include <ctype.h>
+#include "process.h"
+#include "readline.h"
+
+// --- cInfosatevent
+cInfosatevent::cInfosatevent()
+{
+    title = NULL;
+    shortText = NULL;
+    description = NULL;
+    announcement=NULL;
+    country=NULL;
+    genre=NULL;
+    original=NULL;
+    year=-1;
+    fsk=-1;
+    category=-1;
+    startTime = 0;
+    duration = 0;
+}
+
+cInfosatevent::~cInfosatevent()
+{
+    free(title);
+    free(shortText);
+    free(description);
+    free(announcement);
+    free(country);
+    free(genre);
+    free(original);
+}
+
+void cInfosatevent::SetOriginal(const char *Original)
+{
+    original = strcpyrealloc(original, Original);
+}
+
+void cInfosatevent::SetGenre(const char *Genre)
+{
+    genre = strcpyrealloc(genre, Genre);
+}
+
+void cInfosatevent::SetCountry(const char *Country)
+{
+    country = strcpyrealloc(country, Country);
+}
+
+void cInfosatevent::SetAnnouncement(const char *Announcement)
+{
+    announcement = strcpyrealloc(announcement, Announcement);
+}
+
+void cInfosatevent::SetTitle(const char *Title)
+{
+    title = strcpyrealloc(title, Title);
+}
+
+void cInfosatevent::SetShortText(const char *ShortText)
+{
+    shortText = strcpyrealloc(shortText, ShortText);
+}
+
+void cInfosatevent::SetDescription(const char *Description)
+{
+    description = strcpyrealloc(description, Description);
+}
+
+const char *cInfosatevent::Description(const char *oldDescription)
+{
+    //  Add Category:, Genre:, Year:, Country:, Originaltitle:, FSK: , Rating: [if available] ...
+    /*
+      char fmt[100];
+      char *descr;
+
+      if (oldDescription) {
+         descr=(char *) oldDescription;
+         char *extEPG=strstr(descr,"\n\n");
+         if (extEPG) *extEPG=0;
+      } else {
+         descr=description;
+      }
+
+      descr=strcatrealloc(descr,"\n\n");
+
+      if (category!=-1) {
+        descr=strcatrealloc(descr,"Category: ");
+        descr=strcatrealloc(descr,Category());
+        descr=strcatrealloc(descr,"\n");
+      }
+      if (genre) {
+        descr=strcatrealloc(descr,"Genre: ");
+        descr=strcatrealloc(descr,Genre());
+        descr=strcatrealloc(descr,"\n");
+      }
+      if (year!=-1) {
+        strncat(fmt,"Year: %i\n",9);
+      }
+      if (country) {
+        descr=strcatrealloc(descr,"Country: ");
+        descr=strcatrealloc(descr,Country());
+        descr=strcatrealloc(descr,"\n");
+      }
+      if (original) {
+        descr=strcatrealloc(descr,"Originaltitle: ");
+        descr=strcatrealloc(descr,Original());
+        descr=strcatrealloc(descr,"\n");
+      }
+      if (fsk!=-1) {
+        strncat(fmt,"FSK: %i\n",8);
+      }
+      if (announcement) {
+        descr=strcatrealloc(descr,"Rating: ");
+        descr=strcatrealloc(descr,Announcement());
+        descr=strcatrealloc(descr,"\n");
+      }
+
+    */
+    return (const char*) description;
+}
+
+// --- cProcessInfosatepg
+cProcessInfosatepg::cProcessInfosatepg(int Mac, cGlobalInfosatepg *Global)
+{
+    global=Global;
+
+    FILE *f;
+    const char *file = global->Infosatdata[Mac].GetFile();
+    f=fopen(file,"r");
+    if (f)
+    {
+        if (ParseInfosatepg(f))
+        {
+            Global->Infosatdata[Mac].SetProcessed();
+        }
+        else
+        {
+            esyslog("infosatepg: failed to get writelock while parsing %s",global->Infosatdata[Mac].GetFile());
+        }
+        fclose(f);
+    }
+}
+
+cEvent *cProcessInfosatepg::SearchEvent(cSchedule* Schedule, cInfosatevent *iEvent)
+{
+    cEvent *f=NULL;
+    int maxdiff=MAX_EVENTTIMEDIFF*60;
+
+    for (cEvent *p = Schedule->Events()->First(); p; p = Schedule->Events()->Next(p))
+    {
+        if (!strcmp(p->Title(),iEvent->Title()))
+        {
+            // found event with same title
+            int diff=abs((int) difftime(p->StartTime(),iEvent->StartTime()));
+            if (diff<=global->EventTimeDiff)
+            {
+                if (diff<maxdiff)
+                {
+                    f=p;
+                    maxdiff=diff;
+                }
+            }
+        }
+    }
+    return f;
+}
+
+bool cProcessInfosatepg::AddInfosatEvent(cChannel *channel, cInfosatevent *iEvent)
+{
+    if ((!channel) || (!iEvent)) return true;
+    if (iEvent->GetEventUse()==USE_NOTHING) return true; // this should never happen!
+    if (iEvent->StartTime()<time(NULL)) return true; // don't deal with old events
+    if (iEvent->StartTime()>(time(NULL)+86400)) return true; // don't deal with events to far in the future
+
+    cSchedulesLock SchedulesLock(true,2000); // to be safe ;)
+    const cSchedules *Schedules = cSchedules::Schedules(SchedulesLock);
+    if (!Schedules) return false; // No write lock -> try later!
+    cSchedule* Schedule = (cSchedule *) Schedules->GetSchedule(channel,true);
+    if (!Schedule) return true;
+
+    time_t start;
+
+    const cEvent *lastEvent=Schedule->Events()->Last();
+    if ((lastEvent) && (iEvent->StartTime()<lastEvent->EndTime()))
+    {
+        // try to find, 1st with EventID
+        cEvent *Event = (cEvent *) Schedule->GetEvent(iEvent->EventID());
+        // 2nd with StartTime +/- WaitTime
+        if (!Event) Event= (cEvent *) SearchEvent(Schedule,iEvent);
+        if (!Event) return true; // just bail out with ok
+
+	start=iEvent->StartTime();
+	dsyslog("infosatepg: ievent %s [%s]", iEvent->Title(),ctime(&start));
+
+        // change existing event
+        Event->SetShortText(iEvent->ShortText());
+        if (iEvent->GetEventUse()==USE_SHORTLONGTEXT)
+        {
+            Event->SetDescription(iEvent->Description(NULL));
+        }
+        if (iEvent->GetEventUse()==USE_SHORTTEXTEPG)
+        {
+            Event->SetDescription(iEvent->Description(Event->Description()));
+        }
+        if (iEvent->GetEventUse()==USE_INTELLIGENT)
+        {
+            if (!Event->Description() || (!iEvent->Description(NULL)))
+            {
+                Event->SetDescription(iEvent->Description(NULL));
+            }
+            else
+            {
+                if (strlen(iEvent->Description(NULL))>strlen(Event->Description()))
+                {
+                    Event->SetDescription(iEvent->Description(NULL));
+                }
+            }
+        }
+        Event->SetTableID(0);
+        Event->SetSeen();
+        start=Event->StartTime();
+        dsyslog("infosatepg: changed event %s [%s]",Event->Title(),ctime(&start));
+    }
+    else
+    {
+        // we are beyond the last event, so just add (if we should)
+        if (iEvent->GetEventUse()!=USE_ALL) return true;
+        start=iEvent->StartTime();
+        cEvent *Event = new cEvent(iEvent->EventID());
+        if (!Event) return true;
+        Event->SetTitle(iEvent->Title());
+        Event->SetShortText(iEvent->ShortText());
+        Event->SetDescription(iEvent->Description(NULL));
+        Event->SetStartTime(iEvent->StartTime());
+        Event->SetDuration(iEvent->Duration());
+        Event->SetVersion(0);
+        start=iEvent->StartTime();
+        dsyslog("infosatepg: adding new event %s [%s]",iEvent->Title(),ctime(&start));
+        Schedule->AddEvent(Event);
+    }
+
+    return true;
+}
+
+cChannel *cProcessInfosatepg::GetInfosatChannel(int frequency, int sid)
+{
+    cChannel *chan;
+    int source = cSource::FromString("S19.2E"); // only from astra 19.2E
+
+    for (int i=0; i<=Channels.MaxNumber(); i++)
+    {
+        chan = Channels.GetByNumber(i,0);
+        if (chan)
+        {
+            if (chan->Source()!=source) continue;
+            if (chan->Sid()!=sid) continue;
+
+            if ((chan->Frequency()==frequency) || (chan->Frequency()+1==frequency) ||
+                    (chan->Frequency()-1==frequency))
+            {
+                return chan;
+            }
+        }
+    }
+    return NULL;
+}
+
+bool cProcessInfosatepg::CheckOriginal(char *s,cInfosatevent *iEvent,cCharSetConv *conv)
+{
+    char *pOT,*pEOT;
+    pOT=strchr(s,'(');
+    if (!pOT) return false;
+    pEOT=strrchr(pOT,')');
+    if (!pEOT) return false;
+    if (pEOT[1]!=0) return false;
+
+    *pOT=*pEOT=0;
+    *pOT++;
+    if (pOT[-1]==' ') pOT[-1]=0;
+    // check some things
+    if (!strcmp(pOT,"TM")) return false;
+    char *endp=NULL;
+    long int ret =strtol(pOT,&endp,10);
+    if ((ret!=0) && (*endp==0)) return false;
+    // with original title
+    iEvent->SetOriginal(conv->Convert(pOT));
+    return true;
+}
+
+bool cProcessInfosatepg::CheckAnnouncement(char *s,cInfosatevent *iEvent)
+{
+    bool ret=true;
+    if ((strlen(s)==16) && (!strcmp(s,"Erstausstrahlung")))
+    {
+        iEvent->SetAnnouncement("Erstausstrahlung");
+    }
+    else if ((strlen(s)==20) && (!strcmp(s,"Deutschland-Premiere")))
+    {
+        iEvent->SetAnnouncement("Deutschland-Premiere");
+    }
+    else if ((strlen(s)>=16) && (!strncmp(s,"Free-TV-Premiere",16)))
+    {
+        iEvent->SetAnnouncement("Free-TV-Premiere!");
+    }
+    else if ((strlen(s)>=9) && (!strncmp(s,"Highlight",9)))
+    {
+        iEvent->SetAnnouncement("Tipp");
+    }
+    else ret=false;
+    return ret;
+}
+
+bool cProcessInfosatepg::ParseInfosatepg(FILE *f)
+{
+    char *s,tag;
+    int fields,index;
+    size_t size;
+    time_t oldstart;
+    bool ignore=true;
+    bool abort=false;
+    struct tm tm;
+    int ieventnr=1;
+    cChannel *chan=NULL;
+    cInfosatevent *ievent=NULL;
+#if VDRVERSNUM < 10701
+    cCharSetConv *conv = new cCharSetConv("ISO-8859-1",cCharSetConv::SystemCharacterTable() ?
+                                          cCharSetConv::SystemCharacterTable() : "UTF-8");
+#else
+    cCharSetConv *conv = new cCharSetConv("ISO-8859-1",NULL);
+#endif
+
+    cReadLineInfosatepg ReadLine;
+    while ((s=ReadLine.Read(f,&size))!=NULL)
+    {
+        if (size<3) continue;
+        if (isdigit(s[0]))
+        {
+            if (ievent)
+            {
+                int category,fsk;
+                fields=sscanf(s,"%d %d",&category,&fsk);
+                if (fields==1) ievent->SetCategory(category);
+                if (fields==2)
+                {
+                    ievent->SetCategory(category);
+                    ievent->SetFSK(fsk);
+                }
+            }
+            continue;
+        }
+
+        s++;
+        tag=*s;
+        s+=2;
+        switch (tag)
+        {
+        case 'P':
+            // contains program (schedule)
+            ignore=true;
+            int frequency,sid,day,month,year,hour,minute;
+            fields=sscanf(s,"%*[^(](%*[^,],%d,%d) %d.%d.%d %*s %*s %d:%d",
+                          &frequency,&sid,&day,&month,&year,&hour,&minute);
+            if (fields==7)
+            {
+                // got all fields
+                chan=GetInfosatChannel(frequency,sid);
+                if (chan)
+                {
+                    if (!global->ChannelExists(chan->GetChannelID(),&index))
+                    {
+                        if (!global->InfosatChannels())
+                        {
+                            // Send message
+                            Skins.QueueMessage(mtInfo,tr("Infosat channellist available"));
+                        }
+                        // Channel is not in global list->add
+                        global->AddChannel(chan->GetChannelID(),USE_NOTHING);
+                    }
+                    else
+                    {
+                        if (global->GetChannelUse(index)!=USE_NOTHING)
+                        {
+                            memset(&tm,0,sizeof(tm));
+                            tm.tm_sec=0;
+                            tm.tm_min=minute;
+                            tm.tm_hour=hour;
+                            tm.tm_mday=day;
+                            tm.tm_mon=month-1;
+                            tm.tm_year=year-1900;
+                            tm.tm_isdst=-1;
+                            oldstart=mktime(&tm);
+                            dsyslog("infosatepg: using '%s'",s);
+                            dsyslog("infosatepg: start on %02i.%02i.%04i %02i:%02i",tm.tm_mday,tm.tm_mon+1,tm.tm_year+1900,
+                                    tm.tm_hour,tm.tm_min);
+                            ignore=false;
+                            ieventnr=1;
+                        }
+                    }
+                }
+            }
+            break;
+        case 'E':
+            // contains event / title
+            if (ignore) continue;
+            if (ievent)
+            {
+                // There was an event without long description -> add
+                if (!AddInfosatEvent(chan,ievent)) abort=true;
+                oldstart=ievent->StartTime();
+                delete ievent; // delete old event
+                ievent =NULL;
+            }
+            else
+            {
+                oldstart=(time_t) -1;
+            }
+            int shour,sminute;
+            char *title;
+            fields=sscanf(s,"%d:%d %a[^^]",&shour,&sminute,&title);
+            if (fields==3)
+            {
+                if (!ievent) ievent = new cInfosatevent;
+                tm.tm_hour=shour;
+                tm.tm_min=sminute;
+                tm.tm_isdst=-1;
+                time_t start=mktime(&tm);
+                if ((oldstart!=(time_t) -1) && (difftime(start,oldstart)<0))
+                {
+                    // last ievent was yesterday
+                    tm.tm_isdst=-1;
+                    tm.tm_mday++;
+                    start=mktime(&tm);
+                }
+                ievent->SetStartTime(start);
+                ievent->SetTitle(conv->Convert(title));
+                free(title);
+                ievent->SetEventUse(global->GetChannelUse(index));
+                ievent->SetEventID(DoSum(ieventnr++,ievent->Title(),strlen(ievent->Title())));
+            }
+            else
+            {
+                dsyslog("infosatepg: ERROR sscanf failed, just %i fields [%s]", fields,s);
+            }
+            break;
+
+        case 'S':
+            if (ignore) continue;
+            if (!ievent) continue;
+            // contains:
+            // short description (,maybe special announcement)<8A>
+            // Genre (,maybe Country and Year), Duration (sometimes missing)
+            char *p8A,*pSA;
+            p8A=strchr(s,0x8A);
+            if (p8A)
+            {
+                *p8A=0;
+                // check for special announcements:
+                // Erstausstrahlung
+                // Deutschland-Premiere
+                // Free-TV-Premiere
+                // Highlight
+                pSA=strrchr(s,',');
+                if (pSA)
+                {
+                    if (CheckAnnouncement(pSA+2,ievent))
+                    {
+                        // announcement was added to short description
+                        *pSA=0;
+                    }
+                    CheckOriginal(s,ievent,conv);
+                    ievent->SetShortText(conv->Convert(s));
+                }
+                else
+                {
+                    if (!CheckAnnouncement(s,ievent))
+                    {
+                        CheckOriginal(s,ievent,conv);
+                        ievent->SetShortText(conv->Convert(s));
+                    }
+                }
+                s=++p8A;
+            }
+            char *pDU;
+            pDU=strrchr(s,',');
+            if (pDU)
+            {
+                *pDU=0;
+                // with genre
+                if (CheckAnnouncement(s,ievent))
+                {
+                    char *tp8A;
+                    tp8A=strchr(s,0x8A);
+                    if (tp8A) s=++tp8A;
+                }
+                char *pCY;
+                pCY=strchr(s,',');
+                if (pCY)
+                {
+                    // with country or year
+                    *pCY=0;
+                    pCY+=2;
+                    if (isdigit(*pCY))
+                    {
+                        // just year (checked)
+                        ievent->SetYear(atoi(pCY));
+                    }
+                    else
+                    {
+                        char *pSP;
+                        pSP=strrchr(pCY,' ');
+                        if (pSP)
+                        {
+                            *pSP=0;
+                            pSP++;
+                            // country and year (check!!)
+                            int year = atoi(pSP);
+                            if (year!=0) ievent->SetYear(year);
+                            ievent->SetCountry(conv->Convert(pCY));
+                        }
+                        else
+                        {
+                            // just country
+                            ievent->SetCountry(conv->Convert(pCY));
+                        }
+                    }
+                }
+                ievent->SetGenre(conv->Convert(s));
+                s=++pDU;
+            }
+            int len;
+            len=strlen(s);
+            if ((s[len-1]=='.') && (s[len-2]=='n') && (s[len-3]=='i') && (s[len-4]=='M'))
+            {
+                // Duration
+                ievent->SetDuration(60*atoi(s));
+            }
+            break;
+
+        case 'L':
+            // contains description
+            if (ignore) continue;
+            if (!ievent) continue;
+            strreplace(s,0x8A,'\n');
+            ievent->SetDescription(conv->Convert(s));
+            if (!AddInfosatEvent(chan,ievent)) abort=true;
+            oldstart=ievent->StartTime();
+            delete ievent;
+            ievent=NULL;
+            break;
+        case 'Q':
+            if (ignore) continue;
+            if (ievent)
+            {
+                if (!AddInfosatEvent(chan,ievent)) abort=true;
+                delete ievent;
+                ievent=NULL;
+            }
+        default:
+            continue;
+        }
+
+        if (abort) break;
+
+    }
+    delete conv;
+    return (!abort);
+}
+
+u_long cProcessInfosatepg::DoSum(u_long sum, const char *buf, int nBytes)
+{
+    int nleft=nBytes;
+    u_short *w = (u_short*)buf;
+
+    while (nleft > 1)
+    {
+        sum += *w++;
+        nleft -= 2;
+    }
+
+    if (nleft == 1)
+    {
+        u_short answer = 0;
+        *(u_char*)(&answer) = *(u_char*)w;
+        sum += answer;
+    }
+    return sum;
+}