/
main.c
77 lines (60 loc) · 1.46 KB
/
main.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
#include <arpa/inet.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/types.h>
#include "session.h"
#include "parseconf.h"
#define BACKLOG 1
int startserver(char *addr, int port)
{
int listenfd = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in servaddr;
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(port);
if (addr != NULL)
{
if (inet_pton(AF_INET, addr, &servaddr.sin_addr) == 0)
perror("addr error");
}
else
servaddr.sin_addr.s_addr = htons(INADDR_ANY);
if (bind(listenfd, (struct sockaddr*)&servaddr, sizeof(servaddr)) == -1)
perror("bind");
if (listen(listenfd, BACKLOG) == -1)
perror("listen");
return listenfd;
}
int main(int argc, char *argv[])
{
pid_t pid;
int listenfd;
int connfd;
socklen_t len;
struct sockaddr_in cliaddr;
loadconf("miniftp.conf");
printf("server_addr: %s\n", server_addr);
printf("server_port: %d\n", server_port);
printf("server_max_connfd: %d\n", server_max_connfd);
if (geteuid() != 0)//need root privilege.
{
printf("please use root privilege.\n");
exit(0);
}
listenfd = startserver(NULL, 7777);
len = sizeof(cliaddr);
if ((connfd = accept(listenfd, (struct sockaddr*)&cliaddr, &len)) == -1)
perror("accept");
pid = fork();
if (pid == -1)
perror("fork");
else if (pid == 0)//child
{
startsession(connfd);
}
wait(NULL); //wait for child to terminate.
return 0;
}