-
Notifications
You must be signed in to change notification settings - Fork 10
/
ProxyServer.h
executable file
·77 lines (62 loc) · 1.97 KB
/
ProxyServer.h
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
/**
tcp_proxy
ProxyServer.h
Copyright 2011 Ramsey Kant
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
#ifndef PROXYSERVER_H_
#define PROXYSERVER_H_
#include <iostream>
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <list>
#include <map>
#include "config.h"
#include "ByteBuffer.h"
#include "Client.h"
#include "ProxyClient.h"
#define SOCKET int
#define INVALID_SOCKET -1
using namespace std;
class ProxyServer {
private:
bool canRun;
SOCKET listenSocket; // Descriptor for the listening socket
map<SOCKET, Client*> *clientMap; // Maps a Socket descriptor to a pointer to a Client object (connected clients only)
struct sockaddr_in serverAddr; // Structure for the server address
fd_set fd_master; // Master file descriptor set (listening socket + client sockets)
fd_set fd_read; // FD set of sockets being read/operated on
int fdmax; // Max FD number (max sockets handle)
private:
bool initSocket(int port);
void closeSockets();
void acceptConnection();
void disconnectClient(Client*);
void handleClient(Client*);
void sendData(Client*, ByteBuffer*);
Client* getClient(SOCKET);
Client* getProxyClientOwner(SOCKET);
void handleData(Client*, ByteBuffer*);
public:
ProxyServer();
~ProxyServer();
void runServer();
void stopServer() {
canRun = false;
}
};
#endif