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