-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsocks.h
81 lines (69 loc) · 1.96 KB
/
socks.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
#ifndef __SOCKS_H__
#include <stdint.h>
#include <sys/ioctl.h>
#define __SOCKS_H__
struct sock_name_param {
char name[64];
};
struct sock_buffer_param {
uint64_t size;
void *buffer;
};
#define IOCTL_SOCKS_INIT _IOWR('s', 1, uint64_t)
#define IOCTL_SOCKS_LISTEN _IOWR('s', 2, struct sock_name_param)
#define IOCTL_SOCKS_CONNECT _IOWR('s', 3, struct sock_name_param)
#define IOCTL_SOCKS_SEND _IOWR('s', 4, struct sock_buffer_param)
#define IOCTL_SOCKS_RECV _IOWR('s', 5, struct sock_buffer_param)
#define IOCTL_SOCKS_RESIZE _IOWR('s', 6, uint64_t)
int sock_init(int fd, size_t size) {
return ioctl(fd, IOCTL_SOCKS_INIT, size);
}
int sock_listen(int fd, char *name) {
struct sock_name_param param;
memset(¶m, 0, sizeof(param));
strncpy(param.name, name, sizeof(param.name) -1);
int err = ioctl(fd, IOCTL_SOCKS_LISTEN, ¶m);
if(err < 0){
perror("ioctl sock_listen");
fail();
}
return err;
}
int sock_connect(int fd, char *name) {
struct sock_name_param param;
memset(¶m, 0, sizeof(param));
strncpy(param.name, name, sizeof(param.name) -1);
int err = ioctl(fd, IOCTL_SOCKS_CONNECT, ¶m);
if(err < 0){
perror("ioctl sock_connect");
fail();
}
return err;
}
int sock_send(int fd, void *buffer, size_t size) {
struct sock_buffer_param param = { .size = size, .buffer = buffer};
int err = ioctl(fd, IOCTL_SOCKS_SEND, ¶m);
if(err < 0){
perror("ioctl sock_send");
fail();
}
return err;
}
int sock_recv(int fd, void *buffer, size_t size) {
struct sock_buffer_param param = { .size = size, .buffer = buffer};
int err = ioctl(fd, IOCTL_SOCKS_RECV, ¶m);
if(err < 0){
perror("ioctl sock_recv");
fail();
}
return err;
}
int sock_resize(int fd, size_t size) {
int err = ioctl(fd, IOCTL_SOCKS_RESIZE, size);
if(err < 0){
perror("ioctl sock_resize");
fail();
}
return err;
}
#endif