]> git.vomp.tv Git - vompclient.git/blob - vdrrequestpacket.cc
Fix segfault in boxstack.cc
[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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
19 */
20
21 #ifndef WIN32
22 #include <arpa/inet.h>
23 #else
24
25 #endif
26 #include <stdlib.h>
27 #include <string.h>
28
29 #include "vdrrequestpacket.h"
30
31 #include "vdr.h"
32
33 ULONG VDR_RequestPacket::serialNumberCounter = 1;
34
35 VDR_RequestPacket::VDR_RequestPacket()
36 {
37   buffer = NULL;
38   bufSize = 0;
39   bufUsed = 0;
40   lengthSet = false;
41   serialNumber = 0;
42   
43   opcode = 0;
44 }
45
46 VDR_RequestPacket::~VDR_RequestPacket()
47 {
48   free(buffer);
49 }
50
51 bool VDR_RequestPacket::init(ULONG topcode, bool setUserDataLength, ULONG userDataLength)
52 {
53   if (buffer) return false;
54   
55   if (setUserDataLength)
56   {
57     bufSize = headerLength + userDataLength;
58     lengthSet = true;
59   }
60   else
61   {
62     bufSize = 512;
63     userDataLength = 0; // so the below will write a zero
64   }
65   
66   buffer = (UCHAR*)malloc(bufSize);
67   if (!buffer) return false;
68   
69   channel = VDR::CHANNEL_REQUEST_RESPONSE;
70   serialNumber = serialNumberCounter++;
71   opcode = topcode;
72   
73   int pos=0;
74   buffer[pos++]=(channel>>24)&0xff;
75   buffer[pos++]=(channel>>16)&0xff;
76   buffer[pos++]=(channel>>8)&0xff;
77   buffer[pos++]=channel &0xff;
78   buffer[pos++]=(serialNumber>>24)&0xff;
79   buffer[pos++]=(serialNumber>>16)&0xff;
80   buffer[pos++]=(serialNumber>>8)&0xff;
81   buffer[pos++]=serialNumber &0xff;
82   buffer[pos++]=(opcode>>24)&0xff;
83   buffer[pos++]=(opcode>>16)&0xff;
84   buffer[pos++]=(opcode>>8)&0xff;
85   buffer[pos++]=opcode &0xff;
86   buffer[pos++]=(userDataLength>>24)&0xff;
87   buffer[pos++]=(userDataLength>>16)&0xff;
88   buffer[pos++]=(userDataLength>>8)&0xff;
89   buffer[pos++]=userDataLength &0xff;
90
91   /**(ULONG*)&buffer[4] = htonl(serialNumber);
92   *(ULONG*)&buffer[8] = htonl(opcode);
93   *(ULONG*)&buffer[userDataLenPos] = htonl(userDataLength);*/
94   bufUsed = headerLength;
95
96   return true;
97 }
98
99 bool VDR_RequestPacket::copyin(const UCHAR* src, ULONG len)
100 {
101   if (!checkExtend(len)) return false;
102   memcpy(buffer + bufUsed, src, len);
103   bufUsed += len;
104   if (!lengthSet) {
105           int pos=userDataLenPos;
106           ULONG tocopy=bufUsed - headerLength;
107           buffer[pos++]=(tocopy>>24)&0xff;
108           buffer[pos++]=(tocopy>>16)&0xff;
109           buffer[pos++]=(tocopy>>8)&0xff;
110           buffer[pos++]=tocopy &0xff;
111   }
112   return true;
113 }
114
115
116
117 bool VDR_RequestPacket::addString(const char* string)
118 {
119   ULONG len = strlen(string) + 1;
120   if (!checkExtend(len)) return false;
121   memcpy(buffer + bufUsed, string, len);
122   bufUsed += len;
123   if (!lengthSet) {
124           int pos=userDataLenPos;
125           ULONG tocopy=bufUsed - headerLength;
126           buffer[pos++]=(tocopy>>24)&0xff;
127           buffer[pos++]=(tocopy>>16)&0xff;
128           buffer[pos++]=(tocopy>>8)&0xff;
129           buffer[pos++]=tocopy &0xff;
130   }
131   return true;
132 }
133
134 bool VDR_RequestPacket::addULONG(ULONG ul)
135 {
136   if (!checkExtend(sizeof(ULONG))) return false;
137 //  *(ULONG*)&buffer[bufUsed] = htonl(ul);
138
139   buffer[bufUsed++]=(ul>>24)&0xff;
140   buffer[bufUsed++]=(ul>>16)&0xff;
141   buffer[bufUsed++]=(ul>>8)&0xff;
142   buffer[bufUsed++]=ul &0xff;
143
144   if (!lengthSet) {
145           int pos=userDataLenPos;
146           ULONG tocopy=bufUsed - headerLength;
147           buffer[pos++]=(tocopy>>24)&0xff;
148           buffer[pos++]=(tocopy>>16)&0xff;
149           buffer[pos++]=(tocopy>>8)&0xff;
150           buffer[pos++]=tocopy &0xff;
151   }
152   return true;
153 }   
154
155 bool VDR_RequestPacket::addULLONG(ULLONG ull)
156 {
157   if (!checkExtend(sizeof(ULLONG))) return false;
158   buffer[bufUsed++]=(ull>>56)&0xff;
159   buffer[bufUsed++]=(ull>>48)&0xff;
160   buffer[bufUsed++]=(ull>>40)&0xff;
161   buffer[bufUsed++]=(ull>>32)&0xff;
162   buffer[bufUsed++]=(ull>>24)&0xff;
163   buffer[bufUsed++]=(ull>>16)&0xff;
164   buffer[bufUsed++]=(ull>>8)&0xff;
165   buffer[bufUsed++]=ull &0xff;
166
167   if (!lengthSet) {
168           int pos=userDataLenPos;
169           ULONG tocopy=bufUsed - headerLength;
170           buffer[pos++]=(tocopy>>24)&0xff;
171           buffer[pos++]=(tocopy>>16)&0xff;
172           buffer[pos++]=(tocopy>>8)&0xff;
173           buffer[pos++]=tocopy &0xff;
174   }
175   return true;
176 }
177
178 bool VDR_RequestPacket::checkExtend(ULONG by)
179 {
180   if (lengthSet) return true;
181   if ((bufUsed + by) < bufSize) return true;
182   UCHAR* newBuf = (UCHAR*)realloc(buffer, bufSize + 512);
183   if (!newBuf) return false;
184   buffer = newBuf;
185   bufSize += 512;
186   return true;
187 }
188