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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
|
/*
** Zabbix
** Copyright (C) 2001-2019 Zabbix SIA
**
** This program is free software; you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation; either version 2 of the License, or
** (at your option) any later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software
** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
**/
#ifndef ZABBIX_COMMS_H
#define ZABBIX_COMMS_H
#include "zbxtypes.h"
#ifdef _WINDOWS
# define ZBX_TCP_WRITE(s, b, bl) ((ssize_t)send((s), (b), (int)(bl), 0))
# define ZBX_TCP_READ(s, b, bl) ((ssize_t)recv((s), (b), (int)(bl), 0))
# define zbx_socket_close(s) if (ZBX_SOCKET_ERROR != (s)) closesocket(s)
# define zbx_socket_last_error() WSAGetLastError()
# define zbx_bind(s, a, l) (bind((s), (a), (int)(l)))
# define zbx_sendto(fd, b, n, f, a, l) (sendto((fd), (b), (int)(n), (f), (a), (l)))
# define ZBX_PROTO_AGAIN WSAEINTR
# define ZBX_PROTO_ERROR SOCKET_ERROR
# define ZBX_SOCKET_ERROR INVALID_SOCKET
# define ZBX_SOCKET_TO_INT(s) ((int)(s))
#else
# define ZBX_TCP_WRITE(s, b, bl) ((ssize_t)write((s), (b), (bl)))
# define ZBX_TCP_READ(s, b, bl) ((ssize_t)read((s), (b), (bl)))
# define zbx_socket_close(s) if (ZBX_SOCKET_ERROR != (s)) close(s)
# define zbx_socket_last_error() errno
# define zbx_bind(s, a, l) (bind((s), (a), (l)))
# define zbx_sendto(fd, b, n, f, a, l) (sendto((fd), (b), (n), (f), (a), (l)))
# define ZBX_PROTO_AGAIN EINTR
# define ZBX_PROTO_ERROR -1
# define ZBX_SOCKET_ERROR -1
# define ZBX_SOCKET_TO_INT(s) (s)
#endif
#ifdef _WINDOWS
typedef SOCKET ZBX_SOCKET;
#else
typedef int ZBX_SOCKET;
#endif
#if defined(HAVE_IPV6)
# define ZBX_SOCKADDR struct sockaddr_storage
#else
# define ZBX_SOCKADDR struct sockaddr_in
#endif
typedef enum
{
ZBX_BUF_TYPE_STAT = 0,
ZBX_BUF_TYPE_DYN
}
zbx_buf_type_t;
#define ZBX_SOCKET_COUNT 256
#define ZBX_STAT_BUF_LEN 2048
#if defined(HAVE_POLARSSL) || defined(HAVE_GNUTLS) || defined(HAVE_OPENSSL)
typedef struct zbx_tls_context zbx_tls_context_t;
#endif
typedef struct
{
ZBX_SOCKET socket;
ZBX_SOCKET socket_orig;
size_t read_bytes;
char *buffer;
char *next_line;
#if defined(HAVE_POLARSSL) || defined(HAVE_GNUTLS) || defined(HAVE_OPENSSL)
zbx_tls_context_t *tls_ctx;
#endif
unsigned int connection_type; /* type of connection actually established: */
/* ZBX_TCP_SEC_UNENCRYPTED, ZBX_TCP_SEC_TLS_PSK or */
/* ZBX_TCP_SEC_TLS_CERT */
int timeout;
zbx_buf_type_t buf_type;
unsigned char accepted;
int num_socks;
ZBX_SOCKET sockets[ZBX_SOCKET_COUNT];
char buf_stat[ZBX_STAT_BUF_LEN];
ZBX_SOCKADDR peer_info; /* getpeername() result */
/* Peer host DNS name or IP address for diagnostics (after TCP connection is established). */
/* TLS connection may be shut down at any time and it will not be possible to get peer IP address anymore. */
char peer[MAX_ZBX_DNSNAME_LEN + 1];
int protocol;
}
zbx_socket_t;
const char *zbx_socket_strerror(void);
#ifndef _WINDOWS
void zbx_gethost_by_ip(const char *ip, char *host, size_t hostlen);
void zbx_getip_by_host(const char *host, char *ip, size_t iplen);
#endif
int zbx_tcp_connect(zbx_socket_t *s, const char *source_ip, const char *ip, unsigned short port, int timeout,
unsigned int tls_connect, const char *tls_arg1, const char *tls_arg2);
#define ZBX_TCP_PROTOCOL 0x01
#define ZBX_TCP_COMPRESS 0x02
#define ZBX_TCP_SEC_UNENCRYPTED 1 /* do not use encryption with this socket */
#define ZBX_TCP_SEC_TLS_PSK 2 /* use TLS with pre-shared key (PSK) with this socket */
#define ZBX_TCP_SEC_TLS_CERT 4 /* use TLS with certificate with this socket */
#define ZBX_TCP_SEC_UNENCRYPTED_TXT "unencrypted"
#define ZBX_TCP_SEC_TLS_PSK_TXT "psk"
#define ZBX_TCP_SEC_TLS_CERT_TXT "cert"
const char *zbx_tcp_connection_type_name(unsigned int type);
#define zbx_tcp_send(s, d) zbx_tcp_send_ext((s), (d), strlen(d), ZBX_TCP_PROTOCOL, 0)
#define zbx_tcp_send_to(s, d, timeout) zbx_tcp_send_ext((s), (d), strlen(d), ZBX_TCP_PROTOCOL, timeout)
#define zbx_tcp_send_bytes_to(s, d, len, timeout) zbx_tcp_send_ext((s), (d), len, ZBX_TCP_PROTOCOL, timeout)
#define zbx_tcp_send_raw(s, d) zbx_tcp_send_ext((s), (d), strlen(d), 0, 0)
int zbx_tcp_send_ext(zbx_socket_t *s, const char *data, size_t len, unsigned char flags, int timeout);
void zbx_tcp_close(zbx_socket_t *s);
#ifdef HAVE_IPV6
int get_address_family(const char *addr, int *family, char *error, int max_error_len);
#endif
int zbx_tcp_listen(zbx_socket_t *s, const char *listen_ip, unsigned short listen_port);
int zbx_tcp_accept(zbx_socket_t *s, unsigned int tls_accept);
void zbx_tcp_unaccept(zbx_socket_t *s);
#define ZBX_TCP_READ_UNTIL_CLOSE 0x01
#define zbx_tcp_recv(s) SUCCEED_OR_FAIL(zbx_tcp_recv_ext(s, 0))
#define zbx_tcp_recv_to(s, timeout) SUCCEED_OR_FAIL(zbx_tcp_recv_ext(s, timeout))
#define zbx_tcp_recv_raw(s) SUCCEED_OR_FAIL(zbx_tcp_recv_raw_ext(s, 0))
ssize_t zbx_tcp_recv_ext(zbx_socket_t *s, int timeout);
ssize_t zbx_tcp_recv_raw_ext(zbx_socket_t *s, int timeout);
const char *zbx_tcp_recv_line(zbx_socket_t *s);
int zbx_validate_peer_list(const char *peer_list, char **error);
int zbx_tcp_check_allowed_peers(const zbx_socket_t *s, const char *peer_list);
int zbx_udp_connect(zbx_socket_t *s, const char *source_ip, const char *ip, unsigned short port, int timeout);
int zbx_udp_send(zbx_socket_t *s, const char *data, size_t data_len, int timeout);
int zbx_udp_recv(zbx_socket_t *s, int timeout);
void zbx_udp_close(zbx_socket_t *s);
#define ZBX_DEFAULT_FTP_PORT 21
#define ZBX_DEFAULT_SSH_PORT 22
#define ZBX_DEFAULT_TELNET_PORT 23
#define ZBX_DEFAULT_SMTP_PORT 25
#define ZBX_DEFAULT_DNS_PORT 53
#define ZBX_DEFAULT_HTTP_PORT 80
#define ZBX_DEFAULT_POP_PORT 110
#define ZBX_DEFAULT_NNTP_PORT 119
#define ZBX_DEFAULT_NTP_PORT 123
#define ZBX_DEFAULT_IMAP_PORT 143
#define ZBX_DEFAULT_LDAP_PORT 389
#define ZBX_DEFAULT_HTTPS_PORT 443
#define ZBX_DEFAULT_AGENT_PORT 10050
#define ZBX_DEFAULT_SERVER_PORT 10051
#define ZBX_DEFAULT_GATEWAY_PORT 10052
#define ZBX_DEFAULT_AGENT_PORT_STR "10050"
#define ZBX_DEFAULT_SERVER_PORT_STR "10051"
int zbx_send_response_ext(zbx_socket_t *sock, int result, const char *info, const char *version, int protocol,
int timeout);
#define zbx_send_response(sock, result, info, timeout) \
zbx_send_response_ext(sock, result, info, NULL, ZBX_TCP_PROTOCOL, timeout)
#define zbx_send_response_same(sock, result, info, timeout) \
zbx_send_response_ext(sock, result, info, NULL, sock->protocol, timeout)
#define zbx_send_proxy_response(sock, result, info, timeout) \
zbx_send_response_ext(sock, result, info, ZABBIX_VERSION, ZBX_TCP_PROTOCOL | ZBX_TCP_COMPRESS, timeout)
int zbx_recv_response(zbx_socket_t *sock, int timeout, char **error);
#ifdef HAVE_IPV6
# define zbx_getnameinfo(sa, host, hostlen, serv, servlen, flags) \
getnameinfo(sa, AF_INET == (sa)->sa_family ? \
sizeof(struct sockaddr_in) : \
sizeof(struct sockaddr_in6), \
host, hostlen, serv, servlen, flags)
#endif
#ifdef _WINDOWS
int zbx_socket_start(char **error);
#endif
#endif
|
