tftpd.c

   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 #include "tftpd.h"
  22
  23 Tftpd::Tftpd()
  24 {
  25   log = Log::getInstance();
  26   baseDir = NULL;
  27 }
  28
  29 Tftpd::~Tftpd()
  30 {
  31   shutdown();
  32 }
  33
  34 int Tftpd::shutdown()
  35 {
  36   if (threadIsActive()) threadCancel();
  37   ds.shutdown();
  38
  39   if (baseDir) delete[] baseDir;
  40   baseDir = NULL;
  41
  42   return 1;
  43 }
  44
  45 int Tftpd::run(char* tbaseDir)
  46 {
  47   if (threadIsActive()) return 1;
  48   log->log("Tftpd", Log::DEBUG, "Starting TFTPd");
  49
  50   if (!ds.init(16869))
  51   {
  52     log->log("Tftpd", Log::DEBUG, "DSock init error");
  53     shutdown();
  54     return 0;
  55   }
  56
  57   baseDir = new char[strlen(tbaseDir) + 1];
  58   strcpy(baseDir, tbaseDir);
  59
  60   if (!threadStart())
  61   {
  62     log->log("Tftpd", Log::DEBUG, "Thread start error");
  63     shutdown();
  64     return 0;
  65   }
  66
  67   log->log("Tftpd", Log::DEBUG, "TFTP server started with base path '%s'", baseDir);
  68   return 1;
  69 }
  70
  71 void Tftpd::threadMethod()
  72 {
  73   int retval;
  74   while(1)
  75   {
  76     log->log("Tftpd", Log::DEBUG, "Starting wait");
  77     retval = ds.waitforMessage(0);
  78     log->log("Tftpd", Log::DEBUG, "Wait finished");
  79
  80     if (retval == 0)
  81     {
  82       log->log("Tftpd", Log::CRIT, "Wait for packet error");
  83       return;
  84     }
  85     else if (retval == 1)
  86     {
  87       continue;
  88     }
  89     else
  90     {
  91       TftpClient* t = new TftpClient();
  92       t->run(baseDir, ds.getFromIPA(), ds.getFromPort(), (UCHAR*)ds.getData(), ds.getDataLength());
  93     }
  94   }
  95 }
  96