-
Notifications
You must be signed in to change notification settings - Fork 1
/
tcp_proxy_stub.h
84 lines (71 loc) · 2.33 KB
/
tcp_proxy_stub.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
78
79
80
81
82
83
84
/*
* Copyright 2010-2020 DatasysLab@iit.edu(http://datasys.cs.iit.edu/index.html)
* Director: Ioan Raicu(iraicu@cs.iit.edu)
*
* 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.
*
* This file is part of ZHT library(http://datasys.cs.iit.edu/projects/ZHT/index.html).
* Tonglin Li(tli13@hawk.iit.edu) with nickname Tony,
* Xiaobing Zhou(xzhou40@hawk.iit.edu) with nickname Xiaobingo,
* Ke Wang(kwang22@hawk.iit.edu) with nickname KWang,
* Dongfang Zhao(dzhao8@@hawk.iit.edu) with nickname DZhao,
* Ioan Raicu(iraicu@cs.iit.edu).
*
* tcp_proxy_stub.h
*
* Created on: Jun 21, 2013
* Author: Xiaobingo
* Contributor: Tony, KWang, DZhao
*/
#ifndef TCP_PROXY_STUB_H_
#define TCP_PROXY_STUB_H_
#include "ip_proxy_stub.h"
#include "HTWorker.h"
#include <pthread.h>
#include <map>
using namespace std;
/*
*
*/
class TCPProxy: public IPProtoProxy {
public:
typedef map<string, int> MAP;
typedef MAP::iterator MIT;
public:
TCPProxy();
virtual ~TCPProxy();
virtual bool sendrecv(const void *sendbuf, const size_t sendcount,
void *recvbuf, size_t &recvcount);
virtual bool teardown();
//virtual bool send_to_dest(const void *sendbuf, string host, int port);//Tony added for ZHT-H
protected:
virtual int getSockCached(const string& host, const uint& port);
virtual int makeClientSocket(const string& host, const uint& port);
virtual int recvFrom(int sock, void* recvbuf);
virtual int loopedrecv(int sock, string &srecv);
private:
int sendTo(int sock, const void* sendbuf, int sendcount);
private:
//static MAP CONN_CACHE;
MAP CONN_CACHE;
};
class TCPStub: public IPProtoStub {
public:
TCPStub();
virtual ~TCPStub();
virtual bool recvsend(ProtoAddr addr, const void *recvbuf);
public:
virtual int sendBack(ProtoAddr addr, const void* sendbuf,
int sendcount) const;
};
#endif /* TCP_PROXY_STUB_H_ */