Commit 4562e816f0ba7b8fb71645b03a8c8ea54eb1faf4
1 parent
7a8bdb3e
Add code to create a socket pair and send filedescriptors
Showing
7 changed files
with
114 additions
and
15 deletions
... | ... | @@ -66,6 +66,7 @@ TR_CLASS(TR_Socket) { |
66 | 66 | socklen_t addrlen; |
67 | 67 | int handle; |
68 | 68 | TR_SocketFin fin_state; |
69 | + int fd_getter; | |
69 | 70 | }; |
70 | 71 | TR_INSTANCE_INIT(TR_Socket); |
71 | 72 | TR_CLASSVARS_DECL(TR_Socket) {}; |
... | ... | @@ -81,6 +82,7 @@ TR_CLASSVARS_DECL(TR_Socket) {}; |
81 | 82 | #define TR_socketAddrlen(socket) ((socket)->addrlen) |
82 | 83 | #define TR_socketHandle(socket) ((socket)->handle) |
83 | 84 | #define TR_socketFinState(socket) ((socket)->fin_state) |
85 | +#define TR_socketFdGetter(socket) ((socket)->fd_getter) | |
84 | 86 | #define TR_socketFinRd(socket) \ |
85 | 87 | ((TR_socketFinState(socket) & TR_FIN_RD) == TR_FIN_RD) |
86 | 88 | #define TR_socketFinWr(socket) \ |
... | ... | @@ -118,7 +120,6 @@ TR_CLASSVARS_DECL(TR_Socket) {}; |
118 | 120 | |
119 | 121 | TR_CLASS(TR_TcpSocket) { |
120 | 122 | TR_EXTENDS(TR_Socket); |
121 | - int listen; | |
122 | 123 | int connected; |
123 | 124 | }; |
124 | 125 | TR_INSTANCE_INIT(TR_TcpSocket); |
... | ... | @@ -151,6 +152,8 @@ TR_TcpSocket TR_socketAccept(TR_TcpSocket); |
151 | 152 | (TR_socketInit((socket), NULL)) |
152 | 153 | |
153 | 154 | void TR_socketPair(TR_Socket[2], int); |
155 | +void TR_socketSendFd(TR_Socket, int); | |
156 | +int TR_socketGetFd(TR_Socket); | |
154 | 157 | |
155 | 158 | #endif // __TR_SOCKET_H__ |
156 | 159 | ... | ... |
... | ... | @@ -42,7 +42,6 @@ TR_socketAccept(TR_TcpSocket this) |
42 | 42 | { |
43 | 43 | TR_Socket remote = |
44 | 44 | TR_new(TR_TcpSocket, TR_socketLog((TR_Socket)this), NULL, 0, 0); |
45 | - //int flags; | |
46 | 45 | |
47 | 46 | remote->addrlen = ((TR_Socket)this)->addrlen; |
48 | 47 | remote->handle = accept( |
... | ... | @@ -50,9 +49,6 @@ TR_socketAccept(TR_TcpSocket this) |
50 | 49 | &(remote->addr.info), |
51 | 50 | &(remote->addrlen)); |
52 | 51 | |
53 | - //flags = fcntl(remote->handle, F_GETFL, 0); | |
54 | - //fcntl(remote->handle, F_SETFL, flags | O_NONBLOCK); | |
55 | - | |
56 | 52 | if (-1 == remote->handle) { |
57 | 53 | TR_delete(remote); |
58 | 54 | } else { | ... | ... |
src/socket_get_fd.c
0 → 100644
1 | +/** | |
2 | + * \file | |
3 | + * | |
4 | + * \author Georg Hopp | |
5 | + * | |
6 | + * \copyright | |
7 | + * Copyright © 2014 Georg Hopp | |
8 | + * | |
9 | + * This program is free software: you can redistribute it and/or modify | |
10 | + * it under the terms of the GNU General Public License as published by | |
11 | + * the Free Software Foundation, either version 3 of the License, or | |
12 | + * (at your option) any later version. | |
13 | + * | |
14 | + * This program is distributed in the hope that it will be useful, | |
15 | + * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
16 | + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
17 | + * GNU General Public License for more details. | |
18 | + * | |
19 | + * You should have received a copy of the GNU General Public License | |
20 | + * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
21 | + */ | |
22 | + | |
23 | +#include <errno.h> // for errno | |
24 | +#include <unistd.h> | |
25 | +#include <fcntl.h> | |
26 | + | |
27 | +#include "tr/socket.h" | |
28 | +#include "tr/logger.h" | |
29 | + | |
30 | +int | |
31 | +TR_socketGetFd(TR_Socket this) | |
32 | +{ | |
33 | + struct msghdr msg; | |
34 | + struct cmsghdr * cmsg; | |
35 | + char cmsgbuf[CMSG_SPACE(sizeof(int))]; | |
36 | + int fd = 0; | |
37 | + | |
38 | + memset(&msg, 0, sizeof(msg)); | |
39 | + msg.msg_control = cmsgbuf; | |
40 | + msg.msg_controllen = sizeof(cmsgbuf); | |
41 | + | |
42 | + if (0 > recvmsg(TR_socketHandle(this), &msg, 0)) { | |
43 | + return -1; | |
44 | + } | |
45 | + | |
46 | + cmsg = CMSG_FIRSTHDR(&msg); | |
47 | + if (cmsg == NULL || cmsg->cmsg_type != SCM_RIGHTS) { | |
48 | + return -1; | |
49 | + } | |
50 | + | |
51 | + memcpy(&fd, CMSG_DATA(cmsg), sizeof(int)); | |
52 | + return fd; | |
53 | +} | |
54 | + | |
55 | +// vim: set ts=4 sw=4: | ... | ... |
... | ... | @@ -20,12 +20,6 @@ |
20 | 20 | * along with this program. If not, see <http://www.gnu.org/licenses/>. |
21 | 21 | */ |
22 | 22 | |
23 | -//#include <errno.h> | |
24 | -//#include <stdlib.h> | |
25 | -//#include <unistd.h> | |
26 | -//#include <stdio.h> | |
27 | -//#include <fcntl.h> | |
28 | - | |
29 | 23 | #include <sys/types.h> |
30 | 24 | #include <sys/socket.h> |
31 | 25 | #include <netdb.h> | ... | ... |
src/socket_send_fd.c
0 → 100644
1 | +/** | |
2 | + * \file | |
3 | + * | |
4 | + * \author Georg Hopp | |
5 | + * | |
6 | + * \copyright | |
7 | + * Copyright © 2014 Georg Hopp | |
8 | + * | |
9 | + * This program is free software: you can redistribute it and/or modify | |
10 | + * it under the terms of the GNU General Public License as published by | |
11 | + * the Free Software Foundation, either version 3 of the License, or | |
12 | + * (at your option) any later version. | |
13 | + * | |
14 | + * This program is distributed in the hope that it will be useful, | |
15 | + * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
16 | + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
17 | + * GNU General Public License for more details. | |
18 | + * | |
19 | + * You should have received a copy of the GNU General Public License | |
20 | + * along with this program. If not, see <http://www.gnu.org/licenses/>. | |
21 | + */ | |
22 | + | |
23 | +#include <errno.h> // for errno | |
24 | +#include <unistd.h> | |
25 | +#include <fcntl.h> | |
26 | + | |
27 | +#include "tr/socket.h" | |
28 | +#include "tr/logger.h" | |
29 | + | |
30 | +void | |
31 | +TR_socketSendFd(TR_Socket this, int fd) | |
32 | +{ | |
33 | + struct msghdr msg; | |
34 | + struct cmsghdr * cmsg; | |
35 | + char cmsgbuf[CMSG_SPACE(sizeof(int))]; | |
36 | + | |
37 | + memset(&msg, 0, sizeof(msg)); | |
38 | + msg.msg_control = cmsgbuf; | |
39 | + msg.msg_controllen = sizeof(cmsgbuf); // necessary for CMSG_FIRSTHDR to | |
40 | + // return the correct value | |
41 | + cmsg = CMSG_FIRSTHDR(&msg); | |
42 | + cmsg->cmsg_level = SOL_SOCKET; | |
43 | + cmsg->cmsg_type = SCM_RIGHTS; | |
44 | + cmsg->cmsg_len = CMSG_LEN(sizeof(int)); | |
45 | + memcpy(CMSG_DATA(cmsg), &fd, sizeof(int)); | |
46 | + msg.msg_controllen = cmsg->cmsg_len; // total size of all control blocks | |
47 | + | |
48 | + sendmsg(TR_socketHandle(this), &msg, 0); | |
49 | + /* TODO error handling... */ | |
50 | +} | |
51 | + | |
52 | +// vim: set ts=4 sw=4: | ... | ... |
... | ... | @@ -41,10 +41,7 @@ tcpSocketCtor(void * _this, va_list * params) |
41 | 41 | TR_TcpSocket this = _this; |
42 | 42 | |
43 | 43 | TR_PARENTCALL(TR_TcpSocket, _this, TR_Class, ctor, params); |
44 | - | |
45 | 44 | TR_socketType((TR_Socket)this) = SOCK_STREAM; |
46 | - this->listen = FALSE; | |
47 | - this->connected = FALSE; | |
48 | 45 | |
49 | 46 | return 0; |
50 | 47 | } |
... | ... | @@ -66,7 +63,7 @@ tcpSocketBind(void * _this) |
66 | 63 | // error |
67 | 64 | return -1; |
68 | 65 | } |
69 | - this->listen = TRUE; | |
66 | + ((TR_Socket)this)->fd_getter = TRUE; | |
70 | 67 | |
71 | 68 | return 0; |
72 | 69 | } | ... | ... |
Please
register
or
login
to post a comment