diff options
Diffstat (limited to 'global.cpp')
| -rw-r--r-- | global.cpp | 334 |
1 files changed, 334 insertions, 0 deletions
diff --git a/global.cpp b/global.cpp new file mode 100644 index 0000000..0cf65ba --- /dev/null +++ b/global.cpp @@ -0,0 +1,334 @@ +#include <vdr/channels.h> +#include "global.h" + +// --- cGlobalInfosatdata +cGlobalInfosatdata::cGlobalInfosatdata() +{ + file[0]=0; + Init(NULL,0,0,0); +} + +bool cGlobalInfosatdata::NeverSeen(int Day, int Month, int Packetcount) +{ + if ((day!=Day) || (month!=Month) || (pktcnt!=Packetcount)) + { + return true; + } + else + { + return false; + } +} + +void cGlobalInfosatdata::Init(char *File, int Day, int Month, int Packetcount) +{ + if (access(file,R_OK)==0) + { + dsyslog("infosatepg: deleting old %s",file); + unlink(file); + } + ready=false; + processed=false; + day=Day; + month=Month; + pktcnt=Packetcount; + if (File) snprintf(file,sizeof(file),"%s",File); + else file[0]=0; + memset(bitfield,0,sizeof(bitfield)); +} + +int cGlobalInfosatdata::Save(int fd) +{ + if (fd==-1) return -1; + ssize_t ret=write(fd,&ready,sizeof(ready)); + if (ret!=sizeof(ready)) return -1; + ret=write(fd,&processed,sizeof(processed)); + if (ret!=sizeof(processed)) return -1; + ret=write(fd,&day,sizeof(day)); + if (ret!=sizeof(day)) return -1; + ret=write(fd,&month,sizeof(month)); + if (ret!=sizeof(month)) return -1; + ret=write(fd,&pktcnt,sizeof(pktcnt)); + if (ret!=sizeof(pktcnt)) return -1; + ret=write(fd,&bitfield,sizeof(bitfield)); + if (ret!=sizeof(bitfield)) return -1; + ret=write(fd,&file,sizeof(file)); + if (ret!=sizeof(file)) return -1; + return ret; +} + +int cGlobalInfosatdata::Load(int fd) +{ + if (fd==-1) return -1; + ssize_t ret=read(fd,&ready,sizeof(ready)); + if (ret!=sizeof(ready)) return -1; + ret=read(fd,&processed,sizeof(processed)); + if (ret!=sizeof(processed)) return -1; + ret=read(fd,&day,sizeof(day)); + if (ret!=sizeof(day)) return -1; + ret=read(fd,&month,sizeof(month)); + if (ret!=sizeof(month)) return -1; + ret=read(fd,&pktcnt,sizeof(pktcnt)); + if (ret!=sizeof(pktcnt)) return -1; + ret=read(fd,&bitfield,sizeof(bitfield)); + if (ret!=sizeof(bitfield)) return -1; + ret=read(fd,&file,sizeof(file)); + if (ret!=sizeof(file)) return -1; + dsyslog("infosatepg: loaded file=*%s*",file); + return ret; +} + +#ifdef INFOSATEPG_DEBUG +void cGlobalInfosatdata::Debug(const char *Directory) +{ + char file[1024]; + snprintf(file,sizeof(file),"%s/infosatepg9999_999.dat",Directory); + Init(file,99,99,999); + ready=true; +} +#endif + +int cGlobalInfosatdata::ReceivedPercent() +{ + if (pktcnt==0) return 0; + int donecnt=0; + for (int i=0; i<pktcnt; i++) + { + if (GetBit(i)==true) donecnt++; + } + return ((donecnt*100)/pktcnt); +} + +bool cGlobalInfosatdata::ReceivedAll() +{ + if (pktcnt==0) return false; + for (int i=0; i<pktcnt; i++) + { + if (GetBit(i)==false) return false; + } + ready=true; + return true; +} + +// --- cGlobalInfosatepg ----------------------------------------------------- +cGlobalInfosatepg::cGlobalInfosatepg() +{ + // set default values + Pid=1809; + EventTimeDiff=480; // default 8 minutes + Channel=1; + MAC[0]=0x01; + MAC[1]=0x00; + MAC[2]=0x5e; + MAC[3]=0x02; + MAC[4]=0x02; + WaitTime=10; // default 10 seconds + Switched=false; + directory=NULL; + SetDirectory("/tmp"); + infosatchannels=NULL; + numinfosatchannels=0; + this_day=-1; + this_month=-1; +} + +cGlobalInfosatepg::~cGlobalInfosatepg() +{ + free((void *) directory); + free(infosatchannels); +} + +bool cGlobalInfosatepg::SetDirectory(const char *Directory) +{ + if (access(Directory,W_OK | R_OK | X_OK)==-1) + { + // cannot access, maybe create? + if (mkdir(Directory,0775)==-1) + { + // even if the Directory exists, we cannot access + return false; + } + if (access(Directory,W_OK | R_OK| X_OK)==-1) return false; + } + free((void *) directory); + directory = Directory ? strdup(Directory) : NULL; + return true; +} + +bool cGlobalInfosatepg::CheckMAC(struct ethhdr *eth_hdr) +{ + if (!eth_hdr) return false; + if (eth_hdr->h_dest[0]!=MAC[0]) return false; + if (eth_hdr->h_dest[1]!=MAC[1]) return false; + if (eth_hdr->h_dest[2]!=MAC[2]) return false; + if (eth_hdr->h_dest[3]!=MAC[3]) return false; + if (eth_hdr->h_dest[4]!=MAC[4]) return false; + + if (eth_hdr->h_dest[5]<EPG_FIRST_DAY_MAC) return false; + if (eth_hdr->h_dest[5]>EPG_LAST_DAY_MAC) return false; + + return true; +} + +void cGlobalInfosatepg::AddChannel(tChannelID ChannelID,int Usage) +{ + infosatchannels=(struct infosatchannels *) realloc(infosatchannels, + (numinfosatchannels+1)*sizeof(struct infosatchannels)); + if (!infosatchannels) return; + infosatchannels[numinfosatchannels].ChannelID=ChannelID; + infosatchannels[numinfosatchannels].Usage=Usage; + numinfosatchannels++; +} + +tChannelID cGlobalInfosatepg::GetChannelID(int Index) +{ + if (numinfosatchannels==0) return tChannelID::InvalidID; + if ((Index<0) || (Index>numinfosatchannels-1)) + return tChannelID::InvalidID; + return infosatchannels[Index].ChannelID; +} + +bool cGlobalInfosatepg::ChannelExists(tChannelID ChannelID,int *Index) +{ + if (numinfosatchannels==0) return false; + for (int i=0; i<numinfosatchannels; i++) + { + if (infosatchannels[i].ChannelID==tChannelID::InvalidID) continue; + if (infosatchannels[i].ChannelID==ChannelID) + { + if (Index) *Index=i; + return true; + } + } + return false; +} + +bool cGlobalInfosatepg::SetChannelUse(int Index,int Usage) +{ + if (numinfosatchannels==0) return false; + if ((Index<0) || (Index>numinfosatchannels-1)) return false; + bool ret=false; + if (infosatchannels[Index].Usage!=Usage) ret=true; + infosatchannels[Index].Usage=Usage; + return ret; +} + +int cGlobalInfosatepg::GetChannelUse(int Index) +{ + if (numinfosatchannels==0) return USE_NOTHING; + if ((Index<0) || (Index>numinfosatchannels-1)) return USE_NOTHING; + return infosatchannels[Index].Usage; +} + +int cGlobalInfosatepg::Save() +{ + char file[1024]; + snprintf(file,sizeof(file),"%s/infosatepg.dat",directory); + + int f=open(file,O_RDWR|O_CREAT|O_TRUNC,0664); + if (f==-1) + { + if (errno!=ENOSPC) + { + esyslog("infosatepg: unable to create file '%s'", file); + } + return -1; + } + + int ret; + for (int mac=EPG_FIRST_DAY_MAC; mac<=EPG_LAST_DAY_MAC; mac++) + { + ret=Infosatdata[mac].Save(f); + if (ret==-1) break; + } + + if (ret!=-1) + { + ret=write(f,&this_day,sizeof(this_day)); + if (ret!=sizeof(this_day)) ret=-1; + } + if (ret!=-1) + { + ret=write(f,&this_month,sizeof(this_month)); + if (ret!=sizeof(this_month)) ret=-1; + } + + close(f); + if (ret==-1) unlink(file); + return ret; +} + +int cGlobalInfosatepg::Load() +{ + char file[1024]; + snprintf(file,sizeof(file),"%s/infosatepg.dat",directory); + + int f=open(file,O_RDONLY); + if (f==-1) return 0; // it's ok if the file doesn't exist + + int ret; + for (int mac=EPG_FIRST_DAY_MAC; mac<=EPG_LAST_DAY_MAC; mac++) + { + ret=Infosatdata[mac].Load(f); + if (ret==-1) break; + } + if (ret!=-1) + { + ret=read(f,&this_day,sizeof(this_day)); + if (ret!=sizeof(this_day)) ret=-1; + } + if (ret!=-1) + { + ret=read(f,&this_month,sizeof(this_month)); + if (ret!=sizeof(this_month)) ret=-1; + } + + close(f); + if (ret==-1) + { + unlink(file); // but it's not ok if the read failed! + for (int mac=EPG_FIRST_DAY_MAC; mac<=EPG_LAST_DAY_MAC; mac++) + { + Infosatdata[mac].Init(NULL,0,0,0); + } + + } + return ret; +} + +void cGlobalInfosatepg::ResetProcessedFlags(void) +{ + dsyslog("infosatepg: reprocess files (later)"); + for (int mac=EPG_FIRST_DAY_MAC; mac<=EPG_LAST_DAY_MAC; mac++) + { + Infosatdata[mac].ResetProcessed(); + } +} + +void cGlobalInfosatepg::Lock(time_t Now) +{ + struct tm tm; + if (!localtime_r(&Now,&tm)) + { + this_day=-1; + this_month=-1; + } + else + { + if ((tm.tm_mday!=this_day) || ((tm.tm_mon+1)!=this_month)) + isyslog("infosatepg: all data processed"); + this_day=tm.tm_mday; + this_month=tm.tm_mon+1; + } +} + +bool cGlobalInfosatepg::isLocked(int *Day, int *Month) +{ + if (Day) *Day=this_day; + if (Month) *Month=this_month; + time_t Now=time(NULL); + struct tm tm; + if (!localtime_r(&Now,&tm)) return false; + if ((tm.tm_mday==this_day) && ((tm.tm_mon+1)==this_month)) return true; + return false; +} |