vdrresponsepacket.cc

   1 /*
   2     Copyright 2007 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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19 */
  20
  21 #include "vdrresponsepacket.h"
  22
  23 #include "vdr.h"
  24 #include "tcp.h"
  25
  26 VDR_ResponsePacket::VDR_ResponsePacket()
  27 {
  28   userDataLength = 0;
  29   packetPos = 0;
  30   userData = NULL;
  31   ownBlock = true;
  32
  33   channelID = 0;
  34
  35   requestID = 0;
  36   streamID = 0;
  37
  38   flag = 0;
  39 }
  40
  41 VDR_ResponsePacket::~VDR_ResponsePacket()
  42 {
  43   if (!ownBlock) return; // don't free if it's a getblock
  44
  45   if (userData) free(userData);
  46 }
  47
  48 void VDR_ResponsePacket::setResponse(ULONG trequestID, UCHAR* tuserData, ULONG tuserDataLength)
  49 {
  50   channelID = VDR::CHANNEL_REQUEST_RESPONSE;
  51   requestID = trequestID;
  52   userData = tuserData;
  53   userDataLength = tuserDataLength;
  54 }
  55
  56 void VDR_ResponsePacket::setStream(ULONG tstreamID, ULONG tflag, UCHAR* tuserData, ULONG tuserDataLength)
  57 {
  58   channelID = VDR::CHANNEL_STREAM;
  59   streamID = tstreamID;
  60   flag = tflag;
  61   userData = tuserData;
  62   userDataLength = tuserDataLength;
  63 }
  64
  65 bool VDR_ResponsePacket::end()
  66 {
  67   return (packetPos >= userDataLength);
  68 }
  69
  70 void VDR_ResponsePacket::dumpUD()
  71 {
  72   TCP::dump(userData, userDataLength);
  73 }
  74
  75 int VDR_ResponsePacket::serverError()
  76 {
  77   if ((packetPos == 0) && (userDataLength == 4) && !ntohl(*(ULONG*)userData)) return 1;
  78   else return 0;
  79 }
  80
  81 char* VDR_ResponsePacket::extractString()
  82 {
  83   if (serverError()) return NULL;
  84
  85   int length = strlen((char*)&userData[packetPos]);
  86   if ((packetPos + length) > userDataLength) return NULL;
  87   char* str = new char[length + 1];
  88   strcpy(str, (char*)&userData[packetPos]);
  89   packetPos += length + 1;
  90   return str;
  91 }
  92
  93 UCHAR VDR_ResponsePacket::extractUCHAR()
  94 {
  95   if ((packetPos + sizeof(UCHAR)) > userDataLength) return 0;
  96   UCHAR uc = userData[packetPos];
  97   packetPos += sizeof(UCHAR);
  98   return uc;
  99 }
 100
 101 ULONG VDR_ResponsePacket::extractULONG()
 102 {
 103   if ((packetPos + sizeof(ULONG)) > userDataLength) return 0;
 104   ULONG ul = userData[packetPos++]<<24;
 105   ul|= userData[packetPos++]<<16;
 106   ul|= userData[packetPos++]<<8;
 107   ul|= userData[packetPos++];
 108   //packetPos += sizeof(ULONG);
 109   return ul;
 110 }
 111
 112 ULLONG VDR_ResponsePacket::extractULLONG()
 113 {
 114   if ((packetPos + sizeof(ULLONG)) > userDataLength) return 0;
 115   ULLONG ull= ((ULLONG)userData[packetPos++])<<56;
 116   ull|= ((ULLONG)userData[packetPos++])<<48;
 117   ull|= ((ULLONG)userData[packetPos++])<<40;
 118   ull|= ((ULLONG)userData[packetPos++])<<32;
 119   ull|= ((ULLONG)userData[packetPos++])<<24;
 120   ull|= ((ULLONG)userData[packetPos++])<<16;
 121   ull|= ((ULLONG)userData[packetPos++])<<8;
 122   ull|= ((ULLONG)userData[packetPos++]);
 123   return ull;
 124 }
 125
 126 double VDR_ResponsePacket::extractdouble()
 127 {
 128   if ((packetPos + sizeof(ULLONG)) > userDataLength) return 0;
 129   ULLONG ull = extractULLONG();
 130   double d;
 131   memcpy(&d,&ull,sizeof(double));
 132   return d;
 133 }
 134
 135 long VDR_ResponsePacket::extractLONG()
 136 {
 137   if ((packetPos + sizeof(long)) > userDataLength) return 0;
 138   long l = userData[packetPos++]<<24;
 139   l|= userData[packetPos++]<<16;
 140   l|= userData[packetPos++]<<8;
 141   l|= userData[packetPos++];
 142   return l;
 143 }
 144
 145 UCHAR* VDR_ResponsePacket::getUserData()
 146 {
 147   ownBlock = false;
 148   return userData;
 149 }
 150