Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 1 | /* |
Willy Tarreau | eb47268 | 2010-05-28 18:46:57 +0200 | [diff] [blame] | 2 | * include/proto/stream_sock.h |
| 3 | * This file contains client-side definitions. |
| 4 | * |
| 5 | * Copyright (C) 2000-2010 Willy Tarreau - w@1wt.eu |
| 6 | * |
| 7 | * This library is free software; you can redistribute it and/or |
| 8 | * modify it under the terms of the GNU Lesser General Public |
| 9 | * License as published by the Free Software Foundation, version 2.1 |
| 10 | * exclusively. |
| 11 | * |
| 12 | * This library is distributed in the hope that it will be useful, |
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 15 | * Lesser General Public License for more details. |
| 16 | * |
| 17 | * You should have received a copy of the GNU Lesser General Public |
| 18 | * License along with this library; if not, write to the Free Software |
| 19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 20 | */ |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 21 | |
| 22 | #ifndef _PROTO_STREAM_SOCK_H |
| 23 | #define _PROTO_STREAM_SOCK_H |
| 24 | |
| 25 | #include <stdlib.h> |
| 26 | #include <sys/socket.h> |
| 27 | #include <sys/types.h> |
| 28 | |
Willy Tarreau | 2dd0d47 | 2006-06-29 17:53:05 +0200 | [diff] [blame] | 29 | #include <common/config.h> |
Willy Tarreau | fa7e102 | 2008-10-19 07:30:41 +0200 | [diff] [blame] | 30 | #include <types/stream_interface.h> |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 31 | |
| 32 | |
Willy Tarreau | f8306d5 | 2006-07-29 19:01:31 +0200 | [diff] [blame] | 33 | /* main event functions used to move data between sockets and buffers */ |
Willy Tarreau | eb47268 | 2010-05-28 18:46:57 +0200 | [diff] [blame] | 34 | int stream_sock_accept(int fd); |
Willy Tarreau | d797128 | 2006-07-29 18:36:34 +0200 | [diff] [blame] | 35 | int stream_sock_read(int fd); |
Willy Tarreau | f8306d5 | 2006-07-29 19:01:31 +0200 | [diff] [blame] | 36 | int stream_sock_write(int fd); |
Willy Tarreau | b025325 | 2008-11-30 21:37:12 +0100 | [diff] [blame] | 37 | void stream_sock_data_finish(struct stream_interface *si); |
Willy Tarreau | 0a5d5dd | 2008-11-23 19:31:35 +0100 | [diff] [blame] | 38 | void stream_sock_shutr(struct stream_interface *si); |
| 39 | void stream_sock_shutw(struct stream_interface *si); |
Willy Tarreau | 3ffeba1 | 2008-12-14 14:42:35 +0100 | [diff] [blame] | 40 | void stream_sock_chk_rcv(struct stream_interface *si); |
| 41 | void stream_sock_chk_snd(struct stream_interface *si); |
Willy Tarreau | a8f55d5 | 2010-05-31 17:44:19 +0200 | [diff] [blame] | 42 | void stream_sock_prepare_interface(struct stream_interface *si); |
Willy Tarreau | baaee00 | 2006-06-26 02:48:02 +0200 | [diff] [blame] | 43 | |
| 44 | |
| 45 | /* This either returns the sockname or the original destination address. Code |
| 46 | * inspired from Patrick Schaaf's example of nf_getsockname() implementation. |
| 47 | */ |
| 48 | static inline int get_original_dst(int fd, struct sockaddr_in *sa, socklen_t *salen) { |
| 49 | #if defined(TPROXY) && defined(SO_ORIGINAL_DST) |
| 50 | return getsockopt(fd, SOL_IP, SO_ORIGINAL_DST, (void *)sa, salen); |
| 51 | #else |
| 52 | #if defined(TPROXY) && defined(USE_GETSOCKNAME) |
| 53 | return getsockname(fd, (struct sockaddr *)sa, salen); |
| 54 | #else |
| 55 | return -1; |
| 56 | #endif |
| 57 | #endif |
| 58 | } |
| 59 | |
| 60 | |
| 61 | #endif /* _PROTO_STREAM_SOCK_H */ |
| 62 | |
| 63 | /* |
| 64 | * Local variables: |
| 65 | * c-indent-level: 8 |
| 66 | * c-basic-offset: 8 |
| 67 | * End: |
| 68 | */ |