vdr.h

   1 /*
   2     Copyright 2004-2005 Chris Tallon
   3
   4     This file is part of VOMP.
   5
   6     VOMP is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     VOMP is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with VOMP; if not, write to the Free Software
  18     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 #ifndef VDR_H
  22 #define VDR_H
  23
  24 #include <stdio.h>
  25 #include <time.h>
  26 #include <pthread.h>
  27 #include <vector>
  28 #include <algorithm>
  29
  30 #include "defines.h"
  31 #include "log.h"
  32 #include "dsock.h"
  33 #include "tcp.h"
  34 #include "directory.h"
  35 #include "recording.h"
  36 #include "channel.h"
  37 #include "event.h"
  38 #include "rectimer.h"
  39
  40 using namespace std;
  41
  42  // FIXME do some tcp connection error checking and kill it!
  43
  44 typedef vector<Event*> EventList;
  45 typedef vector<Channel*> ChannelList;
  46 typedef vector<RecTimer*> RecTimerList;
  47
  48 struct VDRServer
  49 {
  50   char* ip;
  51   char* name;
  52 };
  53
  54 struct RecTimerSorter     // : public binary_function<double, double, bool>
  55 {
  56   bool operator() (const RecTimer* a, const RecTimer* b)
  57   {
  58     return a->startTime < b->startTime;
  59   }
  60 };
  61
  62 struct ServerSorter
  63 {
  64   bool operator() (const VDRServer& a, const VDRServer& b)
  65   {
  66     if (strcmp(b.name, a.name) > 0) return true;
  67     return false;
  68   }
  69 };
  70
  71 struct RecordingSorter
  72 {
  73   bool operator() (const Recording* a, const Recording* b)
  74   {
  75     int c = strcmp(b->getProgName(), a->getProgName());
  76     if (c > 0) return true;
  77     if (c < 0) return false;
  78
  79     return a->start < b->start;
  80   }
  81 };
  82
  83 struct DirectorySorter
  84 {
  85   bool operator() (const Directory* a, const Directory* b)
  86   {
  87     int c = strcmp(b->name, a->name);
  88     if (c > 0) return true;
  89     return false;
  90   }
  91 };
  92
  93 class VDR
  94 {
  95
  96   public:
  97     VDR();
  98     ~VDR();
  99     static VDR* getInstance();
 100
 101     int init(int port);
 102     int shutdown();
 103
 104     void findServers(vector<VDRServer>& servers);
 105     void cancelFindingServer();
 106     void setServerIP(char*);
 107     int connect();
 108     void disconnect();
 109     ULLONG getResumePoint(char* fileName);  // uses configLoad
 110
 111     void setReceiveWindow(size_t size);
 112
 113     // protocol functions
 114
 115     int doLogin();
 116
 117     Directory* getRecordingsList();
 118     char*      getRecordingSummary(char* fileName);
 119     int        deleteRecording(char* fileName);
 120     ULLONG     streamRecording(Recording* rec);
 121     ULLONG     rescanRecording();
 122
 123     ChannelList* getChannelsList(ULONG type);
 124     int          streamChannel(ULONG number);
 125
 126     UCHAR*     getBlock(ULLONG position, UINT maxAmount, UINT* amountReceived);
 127     int        stopStreaming();
 128     EventList* getChannelSchedule(ULONG number);
 129     EventList* getChannelSchedule(ULONG number, time_t start, ULONG duration);
 130     int        configSave(char* section, char* key, const char* value);
 131     char*      configLoad(char* section, char* key);
 132     ULONG      setEventTimer(char* timerString);
 133
 134     RecTimerList* getRecTimersList();
 135
 136     // end
 137
 138     const static ULONG VIDEO = 1;
 139     const static ULONG RADIO = 2;
 140
 141   private:
 142     static VDR* instance;
 143     Log* logger;
 144     int initted;
 145     int findingServer;
 146     TCP* tcp;
 147     int port;
 148     char serverIP[16];
 149     bool connected;
 150     pthread_mutex_t mutex;
 151
 152     UCHAR* packet;
 153     ULONG packetLength;
 154     ULONG packetPos;
 155
 156     const static ULONG VDR_LOGIN               = 1;
 157     const static ULONG VDR_GETRECORDINGLIST    = 2;
 158     const static ULONG VDR_DELETERECORDING     = 3;
 159     const static ULONG VDR_GETSUMMARY          = 4;
 160     const static ULONG VDR_GETCHANNELLIST      = 5;
 161     const static ULONG VDR_STREAMCHANNEL       = 6;
 162     const static ULONG VDR_GETBLOCK            = 7;
 163     const static ULONG VDR_STOPSTREAMING       = 8;
 164     const static ULONG VDR_STREAMRECORDING     = 9;
 165     const static ULONG VDR_GETCHANNELSCHEDULE  = 10;
 166     const static ULONG VDR_CONFIGSAVE          = 11;
 167     const static ULONG VDR_CONFIGLOAD          = 12;
 168     const static ULONG VDR_RESCANRECORDING     = 13;
 169     const static ULONG VDR_GETTIMERS           = 14;
 170     const static ULONG VDR_SETTIMER            = 15;
 171
 172     int  getPacket();
 173     void freePacket();
 174     int  serverError();
 175     char*  extractString();
 176     ULONG  extractULONG();
 177     ULLONG extractULLONG();
 178     long   extractLONG();
 179 };
 180
 181 #endif