]> git.vomp.tv Git - vompclient.git/blob - vdrrequestpacket.cc
Live TV updates
[vompclient.git] / vdrrequestpacket.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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19 */
20
21 #include <arpa/inet.h>
22 #include <stdlib.h>
23 #include <string.h>
24
25 #include "vdrrequestpacket.h"
26
27 #include "vdr.h"
28
29 ULONG VDR_RequestPacket::serialNumberCounter = 1;
30
31 VDR_RequestPacket::VDR_RequestPacket()
32 {
33   buffer = NULL;
34   bufSize = 0;
35   bufUsed = 0;
36   lengthSet = false;
37   serialNumber = 0;
38   
39   opcode = 0;
40 }
41
42 VDR_RequestPacket::~VDR_RequestPacket()
43 {
44   free(buffer);
45 }
46
47 bool VDR_RequestPacket::init(ULONG topcode, bool setUserDataLength, ULONG userDataLength)
48 {
49   if (buffer) return false;
50   
51   if (setUserDataLength)
52   {
53     bufSize = headerLength + userDataLength;
54     lengthSet = true;
55   }
56   else
57   {
58     bufSize = 512;
59     userDataLength = 0; // so the below will write a zero
60   }
61   
62   buffer = (UCHAR*)malloc(bufSize);
63   if (!buffer) return false;
64   
65   channel = VDR::CHANNEL_REQUEST_RESPONSE;
66   serialNumber = serialNumberCounter++;
67   opcode = topcode;
68   
69   *(ULONG*)&buffer[0] = htonl(channel);
70   *(ULONG*)&buffer[4] = htonl(serialNumber);
71   *(ULONG*)&buffer[8] = htonl(opcode);
72   *(ULONG*)&buffer[userDataLenPos] = htonl(userDataLength);
73   bufUsed = headerLength;
74
75   return true;
76 }
77
78 bool VDR_RequestPacket::copyin(const UCHAR* src, ULONG len)
79 {
80   if (!checkExtend(len)) return false;
81   memcpy(buffer + bufUsed, src, len);
82   bufUsed += len;
83   if (!lengthSet) *(ULONG*)&buffer[userDataLenPos] = htonl(bufUsed - headerLength);
84   return true;
85 }
86
87 bool VDR_RequestPacket::addString(const char* string)
88 {
89   ULONG len = strlen(string) + 1;
90   if (!checkExtend(len)) return false;
91   memcpy(buffer + bufUsed, string, len);
92   bufUsed += len;
93   if (!lengthSet) *(ULONG*)&buffer[userDataLenPos] = htonl(bufUsed - headerLength);
94   return true;
95 }
96
97 bool VDR_RequestPacket::addULONG(ULONG ul)
98 {
99   if (!checkExtend(sizeof(ULONG))) return false;
100   *(ULONG*)&buffer[bufUsed] = htonl(ul);
101   bufUsed += sizeof(ULONG);
102   if (!lengthSet) *(ULONG*)&buffer[userDataLenPos] = htonl(bufUsed - headerLength);
103   return true;
104 }   
105
106 bool VDR_RequestPacket::addULLONG(ULLONG ull)
107 {
108   if (!checkExtend(sizeof(ULLONG))) return false;
109   *(ULLONG*)&buffer[bufUsed] = htonll(ull);
110   bufUsed += sizeof(ULLONG);
111   if (!lengthSet) *(ULONG*)&buffer[userDataLenPos] = htonl(bufUsed - headerLength);
112   return true;
113 }
114
115 bool VDR_RequestPacket::checkExtend(ULONG by)
116 {
117   if (lengthSet) return true;
118   if ((bufUsed + by) < bufSize) return true;
119   UCHAR* newBuf = (UCHAR*)realloc(buffer, bufSize + 512);
120   if (!newBuf) return false;
121   buffer = newBuf;
122   bufSize += 512;
123   return true;
124 }
125