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