]> git.vomp.tv Git - vompserver.git/blob - tftpd.c
Merge branch 'master' of http://git.vomp.tv/vompserver
[vompserver.git] / 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