]> git.vomp.tv Git - vompserver.git/blob - mvpserver.c
TFTP server (not quite finished)
[vompserver.git] / mvpserver.c
1 /*
2     Copyright 2004-2005 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 #include "mvpserver.h"
22
23 MVPServer::MVPServer()
24 {
25 }
26
27 MVPServer::~MVPServer()
28 {
29   stop();
30 }
31
32 int MVPServer::stop()
33 {
34   if (threadIsActive()) threadCancel();
35   close(listeningSocket);
36
37   udpr.shutdown();
38   bootpd.shutdown();
39   tftpd.shutdown();
40
41   log.log("Main", Log::INFO, "Stopped main server thread");
42   log.shutdown();
43
44   config.shutdown();
45
46   return 1;
47 }
48
49 int MVPServer::run()
50 {
51   if (threadIsActive()) return 1;
52
53   // Start config
54   const char* configDir = cPlugin::ConfigDirectory();
55   if (!configDir)
56   {
57     dsyslog("VOMP: Could not get config dir from VDR");
58   }
59   else
60   {
61     char configFileName[PATH_MAX];
62     snprintf(configFileName, PATH_MAX, "%s/vomp.conf", configDir);
63     if (config.init(configFileName))
64     {
65       dsyslog("VOMP: Config file found");
66     }
67     else
68     {
69       dsyslog("VOMP: Config file not found");
70     }
71   }
72
73   // Start logging
74
75   char* cfgLogFilename = config.getValueString("General", "Log file");
76   if (cfgLogFilename)
77   {
78     log.init(Log::DEBUG, cfgLogFilename);
79     delete[] cfgLogFilename;
80     log.log("Main", Log::INFO, "Logging started");
81   }
82   else
83   {
84     dsyslog("VOMP: Logging disabled");
85   }
86
87   // Work out a name for this server
88
89   char* serverName;
90
91   // Try to get from vomp.conf
92   serverName = config.getValueString("General", "Server name");
93   if (!serverName) // If not, get the hostname
94   {
95     serverName = new char[1024];
96     if (gethostname(serverName, 1024)) // if not, just use "-"
97     {
98       strcpy(serverName, "-");
99     }
100   }
101
102   int udpSuccess = udpr.run(serverName);
103
104   delete[] serverName;
105
106   if (!udpSuccess)
107   {
108     log.log("Main", Log::CRIT, "Could not start UDP replier");
109     stop();
110     return 0;
111   }
112
113   // Start Bootpd
114   if (config.getValueString("General", "Bootp server"))
115   {
116     if (!bootpd.run())
117     {
118       log.log("Main", Log::CRIT, "Could not start Bootpd");
119       stop();
120       return 0;
121     }
122   }
123   else
124   {
125     log.log("Main", Log::INFO, "Not starting Bootpd");
126   }
127
128   // Start Tftpd
129   if (config.getValueString("General", "TFTP server"))
130   {
131     if (!tftpd.run())
132     {
133       log.log("Main", Log::CRIT, "Could not start TFTPd");
134       stop();
135       return 0;
136     }
137   }
138   else
139   {
140     log.log("Main", Log::INFO, "Not starting TFTPd");
141   }
142
143   // start thread here
144   if (!threadStart())
145   {
146     log.log("Main", Log::CRIT, "Could not start MVPServer thread");
147     stop();
148     return 0;
149   }
150
151   log.log("Main", Log::DEBUG, "MVPServer run success");
152   return 1;
153 }
154
155 void MVPServer::threadMethod()
156 {
157   // I want to die as soon as I am cancelled because I'll be in accept()
158   pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
159   pthread_setcanceltype(PTHREAD_CANCEL_ASYNCHRONOUS, NULL);
160
161   struct sockaddr_in address;
162   address.sin_family = AF_INET;
163   address.sin_port = htons(3024);
164   address.sin_addr.s_addr = INADDR_ANY;
165   socklen_t length = sizeof(address);
166
167   listeningSocket = socket(AF_INET, SOCK_STREAM, 0);
168   if (listeningSocket < 0)
169   {
170     log.log("MVPServer", Log::CRIT, "Could not get TCP socket in vompserver");
171     return;
172   }
173
174   int value=1;
175   setsockopt(listeningSocket,SOL_SOCKET,SO_REUSEADDR,&value,sizeof(value));
176
177   if (bind(listeningSocket,(struct sockaddr *)&address,sizeof(address)) < 0)
178   {
179     log.log("MVPServer", Log::CRIT, "Could not bind to socket in vompserver");
180     close(listeningSocket);
181     return;
182   }
183
184   listen(listeningSocket, 5);
185
186   int clientSocket;
187
188   while(1)
189   {
190     clientSocket = accept(listeningSocket,(struct sockaddr *)&address, &length);
191     MVPClient* m = new MVPClient(clientSocket);
192     m->run();
193   }
194 }