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