diff options
Diffstat (limited to 'svdrpclient.h')
| -rw-r--r-- | svdrpclient.h | 169 |
1 files changed, 169 insertions, 0 deletions
diff --git a/svdrpclient.h b/svdrpclient.h new file mode 100644 index 0000000..10bdbae --- /dev/null +++ b/svdrpclient.h @@ -0,0 +1,169 @@ +/* +Copyright (C) 2004-2007 Christian Wieninger + +This program is free software; you can redistribute it and/or +modify it under the terms of the GNU General Public License +as published by the Free Software Foundation; either version 2 +of the License, or (at your option) any later version. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License +along with this program; if not, write to the Free Software +Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +Or, point your browser to http://www.gnu.org/licenses/old-licenses/gpl-2.0.html + +The author can be reached at cwieninger@gmx.de + +The project's page is at http://winni.vdr-developer.org/epgsearch +*/ + +#ifndef __SVDRPCLIENT_H +#define __SVDRPCLIENT_H + +#include <sys/socket.h> +#include <netdb.h> +#include <arpa/inet.h> +#include "log.h" + +#define SVDRPCONNECT 220 +#define SVDRPDISCONNECT 221 +#define CMDSUCCESS 250 + +class cSVDRPClient { + private: + int sock; +public: + bool bConnected; + + cSVDRPClient(int Port) + { + bConnected = false; + sock = socket(PF_INET, SOCK_STREAM, 0); + if (sock < 0) + { + LogFile.eSysLog("error creating socket!"); + return; + } + + struct sockaddr_in AdrSock; + AdrSock.sin_family = AF_INET; + AdrSock.sin_port = htons(EPGSearchConfig.SVDRPPort); + inet_aton("127.0.0.1", &AdrSock.sin_addr); + + if (connect(sock, (struct sockaddr*)&AdrSock, sizeof(AdrSock)) == -1) + { + LogFile.eSysLog("error connecting to socket!"); + return; + } + bConnected = (Receive() == SVDRPCONNECT); + if (!bConnected) + LogFile.eSysLog("EPGSearch: could not connect to VDR!"); + } + + long getAddrFromString(char* hostnameOrIp, struct sockaddr_in* addr) + { + unsigned long ip; + + struct hostent * he; + + if(hostnameOrIp==NULL || addr==NULL) + return -1; + + ip=inet_addr(hostnameOrIp); + + if(ip!=INADDR_NONE) + { + addr->sin_addr.s_addr=ip; + return 0; + } + else + { + he=gethostbyname(hostnameOrIp); + if(he==NULL) + return -1; + else + memcpy(&(addr->sin_addr),he->h_addr_list[0],4); + return 0; + } + } + + bool SendCmd(char* cmd) + { + if (!bConnected) + return false; + + char* szCmd = NULL; + + asprintf(&szCmd, "%s\r\n", cmd); + Send(szCmd); + free(szCmd); + bool cmdret = (Receive() == CMDSUCCESS); + + asprintf(&szCmd, "QUIT\r\n"); + Send(szCmd); + free(szCmd); + long rc = 0; + if ((rc = Receive()) != SVDRPDISCONNECT) + LogFile.eSysLog("could not disconnect (%ld)!", rc); + + close(sock); + return cmdret; + } + bool Send(char* szSend) + { + int length = strlen(szSend); + int sent = 0; + do + { + sent += send(sock, szSend + sent, length - sent, 0); + if( sent < 0 ) + { + LogFile.eSysLog("error sending command!"); + return false; + } + } + while ( sent < length ); + return true; + } + long Receive() + { + char* csResp = strdup(""); + char ch; + long rc = 0; + + bool bCheckMultiLine = true; + + while(bCheckMultiLine) + { + while( strlen(csResp) < 2 || strcmp(csResp + strlen(csResp) - 2, "\r\n") != 0) + { + if (recv(sock, &ch, 1, 0 ) < 0) + { + LogFile.eSysLog("EPGSearch: error receiving response!"); + return -1; + } + char* Temp = NULL; + asprintf(&Temp, "%s%c", csResp, ch); + free(csResp); + csResp = Temp; + } + if( csResp[3] == ' ' ) + { + bCheckMultiLine = false; + rc = atol(csResp); + } + free(csResp); + csResp = strdup(""); + } + free(csResp); + return rc; + } + + static const char *SVDRPSendCmd; +}; + +#endif |