source: TI05-delivery/trunk/src/bbftp-server-3.2.0/bbftpd/bbftpd_daemon.c @ 1448

Subversion URL: http://proj.badc.rl.ac.uk/svn/ndg/TI05-delivery/trunk/src/bbftp-server-3.2.0/bbftpd/bbftpd_daemon.c@1448
Revision 1448, 7.2 KB checked in by spascoe, 14 years ago (diff)

Implemented two hooks in the bbftpd source that allow
python code to be executed before and afer a fork(). This is needed
to reset python logging and will probably be usefull for other things
in the future.

Line 
1/*
2 * bbftpd/bbftpd_daemon.c
3 * Copyright (C) 1999, 2000, 2001, 2002 IN2P3, CNRS
4 * bbftp@in2p3.fr
5 * http://doc.in2p3.fr/bbftp
6 *
7 * This program is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version 2
10 * of the License, or any later version.
11 *
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 * GNU General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
20 */ 
21
22/****************************************************************************
23
24 
25 
26 bbftpd_daemon.c    v 0.0.0  1999/11/24
27                    v 1.3.0  2000/03/16
28                    v 1.6.1  2000/03/30 - Use sysconf instead of OPEN_MAX
29                    v 1.8.4  2000/04/21 - Seed the randoma generator
30                    v 1.8.7  2000/05/24 - Include version.h
31                    v 1.8.10 2000/08/11 - Portage to Linux
32                    v 1.9.3  2000/10/12 - Define the socket first to be able
33                                          to give information to the user
34                    v 1.9.4  2000/10/16 - Supress %m
35                    v 2.0.0  2000/12/13 - Supress be_daemon and controlsock
36                                          as global parameters
37                                        - Add parameters incontrolsock and
38                                          outcontrolsock for bbftp v2
39                    v 2.0.1  2001/04/23 - Correct indentation
40                    v 2.0.2  2001/05/04 - Correct include for RFIO
41                    v 2.1.0  2001/06/11 - Change file name
42
43*****************************************************************************/
44#include <bbftpd.h>
45
46#include <netdb.h>
47#include <netinet/in.h>
48#include <netinet/tcp.h>
49#include <stdio.h>
50#include <bbftpd_private_log.h>
51#include <bbftpd_private_fork.h>
52#include <sys/socket.h>
53#if TIME_WITH_SYS_TIME
54# include <sys/time.h>
55# include <time.h>
56#else
57# if HAVE_SYS_TIME_H
58#  include <sys/time.h>
59# else
60#  include <time.h>
61# endif
62#endif
63#include <unistd.h>
64#include <utime.h>
65#if HAVE_STRING_H
66# include <string.h>
67#endif
68
69#include <common.h>
70#include <daemon.h>
71#include <daemon_proto.h>
72#include <version.h>
73#include <openssl/err.h>
74#include <openssl/rand.h>
75/*
76** Common variables for BBFTP protocole version 1 and 2
77*/
78extern int  newcontrolport ;
79extern char daemonchar[50] ;
80/*
81** Variables for BBFTP protocole version 1
82*/
83extern int msgsock ;
84/*
85** Variables for BBFTP protocole version 2
86*/
87extern int incontrolsock ;
88extern int outcontrolsock ;
89
90#ifdef NDG_PYTHON_EMBED
91int do_daemon(int argc, char **argv, char **envp)
92#else
93void do_daemon(int argc,char **argv,char **envp)
94#endif
95{
96
97    int        prpg ;
98    int        i ;
99    int     retcode ;
100    int     controlsock ;
101    struct sockaddr_in server;
102    int     on = 1;
103    int     nfds ;
104    int     pid;
105   
106    char    buffrand[NBITSINKEY] ;
107    struct timeval tp ;
108    unsigned int seed ;
109   
110    controlsock = socket ( AF_INET, SOCK_STREAM, IPPROTO_TCP ) ;
111    if ( controlsock < 0 ) {
112        bbftpd_log(BBFTPD_ERR, "Cannot create socket to listen on: %s",strerror(errno));
113        fprintf(stderr,"Cannot create socket to listen on: %s\n",strerror(errno)) ;
114        exit(1) ;
115    }
116    if ( setsockopt(controlsock,SOL_SOCKET, SO_REUSEADDR,(char *)&on,sizeof(on)) < 0 ) {
117        bbftpd_log(BBFTPD_ERR,"Cannot set SO_REUSEADDR on control socket : %s",strerror(errno)) ;
118        fprintf(stderr,"Cannot set SO_REUSEADDR on control socket : %s\n ",strerror(errno)) ;
119        exit(1) ;
120    }
121    server.sin_family = AF_INET ;
122    server.sin_addr.s_addr = INADDR_ANY ;
123    server.sin_port = htons(newcontrolport);
124    if ( bind (controlsock,(struct sockaddr *) &server, sizeof(server) ) < 0 ) {
125        bbftpd_log(BBFTPD_ERR,"Error binding control socket : %s",strerror(errno)) ;
126        fprintf(stderr,"Error binding control socket : %s\n ",strerror(errno)) ;
127        exit(1) ;
128    }
129    if ( listen(controlsock,100) < 0 ) {
130        bbftpd_log(BBFTPD_ERR,"Error listening control socket : %s",strerror(errno)) ;
131        fprintf(stderr,"Error listening control socket : %s\n ",strerror(errno)) ;
132        exit(1) ;
133    }
134    /* Fork - so I'm not the owner of the process group any more */
135    retcode = bbftpd_fork();
136    if (retcode < 0) {
137        bbftpd_log(BBFTPD_ERR, "Cannot fork %s",strerror(errno));
138        exit(1);
139    }
140#ifdef NDG_PYTHON_EMBED
141    /* Return the daemon's PID */
142    if (retcode > 0) {
143      close(controlsock);
144      return retcode;
145    }
146#else
147    /* No need for the parent any more */
148    if (retcode > 0) _exit(0);
149#endif
150    prpg = 0 ;
151    prpg = setsid () ;     /* disassoiciate from control terminal */
152    if ( prpg < 0 ) {
153        bbftpd_log(BBFTPD_ERR,"Cannot daemonise: %s",strerror(errno)) ;
154        exit(1) ;
155    }
156    /* Close off all file descriptors and reopen bbftpd_log */
157
158    nfds = sysconf(_SC_OPEN_MAX) ;
159
160    bbftpd_closelog();
161    for (i = 0; i <= nfds; i++) {
162        if ( i != controlsock) close(i);
163    }
164    bbftpd_openlog(daemonchar, LOG_PID | LOG_NDELAY, BBFTPD_FACILITY);
165    /* log PID in /var/run/bbftpd.pid */
166    {
167        FILE *f;
168        if ((f = fopen("/var/run/bbftpd.pid", "w")) != NULL) {
169            fprintf(f, "%d", getpid());
170            fclose(f);
171        }
172    }
173
174    /* junk stderr */
175    (void) freopen("/dev/null", "w", stderr);
176   
177    bbftpd_log(BBFTPD_DEBUG,"Starting bbftpd in background mode") ;
178    if ( bbftpd_blockallsignals() < 0 ) {
179        exit(1) ;
180    }
181    /*
182    ** Load the error message from the crypto lib
183    */
184    ERR_load_crypto_strings() ;
185    /*
186    ** Initialize the buffrand buffer which is giong to be used to initialize the
187    ** random generator
188    */
189    /*
190    ** Take the usec to initialize the random session
191    */
192    gettimeofday(&tp,NULL) ;
193    seed = tp.tv_usec ;
194    srandom(seed) ;
195    for (i=0; i < sizeof(buffrand) ; i++) {
196        buffrand[i] = random() ;
197    }
198    /*
199    ** Initialize the random generator
200    */
201    RAND_seed(buffrand,NBITSINKEY) ;
202    while (1) {
203
204        msgsock = accept(controlsock, 0, 0);
205        if (msgsock < 0) {
206             bbftpd_log(BBFTPD_ERR, "Accept failed: %s",strerror(errno));
207            sleep(1);
208            continue;
209        }
210            /* Fork off a handler */
211        pid = bbftpd_fork();
212        if (pid < 0) {
213            bbftpd_log(BBFTPD_ERR, "failed to fork: %s",strerror(errno));
214             sleep(1);
215            continue;
216        }
217        if (pid == 0) {
218            /* I am that forked off child */
219            bbftpd_closelog();
220            /* Make sure that stdin/stdout are the new socket */
221            /* Only parent needs controlsock */
222            if (controlsock != 0 && controlsock != 1)
223                close(controlsock);
224                bbftpd_openlog(daemonchar, LOG_PID | LOG_NDELAY, BBFTPD_FACILITY);
225                incontrolsock = msgsock ;
226                outcontrolsock = msgsock ;
227#ifdef NDG_PYTHON_EMBED
228                return 0;
229#else
230                return;
231#endif
232           
233                }
234
235            /* I am the parent */
236        close(msgsock);
237   
238    }
239}
240
Note: See TracBrowser for help on using the repository browser.