]> git.vomp.tv Git - vompserver.git/blob - tftpclient.h
Finished Bootp/TFTP server
[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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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 #include <vdr/plugin.h>
28
29 #include "defines.h"
30 #include "log.h"
31 #include "dsock.h"
32 #include "thread.h"
33
34 class TftpClient : public Thread
35 {
36   public:
37     TftpClient();
38     virtual ~TftpClient();
39
40     int run(char* baseDir, char* ip, USHORT port, UCHAR* data, int length);
41     int shutdown();
42
43   private:
44     Log* log;
45     DatagramSocket ds;
46     char* baseDir;
47     char peerIP[17];
48     USHORT peerPort;
49     UCHAR buffer[600];
50     int bufferLength;
51     time_t lastCom;
52     FILE* file;
53     int state;
54     // 0 = start
55     // 1 = awaiting ack
56     // 2 = transfer finished, awaiting last ack
57     USHORT blockNumber;
58
59     void threadMethod();
60     void threadPostStopCleanup();
61
62     int processMessage(UCHAR* data, int length);
63     int processReadRequest(UCHAR* data, int length);
64     int processAck(UCHAR* data, int length);
65
66     int openFile(char* filename);
67     int sendBlock();
68     void transmitBuffer();
69 };
70
71 #endif