ModFTP/FTP_Server.h at master · MidCheck/ModFTP · GitHub

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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*************************************************************************
    > File Name: FTP_Server.h
    > Author: MidCHeck
    > Mail: midcheck@foxmail.com
    > Created Time: 2019年06月08日 星期六 02时25分09秒
 ************************************************************************/
#ifndef MIDCHECK_FTP_SERVER_H
#define MIDCHECK_FTP_SERVER_H

#include "server.h"
#include "FTP_Command.h"

#include<thread>
#include<string.h>

#include<vector>

#ifdef __linux__
#include<sys/epoll.h>
#define MAX_EVENT_NUMBER 1024
#endif

namespace MidCHeck{
#ifdef __linux__
inline void addfd(int epollfd, int fd, bool oneshot){
	epoll_event event;
	event.data.fd = fd;
	event.events = EPOLLIN | EPOLLET;
	if(oneshot){
		event.events |= EPOLLONESHOT;
	}
	epoll_ctl(epollfd, EPOLL_CTL_ADD, fd, &event);
	setnonblocking(fd);
}
inline void reset_oneshot(int epollfd, int fd){
	epoll_event event;
	event.data.fd = fd;
	event.events = EPOLLIN | EPOLLET | EPOLLONESHOT;
	epoll_ctl(epollfd, EPOLL_CTL_MOD, fd, &event);
}
#endif
/*网络传输层*/
class Worker{
private:
	int _epollfd;
	int _sockfd;
	MidCHeck::User *&_user;  // 指向当前sockfd的缓冲区
	COMMAND parse(char*, int&);
public:
	Worker(int efd, int sfd):_epollfd(efd), _sockfd(sfd),
		_user(Shardata::GetEntity()->users[sfd])
	{
		// 保证每一个命令都有数据空间
		// 注意到在quit退出或套接字关闭时释放空间
		// 需要加锁，但也可不加
		if(_user == nullptr) {
			Debug("_user == nullptr, 创建新用户");
			_user = new User;
			_user->sockfd = _sockfd;
		}
	}
	~Worker(){
		Debug("work 析够");
	}
	void operator()(){
		char* buf = _user->buffer;
		// 如果命令后的参数太长，则可能出bug
		while(1){
			// 非第一次接受数据
			// 可能不需要清空内存，因为解析时以结束符为标志
			memset(buf, '\0', _user->rw_cur);
			int ret = recv(_sockfd, buf, 128, 0);
			Debug("recv _sockfd: %d, ret: %d, buf: %s", _sockfd, ret, buf);
			if(ret == 0){
				close(_sockfd);
				Debug("关闭连接套接字close _sockfd: %d", _sockfd);
				//delete d
				break;
			}else if(ret < 0){
				if(errno == EAGAIN){
					reset_oneshot(_epollfd, _sockfd);
					// read later
					break;
				}
			}else{
				// 解析命令
				COMMAND recv_cmd = parse(buf, _user->rw_cur);
				Command* cmd = CommandFactory(_user).CreateCmd(recv_cmd);
				cmd->process();
				//　阻塞与非阻塞可能有bug
				if(recv_cmd != QUIT)
					send(_sockfd, buf, _user->rw_cur, 0);
			}
		}
	}
};
class FTP_Server: public Server{
private:
	epoll_event events[MAX_EVENT_NUMBER];
	std::vector<Socket> clients;
	int epollfd;
public:
	FTP_Server(const char* ip, int port):Server(ip, port){
		if((epollfd = epoll_create(5)) == -1) mcthrow("Create epoll Failed!");
		addfd(epollfd, this->sock, false);
	}
	~FTP_Server(){ this->~Server(); }
	void start();

	#ifdef __linux__
	friend void addfd(int, int, bool);
	friend void reset_oneshot(int, int);
	#endif
};
} // 命名空间MidCHeck结束

#endif