1 /*
2  * Copyright (c) 2016 Cyril Hrubis <chrubis@suse.cz>
3  * Copyright (c) 2015 Fujitsu Ltd.
4  *
5  * This program is free software: you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation, either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with this program. If not, see <http://www.gnu.org/licenses/>.
17  */
18 
19 #ifndef SAFE_NET_FN_H__
20 #define SAFE_NET_FN_H__
21 
22 #include <sys/types.h>
23 #include <sys/socket.h>
24 #include <netinet/in.h>
25 #include <arpa/inet.h>
26 #include <sys/un.h>
27 
28 int safe_socket(const char *file, const int lineno, void (cleanup_fn)(void),
29 		int domain, int type, int protocol);
30 
31 int safe_socketpair(const char *file, const int lineno, int domain, int type,
32 		    int protocol, int sv[]);
33 
34 int safe_getsockopt(const char *file, const int lineno, int sockfd, int level,
35 		    int optname, void *optval, socklen_t *optlen);
36 
37 int safe_setsockopt(const char *file, const int lineno, int sockfd, int level,
38 		    int optname, const void *optval, socklen_t optlen);
39 
40 ssize_t safe_send(const char *file, const int lineno, char len_strict,
41 		  int sockfd, const void *buf, size_t len, int flags);
42 
43 ssize_t safe_sendto(const char *file, const int lineno, char len_strict,
44 		    int sockfd, const void *buf, size_t len, int flags,
45 		    const struct sockaddr *dest_addr, socklen_t addrlen);
46 
47 ssize_t safe_sendmsg(const char *file, const int lineno, size_t msg_len,
48 		  int sockfd, const struct msghdr *msg, int flags);
49 
50 ssize_t safe_recvmsg(const char *file, const int lineno, size_t msg_len,
51 		  int sockfd, struct msghdr *msg, int flags);
52 
53 int safe_bind(const char *file, const int lineno, void (cleanup_fn)(void),
54 	      int socket, const struct sockaddr *address,
55 	      socklen_t address_len);
56 
57 int safe_listen(const char *file, const int lineno, void (cleanup_fn)(void),
58 		int socket, int backlog);
59 
60 int safe_accept(const char *file, const int lineno, void (cleanup_fn)(void),
61 		int sockfd, struct sockaddr *addr, socklen_t *addrlen);
62 
63 int safe_connect(const char *file, const int lineno, void (cleanup_fn)(void),
64 		 int sockfd, const struct sockaddr *addr, socklen_t addrlen);
65 
66 int safe_getsockname(const char *file, const int lineno,
67 		     void (cleanup_fn)(void), int sockfd, struct sockaddr *addr,
68 		     socklen_t *addrlen);
69 
70 int safe_gethostname(const char *file, const int lineno,
71 		     char *name, size_t size);
72 
73 int tst_getsockport(const char *file, const int lineno, int sockfd);
74 
75 unsigned short tst_get_unused_port(const char *file, const int lineno,
76 	      void (cleanup_fn)(void), unsigned short family, int type);
77 
78 char *tst_sock_addr(const struct sockaddr *sa, socklen_t salen, char *res,
79 		    size_t len);
80 
81 #endif /* SAFE_NET_FN_H__ */
82