130 lines
4.1 KiB
C++
130 lines
4.1 KiB
C++
/*
|
|
* discover.h: SAT>IP plugin for the Video Disk Recorder
|
|
*
|
|
* See the README file for copyright information and how to reach the author.
|
|
*
|
|
*/
|
|
|
|
#ifndef __SATIP_DISCOVER_H
|
|
#define __SATIP_DISCOVER_H
|
|
|
|
#include <curl/curl.h>
|
|
#include <curl/easy.h>
|
|
|
|
#include <vdr/thread.h>
|
|
#include <vdr/tools.h>
|
|
|
|
#include "socket.h"
|
|
|
|
class cSatipServer : public cListObject {
|
|
private:
|
|
enum eSatipServer {
|
|
eSatipServerDVBS2 = 0,
|
|
eSatipServerDVBT,
|
|
eSatipServerDVBT2,
|
|
eSatipServerCount
|
|
};
|
|
cString addressM;
|
|
cString descriptionM;
|
|
cString modelM;
|
|
int modelCountM[eSatipServerCount];
|
|
int modelTypeM;
|
|
cTimeMs lastSeenM;
|
|
|
|
public:
|
|
enum eSatipModelType {
|
|
eSatipModelTypeDVBS2 = 0x01,
|
|
eSatipModelTypeDVBT = 0x02,
|
|
eSatipModelTypeDVBT2 = 0x04,
|
|
eSatipModelTypeMask = 0x0F
|
|
};
|
|
cSatipServer(const char *addressP, const char *descriptionP, const char *modelP) : addressM(addressP), descriptionM(descriptionP), modelM(modelP), modelTypeM(eSatipModelTypeMask), lastSeenM(0)
|
|
{
|
|
memset(modelCountM, 0, sizeof(modelCountM));
|
|
if (isempty(*modelM))
|
|
modelM = "DVBS-1,DVBT2-1";
|
|
char *s, *p = strdup(*modelM);
|
|
char *r = strtok_r(p, ",", &s);
|
|
while (r) {
|
|
if (strstr(r, "DVBS2")) {
|
|
modelTypeM |= cSatipServer::eSatipModelTypeDVBS2;
|
|
if (char *c = strstr(r, "-"))
|
|
modelCountM[eSatipServerDVBS2] = atoi(++c);
|
|
}
|
|
if (strstr(r, "DVBT2")) {
|
|
modelTypeM |= cSatipServer::eSatipModelTypeDVBT | cSatipServer::eSatipModelTypeDVBT2;
|
|
if (char *c = strstr(r, "-"))
|
|
modelCountM[eSatipServerDVBT2] = atoi(++c);
|
|
}
|
|
if (strstr(r, "DVBT")) {
|
|
modelTypeM |= cSatipServer::eSatipModelTypeDVBT;
|
|
if (char *c = strstr(r, "-"))
|
|
modelCountM[eSatipServerDVBT] = atoi(++c);
|
|
}
|
|
r = strtok_r(NULL, ",", &s);
|
|
}
|
|
free(p);
|
|
}
|
|
virtual int Compare(const cListObject &listObjectP) const { const cSatipServer *s = (const cSatipServer *)&listObjectP; return strcasecmp(*addressM, *s->addressM); }
|
|
const char *Description() { return *descriptionM; }
|
|
const char *Address() { return *addressM; }
|
|
const char *Model(void) { return modelM; }
|
|
int ModelType(void) { return modelTypeM; }
|
|
bool Match(int modelP) { return ((modelP & eSatipModelTypeMask) & modelTypeM); }
|
|
int Satellite() { return (modelTypeM & eSatipModelTypeDVBS2) ? modelCountM[eSatipServerDVBS2] : 0; }
|
|
int Terrestrial() { return (modelTypeM & eSatipModelTypeDVBT) ? modelCountM[eSatipServerDVBT] : 0; }
|
|
int Terrestrial2() { return (modelTypeM & eSatipModelTypeDVBT2) ? modelCountM[eSatipServerDVBT2] : 0; }
|
|
int LastSeen(void) { return lastSeenM.Elapsed(); }
|
|
void Update(void) { lastSeenM.Set(); }
|
|
};
|
|
|
|
class cSatipServers : public cList<cSatipServer> {
|
|
};
|
|
|
|
|
|
class cSatipDiscover : public cThread {
|
|
private:
|
|
enum {
|
|
eConnectTimeoutMs = 1500, // in milliseconds
|
|
eDiscoveryPort = 1900,
|
|
eProbeBufferSize = 1024, // in bytes
|
|
eProbeTimeoutMs = 2000, // in milliseconds
|
|
eProbeIntervalMs = 60000 // in milliseconds
|
|
};
|
|
static cSatipDiscover *instanceS;
|
|
static const char *bcastAddressS;
|
|
static const char *bcastMessageS;
|
|
static size_t WriteCallback(void *ptrP, size_t sizeP, size_t nmembP, void *dataP);
|
|
cMutex mutexM;
|
|
CURL *handleM;
|
|
cSatipSocket *socketM;
|
|
cCondWait sleepM;
|
|
cTimeMs probeIntervalM;
|
|
cSatipServers *serversM;
|
|
void Janitor(void);
|
|
void Probe(void);
|
|
void Read(void);
|
|
void AddServer(const char *addrP, const char *descP, const char *modelP);
|
|
// constructor
|
|
cSatipDiscover();
|
|
// to prevent copy constructor and assignment
|
|
cSatipDiscover(const cSatipDiscover&);
|
|
cSatipDiscover& operator=(const cSatipDiscover&);
|
|
|
|
protected:
|
|
virtual void Action(void);
|
|
|
|
public:
|
|
static cSatipDiscover *GetInstance(void);
|
|
static bool Initialize(void);
|
|
static void Destroy(void);
|
|
virtual ~cSatipDiscover();
|
|
bool IsValidServer(cSatipServer *serverP);
|
|
cSatipServer *GetServer(int modelP);
|
|
cSatipServers *GetServers(void);
|
|
cString GetServerList(void);
|
|
int NumProvidedSystems(void);
|
|
};
|
|
|
|
#endif // __SATIP_DISCOVER_H
|