-
Notifications
You must be signed in to change notification settings - Fork 2
/
sendmsg.h
207 lines (180 loc) · 4.88 KB
/
sendmsg.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
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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
#ifndef __SENDMSG_H__
#define __SENDMSG_H__
#include "config.h"
#include "nonblocking.h"
#include <string>
#include <map>
#include "sslbio.h"
#if WIN32
#include <windows.h>
#else
#include <errno.h>
#include <netinet/in.h>
//typedef unsigned long long __int64;
#include <string.h>
#ifndef __UCLIBC__
#include <sys/select.h>
#endif
#endif
#include "bytestool.h"
using namespace std;
inline int get_curr_unixtime()
{
time_t now;
return time(&now);
}
int loadargs( int argc, char **argv ,char *s_name,int * s_port,char * authtoken,char *password_c,string *ClientId);
inline int strpos( char *str, char c )
{
char *sc = strchr( str, c );
if ( sc == NULL )
return(-1);
return(sc - str);
}
inline void str_replace(char * cp, int n, char * str)
{
int lenofstr;
char * tmp;
lenofstr = strlen(str);
//str3比str2短,往前移动
if(lenofstr < n)
{
tmp = cp+n;
while(*tmp)
{
*(tmp-(n-lenofstr)) = *tmp; //n-lenofstr是移动的距离
tmp++;
}
*(tmp-(n-lenofstr)) = *tmp; //move '\0'
}
else
//str3比str2长,往后移动
if(lenofstr > n)
{
tmp = cp;
while(*tmp) tmp++;
while(tmp>=cp+n)
{
*(tmp+(lenofstr-n)) = *tmp;
tmp--;
}
}
strncpy(cp,str,lenofstr);
}
inline int getvalue(char * str,const char *key,char * value){
int ypos=0;
if ( strncmp(str,key,strlen(key)) == 0 )
{
ypos = strpos( str, ':' );
if ( ypos != -1 )
{
memcpy(value, str + ypos + 1, strlen( str + ypos ));
return 0;
}
}
return -1;
}
inline int sendremote(int sock,ssl_context *ssl,const char *buf,int buflen,int isblock){
int sendlen=0;
if(isblock)
{
setnonblocking(sock,0);
}
#if OPENSSL
#if OPENSSLDL
sendlen=SslWrite(ssl, buf, buflen );
#else
sendlen=SSL_write(ssl, buf, buflen );
#endif // OPENSSLDL
#else
#if ISMBEDTLS
sendlen=mbedtls_ssl_write( ssl, (unsigned char *)buf, buflen );
#else
sendlen=ssl_write( ssl,(unsigned char *) buf, buflen );
#endif // ISMBEDTLS
#endif
if(isblock)
{
setnonblocking(sock,1);
}
return sendlen;
}
inline int sendlocal(int sock,const char *buf,int buflen,int isblock){
int sendlen=0;
if(isblock)
{
setnonblocking(sock,0);
}
#if WIN32
sendlen=send( sock, (char *) buf, buflen, 0 );
#else
sendlen=send( sock, buf, buflen, 0 );
#endif
if(isblock)
{
setnonblocking(sock,1);
}
return sendlen;
}
inline int sendpack(int sock,ssl_context *ssl,const char *msgstr,int isblock)
{
unsigned char buffer[strlen(msgstr)+9];
memset(buffer,0,strlen(msgstr)+9);
#if WIN32
unsigned __int64 packlen;
#else
unsigned long long packlen;
#endif
packlen=strlen(msgstr);
packlen=LittleEndian_64(packlen);
memcpy(buffer,&packlen,8);
memcpy(buffer+8,msgstr,strlen(msgstr));
if(isblock)
{
setnonblocking(sock,0);
}
#if OPENSSL
#if OPENSSLDL
int len=SslWrite( ssl, buffer, 8+strlen(msgstr));
#else
int len=SSL_write( ssl, buffer, 8+strlen(msgstr));
#endif // OPENSSL
#else
#if ISMBEDTLS
int len=mbedtls_ssl_write(ssl, buffer, 8+strlen(msgstr));
#else
int len=ssl_write(ssl, buffer, 8+strlen(msgstr));
#endif // ISM
#endif // ISM
if(isblock)
{
setnonblocking(sock,1);
}
return len;
}
inline int SendAuth(int sock,ssl_context *ssl,string ClientId,string user,string password_c)
{
// string str="{\"Type\":\"Auth\",\"Payload\":{\"Version\":\"2\",\"MmVersion\":\"1.7\",\"User\":\""+user+"\",\"Password\": \"\",\"OS\":\"darwin\",\"Arch\":\"amd64\",\"ClientId\":\""+ClientId+"\"}}";
char str[255];
memset(str,0,255);
sprintf(str,"{\"Type\":\"Auth\",\"Payload\":{\"Version\":\"2\",\"MmVersion\":\"1.7\",\"User\":\"%s\",\"Password\": \"%s\",\"OS\":\"darwin\",\"Arch\":\"amd64\",\"ClientId\":\"%s\"}}",user.c_str(),password_c.c_str(),ClientId.c_str());
return sendpack(sock,ssl,str,1);
}
inline int SendRegProxy(int sock,ssl_context *ssl,string &ClientId)
{
char str[255];
memset(str,0,255);
sprintf(str,"{\"Type\":\"RegProxy\",\"Payload\":{\"ClientId\":\"%s\"}}",ClientId.c_str());
return sendpack(sock,ssl,str,1);
}
inline int SendPing(int sock,ssl_context *ssl)
{
return sendpack(sock,ssl,"{\"Type\":\"Ping\",\"Payload\":{}}",1);
}
inline int SendPong(int sock,ssl_context *ssl)
{
return sendpack(sock,ssl,"{\"Type\":\"Pong\",\"Payload\":{}}",1);
}
int SendReqTunnel(int sock,ssl_context *ssl,char *ReqId,const char *protocol,const char *HostName,const char * Subdomain,int RemotePort,char *authtoken);
//#endif
#endif