]> git.vomp.tv Git - vompserver.git/blob - tftpclient.h
IPv6 support for discovery protocol
[vompserver.git] / tftpclient.h
1 /*
2     Copyright 2006 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 TFTPCLIENT_H
22 #define TFTPCLIENT_H
23
24 #include <stdio.h>
25 #include <signal.h>
26 #include <ctype.h>
27 #ifndef VOMPSTANDALONE
28 #include <vdr/plugin.h>
29 #endif
30
31 #include "defines.h"
32 #include "log.h"
33 #include "dsock.h"
34 #include "thread.h"
35
36 class TftpClient : public Thread
37 {
38   public:
39     TftpClient();
40     virtual ~TftpClient();
41
42     int run(char* baseDir, char* ip, USHORT port, UCHAR* data, int length);
43     int shutdown();
44
45   private:
46     Log* log;
47     DatagramSocket ds;
48     char* baseDir;
49     char peerIP[17];
50     USHORT peerPort;
51     UCHAR buffer[600];
52     int bufferLength;
53     time_t lastCom;
54     FILE* file;
55     int state;
56     // 0 = start
57     // 1 = awaiting ack
58     // 2 = transfer finished, awaiting last ack
59     USHORT blockNumber;
60
61     void threadMethod();
62     void threadPostStopCleanup();
63
64     int processMessage(UCHAR* data, int length);
65     int processReadRequest(UCHAR* data, int length);
66     int processAck(UCHAR* data, int length);
67
68     int openFile(char* filename);
69     int sendBlock();
70     void transmitBuffer();
71 };
72
73 #endif