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 = ntohl(*(ULONG*)&userData[packetPos]);
 105   packetPos += sizeof(ULONG);
 106   return ul;
 107 }
 108
 109 ULLONG VDR_ResponsePacket::extractULLONG()
 110 {
 111   if ((packetPos + sizeof(ULLONG)) > userDataLength) return 0;
 112   ULLONG ull = ntohll(*(ULLONG*)&userData[packetPos]);
 113   packetPos += sizeof(ULLONG);
 114   return ull;
 115 }
 116
 117 long VDR_ResponsePacket::extractLONG()
 118 {
 119   if ((packetPos + sizeof(long)) > userDataLength) return 0;
 120   long l = ntohl(*(long*)&userData[packetPos]);
 121   packetPos += sizeof(long);
 122   return l;
 123 }
 124
 125 UCHAR* VDR_ResponsePacket::getUserData()
 126 {
 127   ownBlock = false;
 128   return userData;
 129 }
 130