forked from dosgo/ngrok-c
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsendmsg.h
164 lines (137 loc) · 3.71 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
#ifndef __SENDMSG_H__
#define __SENDMSG_H__
#include "config.h"
#include "nonblocking.h"
#include <string>
#include <map>
#if OPENSSL
#include "openssl/ssl.h"
typedef SSL ssl_context;
#else
#if ISMBEDTLS
#include <mbedtls/ssl.h>
typedef mbedtls_ssl_context ssl_context;
#else
#include "polarssl/ssl.h"
#endif // ISMBEDTLS
#endif
#if WIN32
#include <windows.h>
#else
#include <errno.h>
#include <netinet/in.h>
typedef long long __int64;
#include <string.h>
#ifndef __UCLIBC__
#include <sys/select.h>
#endif
#endif
#include "bytestool.h"
using namespace std;
struct TunnelInfo
{
char localhost[255];
char subdomain[255];
char hostname[255];
char httpauth[255];
int localport;
struct sockaddr_in local_addr;
int remoteport;
};
inline int get_curr_unixtime()
{
time_t now;
return time(&now);
}
int loadargs( int argc, char **argv ,map<string, TunnelInfo*>*tunnellist,char *s_name,int * s_port,char * authtoken);
inline int strpos( char *str, char c )
{
char *sc = strchr( str, c );
if ( sc == NULL )
return(-1);
return(sc - str);
}
inline int GetProtocol(char *url,char *Protocol)
{
int plen= strpos( url, ':' );
if(plen>0)
{
memcpy( Protocol, url, plen );
return 0;
}
return -1;
}
int getlocaladdr( map<string,TunnelInfo *> *tunnellist,char *url, struct sockaddr_in* local_addr );
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 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
int len=SSL_write( ssl, buffer, 8+strlen(msgstr));
#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 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\": \"\",\"OS\":\"darwin\",\"Arch\":\"amd64\",\"ClientId\":\"%s\"}}",user.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,const char *protocol,const char *HostName,const char * Subdomain,int RemotePort);
//#endif
__int64 ntoh64(__int64 val );
__int64 hton64(__int64 val );
#endif