blob: 2db7d3de525b1fef76480c8d48b2c44c188026b8 [file] [log] [blame]
Michal Vasko086311b2016-01-08 09:53:11 +01001/**
Michal Vasko95ea9ff2021-11-09 12:29:14 +01002 * @file session_server.c
3 * @author Michal Vasko <mvasko@cesnet.cz>
4 * @brief libnetconf2 server session manipulation functions
Michal Vasko086311b2016-01-08 09:53:11 +01005 *
Michal Vasko95ea9ff2021-11-09 12:29:14 +01006 * @copyright
7 * Copyright (c) 2015 - 2021 CESNET, z.s.p.o.
Michal Vasko086311b2016-01-08 09:53:11 +01008 *
Radek Krejci9b81f5b2016-02-24 13:14:49 +01009 * This source code is licensed under BSD 3-Clause License (the "License").
10 * You may not use this file except in compliance with the License.
11 * You may obtain a copy of the License at
Michal Vaskoafd416b2016-02-25 14:51:46 +010012 *
Radek Krejci9b81f5b2016-02-24 13:14:49 +010013 * https://opensource.org/licenses/BSD-3-Clause
Michal Vasko086311b2016-01-08 09:53:11 +010014 */
apropp-molex4e903c32020-04-20 03:06:58 -040015#define _QNX_SOURCE /* getpeereid */
Olivier Matzac7fa2f2018-10-11 10:02:04 +020016#define _GNU_SOURCE /* signals, threads, SO_PEERCRED */
Michal Vasko086311b2016-01-08 09:53:11 +010017
Michal Vaskob83a3fa2021-05-26 09:53:42 +020018#include <arpa/inet.h>
Michal Vasko77e83572022-07-21 15:31:15 +020019#include <assert.h>
Michal Vasko086311b2016-01-08 09:53:11 +010020#include <errno.h>
Michal Vaskob83a3fa2021-05-26 09:53:42 +020021#include <fcntl.h>
Olivier Matzac7fa2f2018-10-11 10:02:04 +020022#include <netinet/in.h>
23#include <netinet/tcp.h>
Michal Vaskob83a3fa2021-05-26 09:53:42 +020024#include <poll.h>
Michal Vaskob48aa812016-01-18 14:13:09 +010025#include <pthread.h>
Olivier Matzac7fa2f2018-10-11 10:02:04 +020026#include <pwd.h>
Michal Vaskob83a3fa2021-05-26 09:53:42 +020027#include <signal.h>
28#include <stdint.h>
29#include <stdlib.h>
30#include <string.h>
31#include <sys/socket.h>
32#include <sys/types.h>
33#include <sys/un.h>
34#include <time.h>
35#include <unistd.h>
Michal Vasko086311b2016-01-08 09:53:11 +010036
Michal Vasko7a20d2e2021-05-19 16:40:23 +020037#include "compat.h"
Michal Vasko1a38c862016-01-15 15:50:07 +010038#include "libnetconf.h"
Michal Vasko086311b2016-01-08 09:53:11 +010039#include "session_server.h"
Michal Vasko0bdf70b2019-06-24 19:20:20 +020040#include "session_server_ch.h"
Michal Vasko086311b2016-01-08 09:53:11 +010041
Michal Vaskob48aa812016-01-18 14:13:09 +010042struct nc_server_opts server_opts = {
Michal Vaskoade892d2017-02-22 13:40:35 +010043#ifdef NC_ENABLED_SSH
44 .authkey_lock = PTHREAD_MUTEX_INITIALIZER,
45#endif
46 .bind_lock = PTHREAD_MUTEX_INITIALIZER,
Michal Vasko2e6defd2016-10-07 15:48:15 +020047 .endpt_lock = PTHREAD_RWLOCK_INITIALIZER,
48 .ch_client_lock = PTHREAD_RWLOCK_INITIALIZER
Michal Vaskob48aa812016-01-18 14:13:09 +010049};
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +010050
fanchanghu966f2de2016-07-21 02:28:57 -040051static nc_rpc_clb global_rpc_clb = NULL;
52
Michal Vasko3031aae2016-01-27 16:07:18 +010053struct nc_endpt *
Michal Vaskoade892d2017-02-22 13:40:35 +010054nc_server_endpt_lock_get(const char *name, NC_TRANSPORT_IMPL ti, uint16_t *idx)
Michal Vasko3031aae2016-01-27 16:07:18 +010055{
56 uint16_t i;
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +010057 struct nc_endpt *endpt = NULL;
58
Michal Vaskoddce1212019-05-24 09:58:49 +020059 if (!name) {
60 ERRARG("endpt_name");
61 return NULL;
62 }
63
Michal Vaskoade892d2017-02-22 13:40:35 +010064 /* WRITE LOCK */
65 pthread_rwlock_wrlock(&server_opts.endpt_lock);
Michal Vasko3031aae2016-01-27 16:07:18 +010066
67 for (i = 0; i < server_opts.endpt_count; ++i) {
Michal Vasko2e6defd2016-10-07 15:48:15 +020068 if (!strcmp(server_opts.endpts[i].name, name) && (!ti || (server_opts.endpts[i].ti == ti))) {
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +010069 endpt = &server_opts.endpts[i];
70 break;
Michal Vasko3031aae2016-01-27 16:07:18 +010071 }
72 }
73
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +010074 if (!endpt) {
Michal Vasko05532772021-06-03 12:12:38 +020075 ERR(NULL, "Endpoint \"%s\" was not found.", name);
Michal Vaskoade892d2017-02-22 13:40:35 +010076 /* UNLOCK */
Michal Vasko2e6defd2016-10-07 15:48:15 +020077 pthread_rwlock_unlock(&server_opts.endpt_lock);
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +010078 return NULL;
79 }
80
Michal Vaskoe2713da2016-08-22 16:06:40 +020081 if (idx) {
82 *idx = i;
83 }
84
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +010085 return endpt;
86}
87
Michal Vaskoadf30f02019-06-24 09:34:47 +020088struct nc_ch_endpt *
89nc_server_ch_client_lock(const char *name, const char *endpt_name, NC_TRANSPORT_IMPL ti, struct nc_ch_client **client_p)
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +010090{
Michal Vaskoadf30f02019-06-24 09:34:47 +020091 uint16_t i, j;
Michal Vasko2e6defd2016-10-07 15:48:15 +020092 struct nc_ch_client *client = NULL;
Michal Vaskoadf30f02019-06-24 09:34:47 +020093 struct nc_ch_endpt *endpt = NULL;
94
95 *client_p = NULL;
Michal Vasko2e6defd2016-10-07 15:48:15 +020096
Michal Vaskoddce1212019-05-24 09:58:49 +020097 if (!name) {
98 ERRARG("client_name");
99 return NULL;
100 }
101
Michal Vasko2e6defd2016-10-07 15:48:15 +0200102 /* READ LOCK */
103 pthread_rwlock_rdlock(&server_opts.ch_client_lock);
104
105 for (i = 0; i < server_opts.ch_client_count; ++i) {
Michal Vaskoadf30f02019-06-24 09:34:47 +0200106 if (!strcmp(server_opts.ch_clients[i].name, name)) {
Michal Vasko2e6defd2016-10-07 15:48:15 +0200107 client = &server_opts.ch_clients[i];
Michal Vaskoadf30f02019-06-24 09:34:47 +0200108 if (!endpt_name && !ti) {
109 /* return only client */
110 break;
111 }
112 for (j = 0; j < client->ch_endpt_count; ++j) {
Michal Vasko530d95c2021-05-28 13:32:02 +0200113 if ((!endpt_name || !strcmp(client->ch_endpts[j].name, endpt_name)) &&
114 (!ti || (ti == client->ch_endpts[j].ti))) {
Michal Vaskoadf30f02019-06-24 09:34:47 +0200115 endpt = &client->ch_endpts[j];
116 break;
117 }
118 }
Michal Vasko2e6defd2016-10-07 15:48:15 +0200119 break;
120 }
121 }
122
123 if (!client) {
Michal Vasko05532772021-06-03 12:12:38 +0200124 ERR(NULL, "Call Home client \"%s\" was not found.", name);
Michal Vaskoadf30f02019-06-24 09:34:47 +0200125
Michal Vasko2e6defd2016-10-07 15:48:15 +0200126 /* READ UNLOCK */
127 pthread_rwlock_unlock(&server_opts.ch_client_lock);
Michal Vaskoadf30f02019-06-24 09:34:47 +0200128 } else if (endpt_name && ti && !endpt) {
Michal Vasko05532772021-06-03 12:12:38 +0200129 ERR(NULL, "Call Home client \"%s\" endpoint \"%s\" was not found.", name, endpt_name);
Michal Vaskoadf30f02019-06-24 09:34:47 +0200130
131 /* READ UNLOCK */
132 pthread_rwlock_unlock(&server_opts.ch_client_lock);
133 } else {
134 /* CH CLIENT LOCK */
135 pthread_mutex_lock(&client->lock);
136
137 *client_p = client;
Michal Vasko2e6defd2016-10-07 15:48:15 +0200138 }
139
Michal Vaskoadf30f02019-06-24 09:34:47 +0200140 return endpt;
Michal Vasko2e6defd2016-10-07 15:48:15 +0200141}
142
143void
144nc_server_ch_client_unlock(struct nc_ch_client *client)
145{
146 /* CH CLIENT UNLOCK */
147 pthread_mutex_unlock(&client->lock);
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +0100148
149 /* READ UNLOCK */
Michal Vasko2e6defd2016-10-07 15:48:15 +0200150 pthread_rwlock_unlock(&server_opts.ch_client_lock);
Michal Vasko3031aae2016-01-27 16:07:18 +0100151}
Michal Vasko086311b2016-01-08 09:53:11 +0100152
Michal Vasko1a38c862016-01-15 15:50:07 +0100153API void
154nc_session_set_term_reason(struct nc_session *session, NC_SESSION_TERM_REASON reason)
155{
Michal Vasko45e53ae2016-04-07 11:46:03 +0200156 if (!session) {
157 ERRARG("session");
158 return;
159 } else if (!reason) {
160 ERRARG("reason");
Michal Vasko1a38c862016-01-15 15:50:07 +0100161 return;
162 }
163
Michal Vasko142cfea2017-08-07 10:12:11 +0200164 if ((reason != NC_SESSION_TERM_KILLED) && (session->term_reason == NC_SESSION_TERM_KILLED)) {
165 session->killed_by = 0;
166 }
Michal Vasko1a38c862016-01-15 15:50:07 +0100167 session->term_reason = reason;
168}
169
Michal Vasko142cfea2017-08-07 10:12:11 +0200170API void
171nc_session_set_killed_by(struct nc_session *session, uint32_t sid)
172{
173 if (!session || (session->term_reason != NC_SESSION_TERM_KILLED)) {
174 ERRARG("session");
175 return;
176 } else if (!sid) {
177 ERRARG("sid");
178 return;
179 }
180
181 session->killed_by = sid;
182}
183
184API void
185nc_session_set_status(struct nc_session *session, NC_STATUS status)
186{
187 if (!session) {
188 ERRARG("session");
189 return;
190 } else if (!status) {
191 ERRARG("status");
192 return;
193 }
194
195 session->status = status;
196}
197
Michal Vasko086311b2016-01-08 09:53:11 +0100198int
Michal Vaskoe49a15f2019-05-27 14:18:36 +0200199nc_sock_listen_inet(const char *address, uint16_t port, struct nc_keepalives *ka)
Michal Vasko086311b2016-01-08 09:53:11 +0100200{
Michal Vasko06c860d2018-07-09 16:08:52 +0200201 int opt;
Michal Vasko086311b2016-01-08 09:53:11 +0100202 int is_ipv4, sock;
203 struct sockaddr_storage saddr;
204
205 struct sockaddr_in *saddr4;
206 struct sockaddr_in6 *saddr6;
207
Michal Vasko086311b2016-01-08 09:53:11 +0100208 if (!strchr(address, ':')) {
209 is_ipv4 = 1;
210 } else {
211 is_ipv4 = 0;
212 }
213
214 sock = socket((is_ipv4 ? AF_INET : AF_INET6), SOCK_STREAM, 0);
215 if (sock == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200216 ERR(NULL, "Failed to create socket (%s).", strerror(errno));
Michal Vasko086311b2016-01-08 09:53:11 +0100217 goto fail;
218 }
219
Michal Vaskobe52dc22018-10-17 09:28:17 +0200220 /* these options will be inherited by accepted sockets */
Michal Vasko06c860d2018-07-09 16:08:52 +0200221 opt = 1;
222 if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof opt) == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200223 ERR(NULL, "Could not set SO_REUSEADDR socket option (%s).", strerror(errno));
Michal Vasko06c860d2018-07-09 16:08:52 +0200224 goto fail;
225 }
Michal Vasko83ad17e2019-01-30 10:11:37 +0100226 if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200227 ERR(NULL, "Could not set TCP_NODELAY socket option (%s).", strerror(errno));
Michal Vasko83ad17e2019-01-30 10:11:37 +0100228 goto fail;
229 }
Michal Vaskobe52dc22018-10-17 09:28:17 +0200230
Michal Vaskoe49a15f2019-05-27 14:18:36 +0200231 if (nc_sock_enable_keepalive(sock, ka)) {
Michal Vasko086311b2016-01-08 09:53:11 +0100232 goto fail;
233 }
234
Michal Vaskof22d5ff2020-04-15 11:10:27 +0200235 memset(&saddr, 0, sizeof(struct sockaddr_storage));
Michal Vasko086311b2016-01-08 09:53:11 +0100236 if (is_ipv4) {
237 saddr4 = (struct sockaddr_in *)&saddr;
238
239 saddr4->sin_family = AF_INET;
240 saddr4->sin_port = htons(port);
241
242 if (inet_pton(AF_INET, address, &saddr4->sin_addr) != 1) {
Michal Vasko05532772021-06-03 12:12:38 +0200243 ERR(NULL, "Failed to convert IPv4 address \"%s\".", address);
Michal Vasko086311b2016-01-08 09:53:11 +0100244 goto fail;
245 }
246
247 if (bind(sock, (struct sockaddr *)saddr4, sizeof(struct sockaddr_in)) == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200248 ERR(NULL, "Could not bind \"%s\" port %d (%s).", address, port, strerror(errno));
Michal Vasko086311b2016-01-08 09:53:11 +0100249 goto fail;
250 }
251
252 } else {
253 saddr6 = (struct sockaddr_in6 *)&saddr;
254
255 saddr6->sin6_family = AF_INET6;
256 saddr6->sin6_port = htons(port);
257
258 if (inet_pton(AF_INET6, address, &saddr6->sin6_addr) != 1) {
Michal Vasko05532772021-06-03 12:12:38 +0200259 ERR(NULL, "Failed to convert IPv6 address \"%s\".", address);
Michal Vasko086311b2016-01-08 09:53:11 +0100260 goto fail;
261 }
262
263 if (bind(sock, (struct sockaddr *)saddr6, sizeof(struct sockaddr_in6)) == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200264 ERR(NULL, "Could not bind \"%s\" port %d (%s).", address, port, strerror(errno));
Michal Vasko086311b2016-01-08 09:53:11 +0100265 goto fail;
266 }
267 }
268
Michal Vaskofb89d772016-01-08 12:25:35 +0100269 if (listen(sock, NC_REVERSE_QUEUE) == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200270 ERR(NULL, "Unable to start listening on \"%s\" port %d (%s).", address, port, strerror(errno));
Michal Vasko086311b2016-01-08 09:53:11 +0100271 goto fail;
272 }
273
274 return sock;
275
276fail:
277 if (sock > -1) {
278 close(sock);
279 }
280
281 return -1;
282}
283
284int
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200285nc_sock_listen_unix(const char *address, const struct nc_server_unix_opts *opts)
286{
287 struct sockaddr_un sun;
288 int sock = -1;
289
Michal Vasko93e96f12021-09-30 10:02:09 +0200290 if (strlen(address) > sizeof(sun.sun_path) - 1) {
291 ERR(NULL, "Socket path \"%s\" is longer than maximum length %d.", address, (int)(sizeof(sun.sun_path) - 1));
292 goto fail;
293 }
294
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200295 sock = socket(AF_UNIX, SOCK_STREAM, 0);
296 if (sock == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200297 ERR(NULL, "Failed to create socket (%s).", strerror(errno));
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200298 goto fail;
299 }
300
301 memset(&sun, 0, sizeof(sun));
302 sun.sun_family = AF_UNIX;
Michal Vasko93e96f12021-09-30 10:02:09 +0200303 snprintf(sun.sun_path, sizeof(sun.sun_path) - 1, "%s", address);
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200304
305 unlink(sun.sun_path);
306 if (bind(sock, (struct sockaddr *)&sun, sizeof(sun)) == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200307 ERR(NULL, "Could not bind \"%s\" (%s).", address, strerror(errno));
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200308 goto fail;
309 }
310
311 if (opts->mode != (mode_t)-1) {
312 if (chmod(sun.sun_path, opts->mode) < 0) {
Michal Vasko05532772021-06-03 12:12:38 +0200313 ERR(NULL, "Failed to set unix socket permissions (%s).", strerror(errno));
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200314 goto fail;
315 }
316 }
317
Michal Vaskob83a3fa2021-05-26 09:53:42 +0200318 if ((opts->uid != (uid_t)-1) || (opts->gid != (gid_t)-1)) {
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200319 if (chown(sun.sun_path, opts->uid, opts->gid) < 0) {
Michal Vasko05532772021-06-03 12:12:38 +0200320 ERR(NULL, "Failed to set unix socket uid/gid (%s).", strerror(errno));
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200321 goto fail;
322 }
323 }
324
325 if (listen(sock, NC_REVERSE_QUEUE) == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200326 ERR(NULL, "Unable to start listening on \"%s\" (%s).", address, strerror(errno));
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200327 goto fail;
328 }
329
330 return sock;
331
332fail:
333 if (sock > -1) {
334 close(sock);
335 }
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200336 return -1;
337}
338
aPiecek90ff0242021-02-14 14:58:01 +0100339/**
340 * @brief Evaluate socket name for AF_UNIX socket.
341 * @param[in] acc_sock_fd is file descriptor for the accepted socket (a nonnegative).
342 * @param[out] host is pointer to char* to which the socket name will be set. It must not be NULL.
343 * @return 0 in case of success. Call free function for parameter host to avoid a memory leak.
344 * @return 0 if the stream socket is unnamed. Parameter host is set to NULL.
345 * @return -1 in case of error. Parameter host is set to NULL.
346 */
347static int
348sock_host_unix(int acc_sock_fd, char **host)
349{
350 char *sun_path;
351 struct sockaddr_storage saddr;
352 socklen_t addr_len;
353
354 *host = NULL;
355 saddr.ss_family = AF_UNIX;
356 addr_len = sizeof(saddr);
357
358 if (getsockname(acc_sock_fd, (struct sockaddr *)&saddr, &addr_len)) {
Michal Vasko05532772021-06-03 12:12:38 +0200359 ERR(NULL, "getsockname failed (%s).", strerror(errno));
aPiecek90ff0242021-02-14 14:58:01 +0100360 return -1;
361 }
362
363 sun_path = ((struct sockaddr_un *)&saddr)->sun_path;
364 if (!sun_path) {
365 /* stream socket is unnamed */
366 return 0;
367 }
368
369 if (!(*host = strdup(sun_path))) {
370 ERRMEM;
371 return -1;
372 }
373
374 return 0;
375}
376
377/**
378 * @brief Evaluate socket name and port number for AF_INET socket.
379 * @param[in] addr is pointing to structure filled by accept function which was successful.
380 * @param[out] host is pointer to char* to which the socket name will be set. It must not be NULL.
381 * @param[out] port is pointer to uint16_t to which the port number will be set. It must not be NULL.
382 * @return 0 in case of success. Call free function for parameter host to avoid a memory leak.
383 * @return -1 in case of error. Parameter host is set to NULL and port is unchanged.
384 */
385static int
386sock_host_inet(const struct sockaddr_in *addr, char **host, uint16_t *port)
387{
388 *host = malloc(INET_ADDRSTRLEN);
389 if (!(*host)) {
390 ERRMEM;
391 return -1;
392 }
393
aPiecek3da9b342021-02-18 15:00:03 +0100394 if (!inet_ntop(AF_INET, &addr->sin_addr, *host, INET_ADDRSTRLEN)) {
Michal Vasko69e98752022-12-14 14:20:17 +0100395 ERR(NULL, "inet_ntop failed (%s).", strerror(errno));
aPiecek90ff0242021-02-14 14:58:01 +0100396 free(*host);
397 *host = NULL;
398 return -1;
399 }
400
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200401 *port = ntohs(addr->sin_port);
aPiecek90ff0242021-02-14 14:58:01 +0100402
403 return 0;
404}
405
406/**
407 * @brief Evaluate socket name and port number for AF_INET6 socket.
408 * @param[in] addr is pointing to structure filled by accept function which was successful.
409 * @param[out] host is pointer to char* to which the socket name will be set. It must not be NULL.
410 * @param[out] port is pointer to uint16_t to which the port number will be set. It must not be NULL.
411 * @return 0 in case of success. Call free function for parameter host to avoid a memory leak.
412 * @return -1 in case of error. Parameter host is set to the NULL and port is unchanged.
413 */
414static int
415sock_host_inet6(const struct sockaddr_in6 *addr, char **host, uint16_t *port)
416{
Michal Vaskob83a3fa2021-05-26 09:53:42 +0200417 *host = malloc(INET6_ADDRSTRLEN);
aPiecek90ff0242021-02-14 14:58:01 +0100418 if (!(*host)) {
419 ERRMEM;
420 return -1;
421 }
422
aPiecek3da9b342021-02-18 15:00:03 +0100423 if (!inet_ntop(AF_INET6, &addr->sin6_addr, *host, INET6_ADDRSTRLEN)) {
Michal Vasko69e98752022-12-14 14:20:17 +0100424 ERR(NULL, "inet_ntop failed (%s).", strerror(errno));
aPiecek90ff0242021-02-14 14:58:01 +0100425 free(*host);
426 *host = NULL;
427 return -1;
428 }
429
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200430 *port = ntohs(addr->sin6_port);
aPiecek90ff0242021-02-14 14:58:01 +0100431
432 return 0;
433}
434
Olivier Matzac7fa2f2018-10-11 10:02:04 +0200435int
Michal Vasko3031aae2016-01-27 16:07:18 +0100436nc_sock_accept_binds(struct nc_bind *binds, uint16_t bind_count, int timeout, char **host, uint16_t *port, uint16_t *idx)
Michal Vasko086311b2016-01-08 09:53:11 +0100437{
Michal Vaskof54cd352017-02-22 13:42:02 +0100438 sigset_t sigmask, origmask;
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200439 uint16_t i, j, pfd_count, client_port;
440 char *client_address;
Michal Vasko086311b2016-01-08 09:53:11 +0100441 struct pollfd *pfd;
442 struct sockaddr_storage saddr;
443 socklen_t saddr_len = sizeof(saddr);
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200444 int ret, client_sock, sock = -1, flags;
Michal Vasko086311b2016-01-08 09:53:11 +0100445
446 pfd = malloc(bind_count * sizeof *pfd);
Michal Vasko4eb3c312016-03-01 14:09:37 +0100447 if (!pfd) {
448 ERRMEM;
449 return -1;
450 }
451
Michal Vaskoac2f6182017-01-30 14:32:03 +0100452 for (i = 0, pfd_count = 0; i < bind_count; ++i) {
Michal Vasko94acafc2016-09-23 13:40:10 +0200453 if (binds[i].sock < 0) {
454 /* invalid socket */
Michal Vasko94acafc2016-09-23 13:40:10 +0200455 continue;
456 }
Michal Vasko0a3f3752016-10-13 14:58:38 +0200457 if (binds[i].pollin) {
458 binds[i].pollin = 0;
459 /* leftover pollin */
460 sock = binds[i].sock;
Michal Vasko086311b2016-01-08 09:53:11 +0100461 break;
462 }
Michal Vaskoac2f6182017-01-30 14:32:03 +0100463 pfd[pfd_count].fd = binds[i].sock;
464 pfd[pfd_count].events = POLLIN;
465 pfd[pfd_count].revents = 0;
466
467 ++pfd_count;
Michal Vasko086311b2016-01-08 09:53:11 +0100468 }
469
Michal Vasko0a3f3752016-10-13 14:58:38 +0200470 if (sock == -1) {
471 /* poll for a new connection */
Michal Vaskof54cd352017-02-22 13:42:02 +0100472 sigfillset(&sigmask);
473 pthread_sigmask(SIG_SETMASK, &sigmask, &origmask);
Michal Vaskoac2f6182017-01-30 14:32:03 +0100474 ret = poll(pfd, pfd_count, timeout);
Michal Vaskof54cd352017-02-22 13:42:02 +0100475 pthread_sigmask(SIG_SETMASK, &origmask, NULL);
476
Michal Vasko0a3f3752016-10-13 14:58:38 +0200477 if (!ret) {
478 /* we timeouted */
479 free(pfd);
480 return 0;
481 } else if (ret == -1) {
Michal Vasko05532772021-06-03 12:12:38 +0200482 ERR(NULL, "Poll failed (%s).", strerror(errno));
Michal Vasko0a3f3752016-10-13 14:58:38 +0200483 free(pfd);
484 return -1;
485 }
Michal Vasko086311b2016-01-08 09:53:11 +0100486
Michal Vaskoac2f6182017-01-30 14:32:03 +0100487 for (i = 0, j = 0; j < pfd_count; ++i, ++j) {
488 /* adjust i so that indices in binds and pfd always match */
489 while (binds[i].sock != pfd[j].fd) {
490 ++i;
491 }
492
493 if (pfd[j].revents & POLLIN) {
Michal Vasko0a3f3752016-10-13 14:58:38 +0200494 --ret;
495
496 if (!ret) {
497 /* the last socket with an event, use it */
Michal Vaskoac2f6182017-01-30 14:32:03 +0100498 sock = pfd[j].fd;
Michal Vasko0a3f3752016-10-13 14:58:38 +0200499 break;
500 } else {
501 /* just remember the event for next time */
502 binds[i].pollin = 1;
503 }
504 }
Michal Vasko086311b2016-01-08 09:53:11 +0100505 }
506 }
507 free(pfd);
508
509 if (sock == -1) {
Michal Vaskod083db62016-01-19 10:31:29 +0100510 ERRINT;
Michal Vasko086311b2016-01-08 09:53:11 +0100511 return -1;
512 }
513
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200514 /* accept connection */
515 client_sock = accept(sock, (struct sockaddr *)&saddr, &saddr_len);
516 if (client_sock < 0) {
Michal Vasko05532772021-06-03 12:12:38 +0200517 ERR(NULL, "Accept failed (%s).", strerror(errno));
Michal Vasko086311b2016-01-08 09:53:11 +0100518 return -1;
519 }
520
Michal Vasko0190bc32016-03-02 15:47:49 +0100521 /* make the socket non-blocking */
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200522 if (((flags = fcntl(client_sock, F_GETFL)) == -1) || (fcntl(client_sock, F_SETFL, flags | O_NONBLOCK) == -1)) {
Michal Vasko05532772021-06-03 12:12:38 +0200523 ERR(NULL, "Fcntl failed (%s).", strerror(errno));
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200524 goto fail;
Michal Vasko0190bc32016-03-02 15:47:49 +0100525 }
526
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200527 /* learn information about the client end */
528 if (saddr.ss_family == AF_UNIX) {
529 if (sock_host_unix(client_sock, &client_address)) {
530 goto fail;
531 }
532 client_port = 0;
533 } else if (saddr.ss_family == AF_INET) {
534 if (sock_host_inet((struct sockaddr_in *)&saddr, &client_address, &client_port)) {
535 goto fail;
536 }
537 } else if (saddr.ss_family == AF_INET6) {
538 if (sock_host_inet6((struct sockaddr_in6 *)&saddr, &client_address, &client_port)) {
539 goto fail;
540 }
541 } else {
542 ERR(NULL, "Source host of an unknown protocol family.");
543 goto fail;
aPiecek90ff0242021-02-14 14:58:01 +0100544 }
Michal Vasko086311b2016-01-08 09:53:11 +0100545
aPiecek90ff0242021-02-14 14:58:01 +0100546 if (saddr.ss_family == AF_UNIX) {
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200547 VRB(NULL, "Accepted a connection on %s.", binds[i].address);
aPiecek90ff0242021-02-14 14:58:01 +0100548 } else {
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200549 VRB(NULL, "Accepted a connection on %s:%u from %s:%u.", binds[i].address, binds[i].port, client_address, client_port);
Michal Vasko086311b2016-01-08 09:53:11 +0100550 }
551
Michal Vasko89ffa8a2021-06-25 08:40:08 +0200552 if (host) {
553 *host = client_address;
554 } else {
555 free(client_address);
556 }
557 if (port) {
558 *port = client_port;
559 }
560 if (idx) {
561 *idx = i;
562 }
563 return client_sock;
564
565fail:
566 close(client_sock);
567 return -1;
Michal Vasko086311b2016-01-08 09:53:11 +0100568}
569
Michal Vasko238b6c12021-12-14 15:14:09 +0100570API struct nc_server_reply *
Michal Vasko05532772021-06-03 12:12:38 +0200571nc_clb_default_get_schema(struct lyd_node *rpc, struct nc_session *session)
Michal Vasko05ba9df2016-01-13 14:40:27 +0100572{
Michal Vasko77367452021-02-16 16:32:18 +0100573 const char *identifier = NULL, *revision = NULL, *format = NULL;
Michal Vasko05ba9df2016-01-13 14:40:27 +0100574 char *model_data = NULL;
Michal Vasko77367452021-02-16 16:32:18 +0100575 struct ly_out *out;
Michal Vasko9b1a9522021-03-15 16:24:26 +0100576 const struct lys_module *module = NULL, *mod;
Michal Vasko77367452021-02-16 16:32:18 +0100577 const struct lysp_submodule *submodule = NULL;
578 struct lyd_node *child, *err, *data = NULL;
579 LYS_OUTFORMAT outformat = 0;
Michal Vasko05ba9df2016-01-13 14:40:27 +0100580
Michal Vasko77367452021-02-16 16:32:18 +0100581 LY_LIST_FOR(lyd_child(rpc), child) {
Michal Vasko05ba9df2016-01-13 14:40:27 +0100582 if (!strcmp(child->schema->name, "identifier")) {
Michal Vaskoe97b10a2021-04-28 08:52:52 +0200583 identifier = lyd_get_value(child);
Michal Vasko05ba9df2016-01-13 14:40:27 +0100584 } else if (!strcmp(child->schema->name, "version")) {
Michal Vaskoe97b10a2021-04-28 08:52:52 +0200585 revision = lyd_get_value(child);
Michal Vaskob83a3fa2021-05-26 09:53:42 +0200586 if (revision && (revision[0] == '\0')) {
Michal Vasko77367452021-02-16 16:32:18 +0100587 revision = NULL;
Radek Krejci1afa7792017-03-26 11:24:16 -0500588 }
Michal Vasko05ba9df2016-01-13 14:40:27 +0100589 } else if (!strcmp(child->schema->name, "format")) {
Michal Vaskoe97b10a2021-04-28 08:52:52 +0200590 format = lyd_get_value(child);
Michal Vasko05ba9df2016-01-13 14:40:27 +0100591 }
592 }
Michal Vasko5ca5d972022-09-14 13:51:31 +0200593 VRB(session, "Module \"%s@%s\" was requested.", identifier, revision ? revision : "<any>");
Michal Vasko05ba9df2016-01-13 14:40:27 +0100594
Michal Vasko77367452021-02-16 16:32:18 +0100595 /* check revision */
596 if (revision && (strlen(revision) != 10) && strcmp(revision, "1.0")) {
Michal Vasko93224072021-11-09 12:14:28 +0100597 err = nc_err(session->ctx, NC_ERR_INVALID_VALUE, NC_ERR_TYPE_APP);
Michal Vasko1a38c862016-01-15 15:50:07 +0100598 nc_err_set_msg(err, "The requested version is not supported.", "en");
599 return nc_server_reply_err(err);
Michal Vasko05ba9df2016-01-13 14:40:27 +0100600 }
601
Michal Vasko77367452021-02-16 16:32:18 +0100602 if (revision) {
603 /* get specific module */
Michal Vasko93224072021-11-09 12:14:28 +0100604 module = ly_ctx_get_module(session->ctx, identifier, revision);
Michal Vasko77367452021-02-16 16:32:18 +0100605 if (!module) {
Michal Vasko93224072021-11-09 12:14:28 +0100606 submodule = ly_ctx_get_submodule(session->ctx, identifier, revision);
Michal Vasko77367452021-02-16 16:32:18 +0100607 }
608 } else {
609 /* try to get implemented, then latest module */
Michal Vasko93224072021-11-09 12:14:28 +0100610 module = ly_ctx_get_module_implemented(session->ctx, identifier);
Michal Vasko77367452021-02-16 16:32:18 +0100611 if (!module) {
Michal Vasko93224072021-11-09 12:14:28 +0100612 module = ly_ctx_get_module_latest(session->ctx, identifier);
Michal Vasko77367452021-02-16 16:32:18 +0100613 }
614 if (!module) {
Michal Vasko93224072021-11-09 12:14:28 +0100615 submodule = ly_ctx_get_submodule_latest(session->ctx, identifier);
Michal Vasko77367452021-02-16 16:32:18 +0100616 }
Michal Vaskod91f6e62016-04-05 11:34:22 +0200617 }
Michal Vasko77367452021-02-16 16:32:18 +0100618 if (!module && !submodule) {
Michal Vasko93224072021-11-09 12:14:28 +0100619 err = nc_err(session->ctx, NC_ERR_INVALID_VALUE, NC_ERR_TYPE_APP);
Michal Vasko5ca5d972022-09-14 13:51:31 +0200620 nc_err_set_msg(err, "The requested module was not found.", "en");
Michal Vasko1a38c862016-01-15 15:50:07 +0100621 return nc_server_reply_err(err);
Michal Vasko05ba9df2016-01-13 14:40:27 +0100622 }
623
624 /* check format */
Radek Krejci90fba642016-12-07 15:59:45 +0100625 if (!format || !strcmp(format, "ietf-netconf-monitoring:yang")) {
Michal Vasko77367452021-02-16 16:32:18 +0100626 outformat = LYS_OUT_YANG;
Radek Krejci90fba642016-12-07 15:59:45 +0100627 } else if (!strcmp(format, "ietf-netconf-monitoring:yin")) {
Michal Vasko77367452021-02-16 16:32:18 +0100628 outformat = LYS_OUT_YIN;
Michal Vasko05ba9df2016-01-13 14:40:27 +0100629 } else {
Michal Vasko93224072021-11-09 12:14:28 +0100630 err = nc_err(session->ctx, NC_ERR_INVALID_VALUE, NC_ERR_TYPE_APP);
Michal Vasko1a38c862016-01-15 15:50:07 +0100631 nc_err_set_msg(err, "The requested format is not supported.", "en");
632 return nc_server_reply_err(err);
Michal Vasko05ba9df2016-01-13 14:40:27 +0100633 }
Michal Vasko77367452021-02-16 16:32:18 +0100634
635 /* print */
636 ly_out_new_memory(&model_data, 0, &out);
637 if (module) {
638 lys_print_module(out, module, outformat, 0, 0);
639 } else {
640 lys_print_submodule(out, submodule, outformat, 0, 0);
641 }
642 ly_out_free(out, NULL, 0);
Michal Vaskod91f6e62016-04-05 11:34:22 +0200643 if (!model_data) {
644 ERRINT;
645 return NULL;
646 }
Michal Vasko05ba9df2016-01-13 14:40:27 +0100647
Michal Vasko9b1a9522021-03-15 16:24:26 +0100648 /* create reply */
Michal Vasko93224072021-11-09 12:14:28 +0100649 mod = ly_ctx_get_module_implemented(session->ctx, "ietf-netconf-monitoring");
Michal Vasko9b1a9522021-03-15 16:24:26 +0100650 if (!mod || lyd_new_inner(NULL, mod, "get-schema", 0, &data)) {
Michal Vasko05ba9df2016-01-13 14:40:27 +0100651 ERRINT;
Michal Vaskod91f6e62016-04-05 11:34:22 +0200652 free(model_data);
Michal Vasko05ba9df2016-01-13 14:40:27 +0100653 return NULL;
654 }
Michal Vasko9b1a9522021-03-15 16:24:26 +0100655 if (lyd_new_any(data, NULL, "data", model_data, 1, LYD_ANYDATA_STRING, 1, NULL)) {
656 ERRINT;
Michal Vaskoa50f68e2022-02-24 16:10:54 +0100657 free(model_data);
Michal Vasko9b1a9522021-03-15 16:24:26 +0100658 lyd_free_tree(data);
659 return NULL;
660 }
Michal Vasko05ba9df2016-01-13 14:40:27 +0100661
Radek Krejci36dfdb32016-09-01 16:56:35 +0200662 return nc_server_reply_data(data, NC_WD_EXPLICIT, NC_PARAMTYPE_FREE);
Michal Vasko05ba9df2016-01-13 14:40:27 +0100663}
664
Michal Vasko238b6c12021-12-14 15:14:09 +0100665API struct nc_server_reply *
Michal Vasko428087d2016-01-14 16:04:28 +0100666nc_clb_default_close_session(struct lyd_node *UNUSED(rpc), struct nc_session *session)
Michal Vasko05ba9df2016-01-13 14:40:27 +0100667{
Michal Vasko428087d2016-01-14 16:04:28 +0100668 session->term_reason = NC_SESSION_TERM_CLOSED;
669 return nc_server_reply_ok();
Michal Vasko05ba9df2016-01-13 14:40:27 +0100670}
671
Michal Vasko93224072021-11-09 12:14:28 +0100672/**
673 * @brief Initialize a context with default RPC callbacks if none are set.
674 *
675 * @param[in] ctx Context to initialize.
676 */
677static void
678nc_server_init_ctx(const struct ly_ctx *ctx)
Michal Vasko086311b2016-01-08 09:53:11 +0100679{
Michal Vasko77367452021-02-16 16:32:18 +0100680 struct lysc_node *rpc;
Michal Vaskoa7b8ca52016-03-01 12:09:29 +0100681
Michal Vasko238b6c12021-12-14 15:14:09 +0100682 if (global_rpc_clb) {
683 /* expect it to handle these RPCs as well */
684 return;
685 }
686
Michal Vasko05ba9df2016-01-13 14:40:27 +0100687 /* set default <get-schema> callback if not specified */
Michal Vasko77367452021-02-16 16:32:18 +0100688 rpc = NULL;
689 if (ly_ctx_get_module_implemented(ctx, "ietf-netconf-monitoring")) {
690 rpc = (struct lysc_node *)lys_find_path(ctx, NULL, "/ietf-netconf-monitoring:get-schema", 0);
691 }
Michal Vasko88639e92017-08-03 14:38:10 +0200692 if (rpc && !rpc->priv) {
Michal Vasko77367452021-02-16 16:32:18 +0100693 rpc->priv = nc_clb_default_get_schema;
Michal Vasko05ba9df2016-01-13 14:40:27 +0100694 }
695
Michal Vasko93224072021-11-09 12:14:28 +0100696 /* set default <close-session> callback if not specified */
Michal Vasko77367452021-02-16 16:32:18 +0100697 rpc = (struct lysc_node *)lys_find_path(ctx, NULL, "/ietf-netconf:close-session", 0);
Michal Vasko88639e92017-08-03 14:38:10 +0200698 if (rpc && !rpc->priv) {
Michal Vasko77367452021-02-16 16:32:18 +0100699 rpc->priv = nc_clb_default_close_session;
Michal Vasko05ba9df2016-01-13 14:40:27 +0100700 }
Michal Vasko93224072021-11-09 12:14:28 +0100701}
Michal Vasko05ba9df2016-01-13 14:40:27 +0100702
Michal Vasko93224072021-11-09 12:14:28 +0100703API int
704nc_server_init(void)
705{
706 pthread_rwlockattr_t attr, *attr_p = NULL;
707 int r;
708
709 nc_init();
Michal Vaskob48aa812016-01-18 14:13:09 +0100710
711 server_opts.new_session_id = 1;
Andrew Langefeld6ed922d2018-09-12 14:08:32 -0500712 server_opts.new_client_id = 1;
Michal Vaskob48aa812016-01-18 14:13:09 +0100713
Michal Vasko93224072021-11-09 12:14:28 +0100714#ifdef HAVE_PTHREAD_RWLOCKATTR_SETKIND_NP
715 if ((r = pthread_rwlockattr_init(&attr))) {
716 ERR(NULL, "%s: failed init attribute (%s).", __func__, strerror(r));
717 goto error;
718 }
719 attr_p = &attr;
720 if ((r = pthread_rwlockattr_setkind_np(&attr, PTHREAD_RWLOCK_PREFER_WRITER_NONRECURSIVE_NP))) {
721 ERR(NULL, "%s: failed set attribute (%s).", __func__, strerror(r));
722 goto error;
723 }
Rosen Penevef2f3ac2019-07-15 18:15:28 -0700724#endif
Michal Vasko93224072021-11-09 12:14:28 +0100725
726 if ((r = pthread_rwlock_init(&server_opts.endpt_lock, attr_p))) {
727 ERR(NULL, "%s: failed to init rwlock(%s).", __func__, strerror(r));
728 goto error;
729 }
730 if ((r = pthread_rwlock_init(&server_opts.ch_client_lock, attr_p))) {
731 ERR(NULL, "%s: failed to init rwlock(%s).", __func__, strerror(r));
732 goto error;
733 }
734
735 if (attr_p) {
736 pthread_rwlockattr_destroy(attr_p);
Frank Rimpler9f838b02018-07-25 06:44:03 +0000737 }
Michal Vasko086311b2016-01-08 09:53:11 +0100738 return 0;
Michal Vasko93224072021-11-09 12:14:28 +0100739
740error:
741 if (attr_p) {
742 pthread_rwlockattr_destroy(attr_p);
743 }
744 return -1;
Michal Vasko086311b2016-01-08 09:53:11 +0100745}
746
Michal Vaskob48aa812016-01-18 14:13:09 +0100747API void
748nc_server_destroy(void)
749{
Michal Vasko1440a742021-03-31 11:11:03 +0200750 uint32_t i;
Radek Krejci658782b2016-12-04 22:04:55 +0100751
752 for (i = 0; i < server_opts.capabilities_count; i++) {
Michal Vasko93224072021-11-09 12:14:28 +0100753 free(server_opts.capabilities[i]);
Radek Krejci658782b2016-12-04 22:04:55 +0100754 }
755 free(server_opts.capabilities);
Michal Vaskodd6e4f72018-06-01 10:21:27 +0200756 server_opts.capabilities = NULL;
757 server_opts.capabilities_count = 0;
Michal Vasko1440a742021-03-31 11:11:03 +0200758 if (server_opts.content_id_data && server_opts.content_id_data_free) {
759 server_opts.content_id_data_free(server_opts.content_id_data);
760 }
Michal Vaskodd6e4f72018-06-01 10:21:27 +0200761
Michal Vaskob83a3fa2021-05-26 09:53:42 +0200762#if defined (NC_ENABLED_SSH) || defined (NC_ENABLED_TLS)
Michal Vasko59050372016-11-22 14:33:55 +0100763 nc_server_del_endpt(NULL, 0);
Michal Vasko0bdf70b2019-06-24 19:20:20 +0200764 nc_server_ch_del_client(NULL);
Michal Vaskob48aa812016-01-18 14:13:09 +0100765#endif
Michal Vasko17dfda92016-12-01 14:06:16 +0100766#ifdef NC_ENABLED_SSH
Michal Vaskoebba7602018-03-23 13:14:08 +0100767 if (server_opts.passwd_auth_data && server_opts.passwd_auth_data_free) {
768 server_opts.passwd_auth_data_free(server_opts.passwd_auth_data);
769 }
Michal Vaskodd6e4f72018-06-01 10:21:27 +0200770 server_opts.passwd_auth_data = NULL;
771 server_opts.passwd_auth_data_free = NULL;
Michal Vaskoebba7602018-03-23 13:14:08 +0100772
Michal Vasko1c2d2652023-10-17 08:53:36 +0200773 if (server_opts.pubkey_auth_data && server_opts.pubkey_auth_data_free) {
774 server_opts.pubkey_auth_data_free(server_opts.pubkey_auth_data);
775 }
776 server_opts.pubkey_auth_data = NULL;
777 server_opts.pubkey_auth_data_free = NULL;
778
779 if (server_opts.interactive_auth_sess_data && server_opts.interactive_auth_sess_data_free) {
780 server_opts.interactive_auth_sess_data_free(server_opts.interactive_auth_sess_data);
781 }
782 server_opts.interactive_auth_sess_data = NULL;
783 server_opts.interactive_auth_sess_data_free = NULL;
784
785 if (server_opts.interactive_auth_data && server_opts.interactive_auth_data_free) {
786 server_opts.interactive_auth_data_free(server_opts.interactive_auth_data);
787 }
788 server_opts.interactive_auth_data = NULL;
789 server_opts.interactive_auth_data_free = NULL;
790
Michal Vasko17dfda92016-12-01 14:06:16 +0100791 nc_server_ssh_del_authkey(NULL, NULL, 0, NULL);
Michal Vasko4c1fb492017-01-30 14:31:07 +0100792
793 if (server_opts.hostkey_data && server_opts.hostkey_data_free) {
794 server_opts.hostkey_data_free(server_opts.hostkey_data);
795 }
Michal Vaskodd6e4f72018-06-01 10:21:27 +0200796 server_opts.hostkey_data = NULL;
797 server_opts.hostkey_data_free = NULL;
roman41a11e42022-06-22 09:27:08 +0200798
799 /* PAM */
800 free(server_opts.conf_name);
801 free(server_opts.conf_dir);
802 server_opts.conf_name = NULL;
803 server_opts.conf_dir = NULL;
Michal Vasko4c1fb492017-01-30 14:31:07 +0100804#endif
805#ifdef NC_ENABLED_TLS
806 if (server_opts.server_cert_data && server_opts.server_cert_data_free) {
807 server_opts.server_cert_data_free(server_opts.server_cert_data);
808 }
Michal Vaskodd6e4f72018-06-01 10:21:27 +0200809 server_opts.server_cert_data = NULL;
810 server_opts.server_cert_data_free = NULL;
Michal Vasko4c1fb492017-01-30 14:31:07 +0100811 if (server_opts.trusted_cert_list_data && server_opts.trusted_cert_list_data_free) {
812 server_opts.trusted_cert_list_data_free(server_opts.trusted_cert_list_data);
813 }
Michal Vaskodd6e4f72018-06-01 10:21:27 +0200814 server_opts.trusted_cert_list_data = NULL;
815 server_opts.trusted_cert_list_data_free = NULL;
Michal Vaskob48aa812016-01-18 14:13:09 +0100816#endif
Michal Vaskoa7b8ca52016-03-01 12:09:29 +0100817 nc_destroy();
Michal Vaskob48aa812016-01-18 14:13:09 +0100818}
819
Michal Vasko086311b2016-01-08 09:53:11 +0100820API int
821nc_server_set_capab_withdefaults(NC_WD_MODE basic_mode, int also_supported)
822{
Michal Vasko45e53ae2016-04-07 11:46:03 +0200823 if (!basic_mode || (basic_mode == NC_WD_ALL_TAG)) {
824 ERRARG("basic_mode");
825 return -1;
826 } else if (also_supported && !(also_supported & (NC_WD_ALL | NC_WD_ALL_TAG | NC_WD_TRIM | NC_WD_EXPLICIT))) {
827 ERRARG("also_supported");
Michal Vasko086311b2016-01-08 09:53:11 +0100828 return -1;
829 }
830
831 server_opts.wd_basic_mode = basic_mode;
832 server_opts.wd_also_supported = also_supported;
833 return 0;
834}
835
Michal Vasko1a38c862016-01-15 15:50:07 +0100836API void
Michal Vasko55f03972016-04-13 08:56:01 +0200837nc_server_get_capab_withdefaults(NC_WD_MODE *basic_mode, int *also_supported)
838{
839 if (!basic_mode && !also_supported) {
840 ERRARG("basic_mode and also_supported");
841 return;
842 }
843
844 if (basic_mode) {
845 *basic_mode = server_opts.wd_basic_mode;
846 }
847 if (also_supported) {
848 *also_supported = server_opts.wd_also_supported;
849 }
850}
851
Michal Vasko55f03972016-04-13 08:56:01 +0200852API int
Radek Krejci658782b2016-12-04 22:04:55 +0100853nc_server_set_capability(const char *value)
Michal Vasko55f03972016-04-13 08:56:01 +0200854{
Michal Vasko93224072021-11-09 12:14:28 +0100855 void *mem;
Radek Krejci658782b2016-12-04 22:04:55 +0100856
857 if (!value || !value[0]) {
858 ERRARG("value must not be empty");
859 return EXIT_FAILURE;
860 }
861
Michal Vasko93224072021-11-09 12:14:28 +0100862 mem = realloc(server_opts.capabilities, (server_opts.capabilities_count + 1) * sizeof *server_opts.capabilities);
863 if (!mem) {
Radek Krejci658782b2016-12-04 22:04:55 +0100864 ERRMEM;
865 return EXIT_FAILURE;
866 }
Michal Vasko93224072021-11-09 12:14:28 +0100867 server_opts.capabilities = mem;
868
869 server_opts.capabilities[server_opts.capabilities_count] = strdup(value);
870 server_opts.capabilities_count++;
Radek Krejci658782b2016-12-04 22:04:55 +0100871
872 return EXIT_SUCCESS;
Michal Vasko55f03972016-04-13 08:56:01 +0200873}
874
Michal Vasko1a38c862016-01-15 15:50:07 +0100875API void
Michal Vasko1440a742021-03-31 11:11:03 +0200876nc_server_set_content_id_clb(char *(*content_id_clb)(void *user_data), void *user_data,
877 void (*free_user_data)(void *user_data))
878{
879 server_opts.content_id_clb = content_id_clb;
880 server_opts.content_id_data = user_data;
881 server_opts.content_id_data_free = free_user_data;
882}
883
884API void
Michal Vasko086311b2016-01-08 09:53:11 +0100885nc_server_set_hello_timeout(uint16_t hello_timeout)
886{
Michal Vasko086311b2016-01-08 09:53:11 +0100887 server_opts.hello_timeout = hello_timeout;
Michal Vasko086311b2016-01-08 09:53:11 +0100888}
889
Michal Vasko55f03972016-04-13 08:56:01 +0200890API uint16_t
891nc_server_get_hello_timeout(void)
892{
893 return server_opts.hello_timeout;
894}
895
Michal Vasko1a38c862016-01-15 15:50:07 +0100896API void
Michal Vasko086311b2016-01-08 09:53:11 +0100897nc_server_set_idle_timeout(uint16_t idle_timeout)
898{
Michal Vasko086311b2016-01-08 09:53:11 +0100899 server_opts.idle_timeout = idle_timeout;
Michal Vasko086311b2016-01-08 09:53:11 +0100900}
901
Michal Vasko55f03972016-04-13 08:56:01 +0200902API uint16_t
903nc_server_get_idle_timeout(void)
904{
905 return server_opts.idle_timeout;
906}
907
Michal Vasko71090fc2016-05-24 16:37:28 +0200908API NC_MSG_TYPE
Michal Vasko93224072021-11-09 12:14:28 +0100909nc_accept_inout(int fdin, int fdout, const char *username, const struct ly_ctx *ctx, struct nc_session **session)
Michal Vasko086311b2016-01-08 09:53:11 +0100910{
Michal Vasko71090fc2016-05-24 16:37:28 +0200911 NC_MSG_TYPE msgtype;
Michal Vasko9fb42272017-10-05 13:50:05 +0200912 struct timespec ts_cur;
Michal Vasko71090fc2016-05-24 16:37:28 +0200913
Michal Vasko93224072021-11-09 12:14:28 +0100914 if (!ctx) {
915 ERRARG("ctx");
Michal Vasko71090fc2016-05-24 16:37:28 +0200916 return NC_MSG_ERROR;
Michal Vasko45e53ae2016-04-07 11:46:03 +0200917 } else if (fdin < 0) {
918 ERRARG("fdin");
Michal Vasko71090fc2016-05-24 16:37:28 +0200919 return NC_MSG_ERROR;
Michal Vasko45e53ae2016-04-07 11:46:03 +0200920 } else if (fdout < 0) {
921 ERRARG("fdout");
Michal Vasko71090fc2016-05-24 16:37:28 +0200922 return NC_MSG_ERROR;
Michal Vasko45e53ae2016-04-07 11:46:03 +0200923 } else if (!username) {
924 ERRARG("username");
Michal Vasko71090fc2016-05-24 16:37:28 +0200925 return NC_MSG_ERROR;
Michal Vasko45e53ae2016-04-07 11:46:03 +0200926 } else if (!session) {
927 ERRARG("session");
Michal Vasko71090fc2016-05-24 16:37:28 +0200928 return NC_MSG_ERROR;
Michal Vasko086311b2016-01-08 09:53:11 +0100929 }
930
Michal Vasko93224072021-11-09 12:14:28 +0100931 /* init ctx as needed */
932 nc_server_init_ctx(ctx);
933
Michal Vasko086311b2016-01-08 09:53:11 +0100934 /* prepare session structure */
Michal Vasko131120a2018-05-29 15:44:02 +0200935 *session = nc_new_session(NC_SERVER, 0);
Michal Vasko1a38c862016-01-15 15:50:07 +0100936 if (!(*session)) {
Michal Vasko086311b2016-01-08 09:53:11 +0100937 ERRMEM;
Michal Vasko71090fc2016-05-24 16:37:28 +0200938 return NC_MSG_ERROR;
Michal Vasko086311b2016-01-08 09:53:11 +0100939 }
Michal Vasko1a38c862016-01-15 15:50:07 +0100940 (*session)->status = NC_STATUS_STARTING;
Michal Vaskoade892d2017-02-22 13:40:35 +0100941
Michal Vasko086311b2016-01-08 09:53:11 +0100942 /* transport specific data */
Michal Vasko1a38c862016-01-15 15:50:07 +0100943 (*session)->ti_type = NC_TI_FD;
944 (*session)->ti.fd.in = fdin;
945 (*session)->ti.fd.out = fdout;
Michal Vasko086311b2016-01-08 09:53:11 +0100946
Michal Vasko93224072021-11-09 12:14:28 +0100947 /* assign context */
Michal Vasko1a38c862016-01-15 15:50:07 +0100948 (*session)->flags = NC_SESSION_SHAREDCTX;
Michal Vasko93224072021-11-09 12:14:28 +0100949 (*session)->ctx = (struct ly_ctx *)ctx;
Michal Vasko086311b2016-01-08 09:53:11 +0100950
Michal Vaskob48aa812016-01-18 14:13:09 +0100951 /* assign new SID atomically */
Michal Vasko5bd4a3f2021-06-17 16:40:10 +0200952 (*session)->id = ATOMIC_INC_RELAXED(server_opts.new_session_id);
Michal Vaskob48aa812016-01-18 14:13:09 +0100953
Michal Vasko086311b2016-01-08 09:53:11 +0100954 /* NETCONF handshake */
Michal Vasko131120a2018-05-29 15:44:02 +0200955 msgtype = nc_handshake_io(*session);
Michal Vasko71090fc2016-05-24 16:37:28 +0200956 if (msgtype != NC_MSG_HELLO) {
957 nc_session_free(*session, NULL);
958 *session = NULL;
959 return msgtype;
Michal Vasko086311b2016-01-08 09:53:11 +0100960 }
Michal Vasko9fb42272017-10-05 13:50:05 +0200961
Michal Vaskod8a74192023-02-06 15:51:50 +0100962 nc_timeouttime_get(&ts_cur, 0);
Michal Vasko9fb42272017-10-05 13:50:05 +0200963 (*session)->opts.server.last_rpc = ts_cur.tv_sec;
Michal Vaskod8a74192023-02-06 15:51:50 +0100964 nc_realtime_get(&ts_cur);
Michal Vasko9fb42272017-10-05 13:50:05 +0200965 (*session)->opts.server.session_start = ts_cur.tv_sec;
966
Michal Vasko1a38c862016-01-15 15:50:07 +0100967 (*session)->status = NC_STATUS_RUNNING;
Michal Vasko086311b2016-01-08 09:53:11 +0100968
Michal Vasko71090fc2016-05-24 16:37:28 +0200969 return msgtype;
Michal Vasko086311b2016-01-08 09:53:11 +0100970}
Michal Vasko9e036d52016-01-08 10:49:26 +0100971
Michal Vaskob30b99c2016-07-26 11:35:43 +0200972static void
Michal Vasko74c345f2018-02-07 10:37:11 +0100973nc_ps_queue_add_id(struct nc_pollsession *ps, uint8_t *id)
974{
975 uint8_t q_last;
976
977 if (ps->queue_len == NC_PS_QUEUE_SIZE) {
978 ERRINT;
979 return;
980 }
981
982 /* get a unique queue value (by adding 1 to the last added value, if any) */
983 if (ps->queue_len) {
984 q_last = (ps->queue_begin + ps->queue_len - 1) % NC_PS_QUEUE_SIZE;
985 *id = ps->queue[q_last] + 1;
986 } else {
987 *id = 0;
988 }
989
990 /* add the id into the queue */
991 ++ps->queue_len;
992 q_last = (ps->queue_begin + ps->queue_len - 1) % NC_PS_QUEUE_SIZE;
993 ps->queue[q_last] = *id;
994}
995
996static void
Michal Vaskob30b99c2016-07-26 11:35:43 +0200997nc_ps_queue_remove_id(struct nc_pollsession *ps, uint8_t id)
998{
Michal Vasko74c345f2018-02-07 10:37:11 +0100999 uint8_t i, q_idx, found = 0;
Michal Vaskob30b99c2016-07-26 11:35:43 +02001000
1001 for (i = 0; i < ps->queue_len; ++i) {
Michal Vasko74c345f2018-02-07 10:37:11 +01001002 /* get the actual queue idx */
1003 q_idx = (ps->queue_begin + i) % NC_PS_QUEUE_SIZE;
Michal Vaskob30b99c2016-07-26 11:35:43 +02001004
1005 if (found) {
Michal Vasko74c345f2018-02-07 10:37:11 +01001006 if (ps->queue[q_idx] == id) {
Michal Vaskob30b99c2016-07-26 11:35:43 +02001007 /* another equal value, simply cannot be */
1008 ERRINT;
1009 }
Michal Vaskod8340032018-02-12 14:41:00 +01001010 if (found == 2) {
1011 /* move the following values */
1012 ps->queue[q_idx ? q_idx - 1 : NC_PS_QUEUE_SIZE - 1] = ps->queue[q_idx];
1013 }
Michal Vasko74c345f2018-02-07 10:37:11 +01001014 } else if (ps->queue[q_idx] == id) {
Michal Vaskob30b99c2016-07-26 11:35:43 +02001015 /* found our id, there can be no more equal valid values */
Michal Vaskod8340032018-02-12 14:41:00 +01001016 if (i == 0) {
1017 found = 1;
1018 } else {
1019 /* this is not okay, our id is in the middle of the queue */
1020 found = 2;
1021 }
Michal Vaskob30b99c2016-07-26 11:35:43 +02001022 }
1023 }
Michal Vaskob30b99c2016-07-26 11:35:43 +02001024 if (!found) {
1025 ERRINT;
Michal Vasko103fe632018-02-12 16:37:45 +01001026 return;
Michal Vaskob30b99c2016-07-26 11:35:43 +02001027 }
Michal Vasko74c345f2018-02-07 10:37:11 +01001028
Michal Vasko103fe632018-02-12 16:37:45 +01001029 --ps->queue_len;
Michal Vaskod8340032018-02-12 14:41:00 +01001030 if (found == 1) {
Michal Vasko103fe632018-02-12 16:37:45 +01001031 /* remove the id by moving the queue, otherwise all the values in the queue were moved */
Michal Vaskod8340032018-02-12 14:41:00 +01001032 ps->queue_begin = (ps->queue_begin + 1) % NC_PS_QUEUE_SIZE;
1033 }
Michal Vaskob30b99c2016-07-26 11:35:43 +02001034}
1035
Michal Vaskof04a52a2016-04-07 10:52:10 +02001036int
Michal Vasko26043172016-07-26 14:08:59 +02001037nc_ps_lock(struct nc_pollsession *ps, uint8_t *id, const char *func)
Michal Vaskobe86fe32016-04-07 10:43:03 +02001038{
1039 int ret;
Michal Vaskobe86fe32016-04-07 10:43:03 +02001040 struct timespec ts;
1041
Michal Vaskobe86fe32016-04-07 10:43:03 +02001042 /* LOCK */
Michal Vasko8c7def52023-06-06 14:48:56 +02001043 ret = pthread_mutex_lock(&ps->lock);
Michal Vaskobe86fe32016-04-07 10:43:03 +02001044 if (ret) {
Michal Vasko05532772021-06-03 12:12:38 +02001045 ERR(NULL, "%s: failed to lock a pollsession (%s).", func, strerror(ret));
Michal Vaskobe86fe32016-04-07 10:43:03 +02001046 return -1;
1047 }
1048
Michal Vasko74c345f2018-02-07 10:37:11 +01001049 /* check that the queue is long enough */
Michal Vasko8bc747c2018-02-09 16:37:04 +01001050 if (ps->queue_len == NC_PS_QUEUE_SIZE) {
Michal Vasko05532772021-06-03 12:12:38 +02001051 ERR(NULL, "%s: pollsession queue size (%d) too small.", func, NC_PS_QUEUE_SIZE);
Michal Vasko8bc747c2018-02-09 16:37:04 +01001052 pthread_mutex_unlock(&ps->lock);
1053 return -1;
1054 }
Michal Vasko74c345f2018-02-07 10:37:11 +01001055
1056 /* add ourselves into the queue */
1057 nc_ps_queue_add_id(ps, id);
Michal Vaskofdba4a32022-01-05 12:13:53 +01001058 DBL(NULL, "PS 0x%p TID %lu queue: added %u, head %u, length %u", ps, (long unsigned int)pthread_self(), *id,
Michal Vasko91290952019-09-27 11:30:55 +02001059 ps->queue[ps->queue_begin], ps->queue_len);
Michal Vaskobe86fe32016-04-07 10:43:03 +02001060
1061 /* is it our turn? */
Michal Vaskob30b99c2016-07-26 11:35:43 +02001062 while (ps->queue[ps->queue_begin] != *id) {
Michal Vaskod8a74192023-02-06 15:51:50 +01001063 nc_timeouttime_get(&ts, NC_PS_QUEUE_TIMEOUT);
Michal Vaskobe86fe32016-04-07 10:43:03 +02001064
Michal Vaskod8a74192023-02-06 15:51:50 +01001065 ret = pthread_cond_clockwait(&ps->cond, &ps->lock, COMPAT_CLOCK_ID, &ts);
Michal Vaskobe86fe32016-04-07 10:43:03 +02001066 if (ret) {
preetbhansalif3edf492018-12-14 17:53:32 +05301067 /**
1068 * This may happen when another thread releases the lock and broadcasts the condition
1069 * and this thread had already timed out. When this thread is scheduled, it returns timed out error
1070 * but when actually this thread was ready for condition.
1071 */
preetbhansali629dfc42018-12-17 16:04:40 +05301072 if ((ETIMEDOUT == ret) && (ps->queue[ps->queue_begin] == *id)) {
preetbhansalif3edf492018-12-14 17:53:32 +05301073 break;
1074 }
Michal Vasko66032bc2019-01-22 15:03:12 +01001075
Michal Vasko05532772021-06-03 12:12:38 +02001076 ERR(NULL, "%s: failed to wait for a pollsession condition (%s).", func, strerror(ret));
Michal Vaskobe86fe32016-04-07 10:43:03 +02001077 /* remove ourselves from the queue */
Michal Vaskob30b99c2016-07-26 11:35:43 +02001078 nc_ps_queue_remove_id(ps, *id);
1079 pthread_mutex_unlock(&ps->lock);
Michal Vaskobe86fe32016-04-07 10:43:03 +02001080 return -1;
1081 }
1082 }
1083
Michal Vaskobe86fe32016-04-07 10:43:03 +02001084 /* UNLOCK */
1085 pthread_mutex_unlock(&ps->lock);
1086
1087 return 0;
1088}
1089
Michal Vaskof04a52a2016-04-07 10:52:10 +02001090int
Michal Vasko26043172016-07-26 14:08:59 +02001091nc_ps_unlock(struct nc_pollsession *ps, uint8_t id, const char *func)
Michal Vaskobe86fe32016-04-07 10:43:03 +02001092{
1093 int ret;
Michal Vaskobe86fe32016-04-07 10:43:03 +02001094
1095 /* LOCK */
Michal Vasko8c7def52023-06-06 14:48:56 +02001096 ret = pthread_mutex_lock(&ps->lock);
Michal Vaskobe86fe32016-04-07 10:43:03 +02001097 if (ret) {
Michal Vasko05532772021-06-03 12:12:38 +02001098 ERR(NULL, "%s: failed to lock a pollsession (%s).", func, strerror(ret));
Michal Vaskobe86fe32016-04-07 10:43:03 +02001099 ret = -1;
1100 }
1101
Michal Vaskob30b99c2016-07-26 11:35:43 +02001102 /* we must be the first, it was our turn after all, right? */
1103 if (ps->queue[ps->queue_begin] != id) {
1104 ERRINT;
Michal Vaskob1a094b2016-10-05 14:04:52 +02001105 /* UNLOCK */
1106 if (!ret) {
1107 pthread_mutex_unlock(&ps->lock);
1108 }
Michal Vaskob30b99c2016-07-26 11:35:43 +02001109 return -1;
1110 }
1111
Michal Vaskobe86fe32016-04-07 10:43:03 +02001112 /* remove ourselves from the queue */
Michal Vaskob30b99c2016-07-26 11:35:43 +02001113 nc_ps_queue_remove_id(ps, id);
Michal Vaskofdba4a32022-01-05 12:13:53 +01001114 DBL(NULL, "PS 0x%p TID %lu queue: removed %u, head %u, length %u", ps, (long unsigned int)pthread_self(), id,
Michal Vasko91290952019-09-27 11:30:55 +02001115 ps->queue[ps->queue_begin], ps->queue_len);
Michal Vaskobe86fe32016-04-07 10:43:03 +02001116
1117 /* broadcast to all other threads that the queue moved */
1118 pthread_cond_broadcast(&ps->cond);
1119
Michal Vaskobe86fe32016-04-07 10:43:03 +02001120 /* UNLOCK */
1121 if (!ret) {
1122 pthread_mutex_unlock(&ps->lock);
1123 }
1124
1125 return ret;
1126}
1127
Michal Vasko428087d2016-01-14 16:04:28 +01001128API struct nc_pollsession *
1129nc_ps_new(void)
1130{
Michal Vasko48a63ed2016-03-01 09:48:21 +01001131 struct nc_pollsession *ps;
1132
1133 ps = calloc(1, sizeof(struct nc_pollsession));
Michal Vasko4eb3c312016-03-01 14:09:37 +01001134 if (!ps) {
1135 ERRMEM;
1136 return NULL;
1137 }
Michal Vaskobe86fe32016-04-07 10:43:03 +02001138 pthread_cond_init(&ps->cond, NULL);
Michal Vasko48a63ed2016-03-01 09:48:21 +01001139 pthread_mutex_init(&ps->lock, NULL);
1140
1141 return ps;
Michal Vasko428087d2016-01-14 16:04:28 +01001142}
1143
1144API void
1145nc_ps_free(struct nc_pollsession *ps)
1146{
fanchanghu3d4e7212017-08-09 09:42:30 +08001147 uint16_t i;
1148
Michal Vasko7f1c78b2016-01-19 09:52:14 +01001149 if (!ps) {
1150 return;
1151 }
1152
Michal Vaskobe86fe32016-04-07 10:43:03 +02001153 if (ps->queue_len) {
Michal Vasko05532772021-06-03 12:12:38 +02001154 ERR(NULL, "FATAL: Freeing a pollsession structure that is currently being worked with!");
Michal Vaskobe86fe32016-04-07 10:43:03 +02001155 }
1156
fanchanghu3d4e7212017-08-09 09:42:30 +08001157 for (i = 0; i < ps->session_count; i++) {
1158 free(ps->sessions[i]);
1159 }
1160
Michal Vasko428087d2016-01-14 16:04:28 +01001161 free(ps->sessions);
Michal Vasko48a63ed2016-03-01 09:48:21 +01001162 pthread_mutex_destroy(&ps->lock);
Michal Vaskobe86fe32016-04-07 10:43:03 +02001163 pthread_cond_destroy(&ps->cond);
Michal Vasko48a63ed2016-03-01 09:48:21 +01001164
Michal Vasko428087d2016-01-14 16:04:28 +01001165 free(ps);
1166}
1167
1168API int
1169nc_ps_add_session(struct nc_pollsession *ps, struct nc_session *session)
1170{
Michal Vaskob30b99c2016-07-26 11:35:43 +02001171 uint8_t q_id;
1172
Michal Vasko45e53ae2016-04-07 11:46:03 +02001173 if (!ps) {
1174 ERRARG("ps");
1175 return -1;
1176 } else if (!session) {
1177 ERRARG("session");
Michal Vasko428087d2016-01-14 16:04:28 +01001178 return -1;
1179 }
1180
Michal Vasko48a63ed2016-03-01 09:48:21 +01001181 /* LOCK */
Michal Vasko26043172016-07-26 14:08:59 +02001182 if (nc_ps_lock(ps, &q_id, __func__)) {
Michal Vaskobe86fe32016-04-07 10:43:03 +02001183 return -1;
1184 }
Michal Vasko48a63ed2016-03-01 09:48:21 +01001185
Michal Vasko428087d2016-01-14 16:04:28 +01001186 ++ps->session_count;
Michal Vasko4eb3c312016-03-01 14:09:37 +01001187 ps->sessions = nc_realloc(ps->sessions, ps->session_count * sizeof *ps->sessions);
Michal Vasko9a327362017-01-11 11:31:46 +01001188 if (!ps->sessions) {
Michal Vasko4eb3c312016-03-01 14:09:37 +01001189 ERRMEM;
1190 /* UNLOCK */
Michal Vasko26043172016-07-26 14:08:59 +02001191 nc_ps_unlock(ps, q_id, __func__);
Michal Vasko4eb3c312016-03-01 14:09:37 +01001192 return -1;
1193 }
fanchanghu3d4e7212017-08-09 09:42:30 +08001194 ps->sessions[ps->session_count - 1] = calloc(1, sizeof **ps->sessions);
1195 if (!ps->sessions[ps->session_count - 1]) {
1196 ERRMEM;
1197 --ps->session_count;
1198 /* UNLOCK */
1199 nc_ps_unlock(ps, q_id, __func__);
1200 return -1;
1201 }
1202 ps->sessions[ps->session_count - 1]->session = session;
1203 ps->sessions[ps->session_count - 1]->state = NC_PS_STATE_NONE;
Michal Vasko428087d2016-01-14 16:04:28 +01001204
Michal Vasko48a63ed2016-03-01 09:48:21 +01001205 /* UNLOCK */
Michal Vasko26043172016-07-26 14:08:59 +02001206 return nc_ps_unlock(ps, q_id, __func__);
Michal Vasko428087d2016-01-14 16:04:28 +01001207}
1208
Michal Vasko48a63ed2016-03-01 09:48:21 +01001209static int
Radek Krejcid5f978f2016-03-03 13:14:45 +01001210_nc_ps_del_session(struct nc_pollsession *ps, struct nc_session *session, int index)
Michal Vasko428087d2016-01-14 16:04:28 +01001211{
1212 uint16_t i;
1213
Radek Krejcid5f978f2016-03-03 13:14:45 +01001214 if (index >= 0) {
1215 i = (uint16_t)index;
1216 goto remove;
1217 }
Michal Vasko428087d2016-01-14 16:04:28 +01001218 for (i = 0; i < ps->session_count; ++i) {
fanchanghu3d4e7212017-08-09 09:42:30 +08001219 if (ps->sessions[i]->session == session) {
Radek Krejcid5f978f2016-03-03 13:14:45 +01001220remove:
Michal Vasko428087d2016-01-14 16:04:28 +01001221 --ps->session_count;
fanchanghu3d4e7212017-08-09 09:42:30 +08001222 if (i <= ps->session_count) {
1223 free(ps->sessions[i]);
Michal Vasko58005732016-02-02 15:50:52 +01001224 ps->sessions[i] = ps->sessions[ps->session_count];
fanchanghu3d4e7212017-08-09 09:42:30 +08001225 }
1226 if (!ps->session_count) {
Michal Vasko58005732016-02-02 15:50:52 +01001227 free(ps->sessions);
1228 ps->sessions = NULL;
Michal Vasko58005732016-02-02 15:50:52 +01001229 }
Michal Vasko11d2f6a2017-02-02 11:15:06 +01001230 ps->last_event_session = 0;
Michal Vasko428087d2016-01-14 16:04:28 +01001231 return 0;
1232 }
1233 }
1234
Michal Vaskof0537d82016-01-29 14:42:38 +01001235 return -1;
Michal Vasko428087d2016-01-14 16:04:28 +01001236}
1237
Michal Vasko48a63ed2016-03-01 09:48:21 +01001238API int
1239nc_ps_del_session(struct nc_pollsession *ps, struct nc_session *session)
1240{
Michal Vaskob30b99c2016-07-26 11:35:43 +02001241 uint8_t q_id;
Michal Vaskobe86fe32016-04-07 10:43:03 +02001242 int ret, ret2;
Michal Vasko48a63ed2016-03-01 09:48:21 +01001243
Michal Vasko45e53ae2016-04-07 11:46:03 +02001244 if (!ps) {
1245 ERRARG("ps");
1246 return -1;
1247 } else if (!session) {
1248 ERRARG("session");
Michal Vasko48a63ed2016-03-01 09:48:21 +01001249 return -1;
1250 }
1251
1252 /* LOCK */
Michal Vasko26043172016-07-26 14:08:59 +02001253 if (nc_ps_lock(ps, &q_id, __func__)) {
Michal Vaskobe86fe32016-04-07 10:43:03 +02001254 return -1;
1255 }
Michal Vasko48a63ed2016-03-01 09:48:21 +01001256
Radek Krejcid5f978f2016-03-03 13:14:45 +01001257 ret = _nc_ps_del_session(ps, session, -1);
Michal Vasko48a63ed2016-03-01 09:48:21 +01001258
1259 /* UNLOCK */
Michal Vasko26043172016-07-26 14:08:59 +02001260 ret2 = nc_ps_unlock(ps, q_id, __func__);
Michal Vasko48a63ed2016-03-01 09:48:21 +01001261
Michal Vaskob83a3fa2021-05-26 09:53:42 +02001262 return ret || ret2 ? -1 : 0;
Michal Vasko48a63ed2016-03-01 09:48:21 +01001263}
1264
Michal Vaskoe1ee05b2017-03-21 10:10:18 +01001265API struct nc_session *
Michal Vasko4871c9d2017-10-09 14:48:39 +02001266nc_ps_get_session(const struct nc_pollsession *ps, uint16_t idx)
Michal Vaskoe1ee05b2017-03-21 10:10:18 +01001267{
1268 uint8_t q_id;
Michal Vaskoe1ee05b2017-03-21 10:10:18 +01001269 struct nc_session *ret = NULL;
1270
1271 if (!ps) {
1272 ERRARG("ps");
1273 return NULL;
1274 }
1275
1276 /* LOCK */
1277 if (nc_ps_lock((struct nc_pollsession *)ps, &q_id, __func__)) {
1278 return NULL;
1279 }
1280
Michal Vasko4871c9d2017-10-09 14:48:39 +02001281 if (idx < ps->session_count) {
1282 ret = ps->sessions[idx]->session;
Michal Vaskoe1ee05b2017-03-21 10:10:18 +01001283 }
1284
1285 /* UNLOCK */
1286 nc_ps_unlock((struct nc_pollsession *)ps, q_id, __func__);
1287
1288 return ret;
1289}
1290
Michal Vasko3ec3b112022-07-21 12:32:33 +02001291API struct nc_session *
1292nc_ps_find_session(const struct nc_pollsession *ps, nc_ps_session_match_cb match_cb, void *cb_data)
1293{
1294 uint8_t q_id;
1295 uint16_t i;
1296 struct nc_session *ret = NULL;
1297
1298 if (!ps) {
1299 ERRARG("ps");
1300 return NULL;
1301 }
1302
1303 /* LOCK */
1304 if (nc_ps_lock((struct nc_pollsession *)ps, &q_id, __func__)) {
1305 return NULL;
1306 }
1307
1308 for (i = 0; i < ps->session_count; ++i) {
1309 if (match_cb(ps->sessions[i]->session, cb_data)) {
1310 ret = ps->sessions[i]->session;
1311 break;
1312 }
1313 }
1314
1315 /* UNLOCK */
1316 nc_ps_unlock((struct nc_pollsession *)ps, q_id, __func__);
1317
1318 return ret;
1319}
1320
Michal Vasko0fdb7ac2016-03-01 09:03:12 +01001321API uint16_t
1322nc_ps_session_count(struct nc_pollsession *ps)
1323{
Michal Vasko47003942019-03-14 12:25:23 +01001324 uint8_t q_id;
1325 uint16_t session_count;
1326
Michal Vasko0fdb7ac2016-03-01 09:03:12 +01001327 if (!ps) {
Michal Vasko45e53ae2016-04-07 11:46:03 +02001328 ERRARG("ps");
Michal Vasko0fdb7ac2016-03-01 09:03:12 +01001329 return 0;
1330 }
1331
Michal Vasko47003942019-03-14 12:25:23 +01001332 /* LOCK (just for memory barrier so that we read the current value) */
1333 if (nc_ps_lock((struct nc_pollsession *)ps, &q_id, __func__)) {
1334 return 0;
1335 }
1336
1337 session_count = ps->session_count;
1338
1339 /* UNLOCK */
1340 nc_ps_unlock((struct nc_pollsession *)ps, q_id, __func__);
1341
1342 return session_count;
Michal Vasko0fdb7ac2016-03-01 09:03:12 +01001343}
1344
Michal Vasko77e83572022-07-21 15:31:15 +02001345static NC_MSG_TYPE
1346recv_rpc_check_msgid(struct nc_session *session, const struct lyd_node *envp)
1347{
1348 struct lyd_attr *attr;
1349
1350 assert(envp && !envp->schema);
1351
1352 /* find the message-id attribute */
1353 LY_LIST_FOR(((struct lyd_node_opaq *)envp)->attr, attr) {
1354 if (!strcmp(attr->name.name, "message-id")) {
1355 break;
1356 }
1357 }
1358
1359 if (!attr) {
1360 ERR(session, "Received an <rpc> without a message-id.");
1361 return NC_MSG_REPLY_ERR_MSGID;
1362 }
1363
1364 return NC_MSG_RPC;
1365}
1366
Michal Vasko131120a2018-05-29 15:44:02 +02001367/* should be called holding the session RPC lock! IO lock will be acquired as needed
Michal Vasko71090fc2016-05-24 16:37:28 +02001368 * returns: NC_PSPOLL_ERROR,
Michal Vasko77367452021-02-16 16:32:18 +01001369 * NC_PSPOLL_TIMEOUT,
Michal Vasko77e83572022-07-21 15:31:15 +02001370 * NC_PSPOLL_BAD_RPC,
Michal Vasko71090fc2016-05-24 16:37:28 +02001371 * NC_PSPOLL_RPC
1372 */
1373static int
Michal Vasko131120a2018-05-29 15:44:02 +02001374nc_server_recv_rpc_io(struct nc_session *session, int io_timeout, struct nc_server_rpc **rpc)
Michal Vasko428087d2016-01-14 16:04:28 +01001375{
Michal Vasko77367452021-02-16 16:32:18 +01001376 struct ly_in *msg;
Radek Krejcif93c7d42016-04-06 13:41:15 +02001377 struct nc_server_reply *reply = NULL;
Michal Vasko939ffce2021-04-12 13:02:01 +02001378 struct lyd_node *e;
Michal Vasko77367452021-02-16 16:32:18 +01001379 int r, ret;
Michal Vasko428087d2016-01-14 16:04:28 +01001380
Michal Vasko45e53ae2016-04-07 11:46:03 +02001381 if (!session) {
1382 ERRARG("session");
Michal Vasko71090fc2016-05-24 16:37:28 +02001383 return NC_PSPOLL_ERROR;
Michal Vasko45e53ae2016-04-07 11:46:03 +02001384 } else if (!rpc) {
1385 ERRARG("rpc");
Michal Vasko71090fc2016-05-24 16:37:28 +02001386 return NC_PSPOLL_ERROR;
Michal Vasko428087d2016-01-14 16:04:28 +01001387 } else if ((session->status != NC_STATUS_RUNNING) || (session->side != NC_SERVER)) {
Michal Vasko05532772021-06-03 12:12:38 +02001388 ERR(session, "Invalid session to receive RPCs.");
Michal Vasko71090fc2016-05-24 16:37:28 +02001389 return NC_PSPOLL_ERROR;
Michal Vasko428087d2016-01-14 16:04:28 +01001390 }
1391
Michal Vasko93224072021-11-09 12:14:28 +01001392 *rpc = NULL;
1393
Michal Vasko77367452021-02-16 16:32:18 +01001394 /* get a message */
1395 r = nc_read_msg_io(session, io_timeout, &msg, 0);
1396 if (r == -2) {
1397 /* malformed message */
Michal Vasko77e83572022-07-21 15:31:15 +02001398 ret = NC_PSPOLL_BAD_RPC;
Michal Vasko93224072021-11-09 12:14:28 +01001399 reply = nc_server_reply_err(nc_err(session->ctx, NC_ERR_MALFORMED_MSG));
Michal Vasko77e83572022-07-21 15:31:15 +02001400 goto cleanup;
Michal Vaskob83a3fa2021-05-26 09:53:42 +02001401 }
1402 if (r == -1) {
Michal Vasko77367452021-02-16 16:32:18 +01001403 return NC_PSPOLL_ERROR;
1404 } else if (!r) {
1405 return NC_PSPOLL_TIMEOUT;
Michal Vasko428087d2016-01-14 16:04:28 +01001406 }
1407
Michal Vasko77367452021-02-16 16:32:18 +01001408 *rpc = calloc(1, sizeof **rpc);
1409 if (!*rpc) {
1410 ERRMEM;
Michal Vasko77e83572022-07-21 15:31:15 +02001411 ret = NC_PSPOLL_BAD_RPC;
Michal Vasko77367452021-02-16 16:32:18 +01001412 goto cleanup;
1413 }
1414
1415 /* parse the RPC */
Michal Vasko77e83572022-07-21 15:31:15 +02001416 if (!lyd_parse_op(session->ctx, NULL, msg, LYD_XML, LYD_TYPE_RPC_NETCONF, &(*rpc)->envp, &(*rpc)->rpc)) {
1417 /* check message-id */
1418 if (recv_rpc_check_msgid(session, (*rpc)->envp) == NC_MSG_RPC) {
1419 /* valid RPC */
1420 ret = NC_PSPOLL_RPC;
1421 } else {
1422 /* no message-id */
1423 ret = NC_PSPOLL_BAD_RPC;
1424 reply = nc_server_reply_err(nc_err(session->ctx, NC_ERR_MISSING_ATTR, NC_ERR_TYPE_RPC, "message-id", "rpc"));
1425 }
1426 } else {
Michal Vasko77367452021-02-16 16:32:18 +01001427 /* bad RPC received */
Michal Vasko77e83572022-07-21 15:31:15 +02001428 ret = NC_PSPOLL_BAD_RPC;
Michal Vasko77367452021-02-16 16:32:18 +01001429
1430 if ((*rpc)->envp) {
1431 /* at least the envelopes were parsed */
Michal Vasko93224072021-11-09 12:14:28 +01001432 e = nc_err(session->ctx, NC_ERR_OP_FAILED, NC_ERR_TYPE_APP);
1433 nc_err_set_msg(e, ly_errmsg(session->ctx), "en");
Michal Vasko939ffce2021-04-12 13:02:01 +02001434 reply = nc_server_reply_err(e);
Michal Vasko77367452021-02-16 16:32:18 +01001435 } else if (session->version == NC_VERSION_11) {
Michal Vasko93224072021-11-09 12:14:28 +01001436 /* completely malformed message, NETCONF version 1.1 defines sending error reply from
1437 * the server (RFC 6241 sec. 3) */
1438 reply = nc_server_reply_err(nc_err(session->ctx, NC_ERR_MALFORMED_MSG));
Michal Vasko77367452021-02-16 16:32:18 +01001439 }
Michal Vasko77367452021-02-16 16:32:18 +01001440 }
1441
1442cleanup:
Michal Vasko77e83572022-07-21 15:31:15 +02001443 if (reply) {
1444 /* send error reply */
1445 r = nc_write_msg_io(session, io_timeout, NC_MSG_REPLY, *rpc ? (*rpc)->envp : NULL, reply);
1446 nc_server_reply_free(reply);
1447 if (r != NC_MSG_REPLY) {
1448 ERR(session, "Failed to write reply (%s), terminating session.", nc_msgtype2str[r]);
1449 if (session->status != NC_STATUS_INVALID) {
1450 session->status = NC_STATUS_INVALID;
1451 session->term_reason = NC_SESSION_TERM_OTHER;
1452 }
1453 }
1454 }
1455
Michal Vasko77367452021-02-16 16:32:18 +01001456 ly_in_free(msg, 1);
1457 if (ret != NC_PSPOLL_RPC) {
1458 nc_server_rpc_free(*rpc);
1459 *rpc = NULL;
1460 }
Michal Vasko71090fc2016-05-24 16:37:28 +02001461 return ret;
Michal Vasko428087d2016-01-14 16:04:28 +01001462}
1463
fanchanghu966f2de2016-07-21 02:28:57 -04001464API void
1465nc_set_global_rpc_clb(nc_rpc_clb clb)
1466{
1467 global_rpc_clb = clb;
1468}
1469
Radek Krejci93e80222016-10-03 13:34:25 +02001470API NC_MSG_TYPE
1471nc_server_notif_send(struct nc_session *session, struct nc_server_notif *notif, int timeout)
1472{
Michal Vasko131120a2018-05-29 15:44:02 +02001473 NC_MSG_TYPE ret;
Radek Krejci93e80222016-10-03 13:34:25 +02001474
1475 /* check parameters */
Michal Vaskodf68e7e2022-04-21 11:04:00 +02001476 if (!session || (session->side != NC_SERVER) || !nc_session_get_notif_status(session)) {
Radek Krejci93e80222016-10-03 13:34:25 +02001477 ERRARG("session");
1478 return NC_MSG_ERROR;
Michal Vasko77367452021-02-16 16:32:18 +01001479 } else if (!notif || !notif->ntf || !notif->eventtime) {
Radek Krejci93e80222016-10-03 13:34:25 +02001480 ERRARG("notif");
1481 return NC_MSG_ERROR;
1482 }
1483
Michal Vasko131120a2018-05-29 15:44:02 +02001484 /* we do not need RPC lock for this, IO lock will be acquired properly */
1485 ret = nc_write_msg_io(session, timeout, NC_MSG_NOTIF, notif);
Michal Vasko8fe604c2020-02-10 15:25:04 +01001486 if (ret != NC_MSG_NOTIF) {
Michal Vasko05532772021-06-03 12:12:38 +02001487 ERR(session, "Failed to write notification (%s).", nc_msgtype2str[ret]);
Radek Krejci93e80222016-10-03 13:34:25 +02001488 }
Michal Vaskoade892d2017-02-22 13:40:35 +01001489
Michal Vasko131120a2018-05-29 15:44:02 +02001490 return ret;
Radek Krejci93e80222016-10-03 13:34:25 +02001491}
1492
Michal Vaskof9467762023-03-28 09:02:08 +02001493/**
1494 * @brief Send a reply acquiring IO lock as needed.
1495 * Session RPC lock must be held!
1496 *
1497 * @param[in] session Session to use.
1498 * @param[in] io_timeout Timeout to use for acquiring IO lock.
1499 * @param[in] rpc RPC to sent.
1500 * @return 0 on success.
1501 * @return Bitmask of NC_PSPOLL_ERROR (any fatal error) and NC_PSPOLL_REPLY_ERROR (reply failed to be sent).
1502 * @return NC_PSPOLL_ERROR on other errors.
Michal Vasko71090fc2016-05-24 16:37:28 +02001503 */
1504static int
Michal Vasko93224072021-11-09 12:14:28 +01001505nc_server_send_reply_io(struct nc_session *session, int io_timeout, const struct nc_server_rpc *rpc)
Michal Vasko428087d2016-01-14 16:04:28 +01001506{
1507 nc_rpc_clb clb;
1508 struct nc_server_reply *reply;
Michal Vasko77367452021-02-16 16:32:18 +01001509 const struct lysc_node *rpc_act = NULL;
1510 struct lyd_node *elem;
Michal Vasko131120a2018-05-29 15:44:02 +02001511 int ret = 0;
1512 NC_MSG_TYPE r;
Michal Vasko428087d2016-01-14 16:04:28 +01001513
Michal Vasko4a827e52016-03-03 10:59:00 +01001514 if (!rpc) {
1515 ERRINT;
Michal Vasko71090fc2016-05-24 16:37:28 +02001516 return NC_PSPOLL_ERROR;
Michal Vasko4a827e52016-03-03 10:59:00 +01001517 }
1518
Michal Vasko77367452021-02-16 16:32:18 +01001519 if (rpc->rpc->schema->nodetype == LYS_RPC) {
Michal Vasko90e8e692016-07-13 12:27:57 +02001520 /* RPC */
Michal Vasko77367452021-02-16 16:32:18 +01001521 rpc_act = rpc->rpc->schema;
fanchanghu966f2de2016-07-21 02:28:57 -04001522 } else {
Michal Vasko90e8e692016-07-13 12:27:57 +02001523 /* action */
Michal Vasko77367452021-02-16 16:32:18 +01001524 LYD_TREE_DFS_BEGIN(rpc->rpc, elem) {
Michal Vasko90e8e692016-07-13 12:27:57 +02001525 if (elem->schema->nodetype == LYS_ACTION) {
1526 rpc_act = elem->schema;
1527 break;
1528 }
Michal Vasko77367452021-02-16 16:32:18 +01001529 LYD_TREE_DFS_END(rpc->rpc, elem);
fanchanghu966f2de2016-07-21 02:28:57 -04001530 }
Michal Vasko90e8e692016-07-13 12:27:57 +02001531 if (!rpc_act) {
1532 ERRINT;
1533 return NC_PSPOLL_ERROR;
1534 }
1535 }
1536
1537 if (!rpc_act->priv) {
Petr A. Sokolov16284ca2019-02-04 09:02:40 +03001538 if (!global_rpc_clb) {
1539 /* no callback, reply with a not-implemented error */
Michal Vasko93224072021-11-09 12:14:28 +01001540 reply = nc_server_reply_err(nc_err(session->ctx, NC_ERR_OP_NOT_SUPPORTED, NC_ERR_TYPE_PROT));
Petr A. Sokolov16284ca2019-02-04 09:02:40 +03001541 } else {
Michal Vaskob83a3fa2021-05-26 09:53:42 +02001542 reply = global_rpc_clb(rpc->rpc, session);
Petr A. Sokolov16284ca2019-02-04 09:02:40 +03001543 }
Michal Vasko428087d2016-01-14 16:04:28 +01001544 } else {
Michal Vasko90e8e692016-07-13 12:27:57 +02001545 clb = (nc_rpc_clb)rpc_act->priv;
Michal Vasko77367452021-02-16 16:32:18 +01001546 reply = clb(rpc->rpc, session);
Michal Vasko428087d2016-01-14 16:04:28 +01001547 }
1548
1549 if (!reply) {
Michal Vasko93224072021-11-09 12:14:28 +01001550 reply = nc_server_reply_err(nc_err(session->ctx, NC_ERR_OP_FAILED, NC_ERR_TYPE_APP));
Michal Vasko428087d2016-01-14 16:04:28 +01001551 }
Michal Vasko77367452021-02-16 16:32:18 +01001552 r = nc_write_msg_io(session, io_timeout, NC_MSG_REPLY, rpc->envp, reply);
Michal Vasko71090fc2016-05-24 16:37:28 +02001553 if (reply->type == NC_RPL_ERROR) {
1554 ret |= NC_PSPOLL_REPLY_ERROR;
1555 }
1556 nc_server_reply_free(reply);
Michal Vasko428087d2016-01-14 16:04:28 +01001557
Michal Vasko131120a2018-05-29 15:44:02 +02001558 if (r != NC_MSG_REPLY) {
Michal Vasko15469492021-06-09 08:40:48 +02001559 ERR(session, "Failed to write reply (%s).", nc_msgtype2str[r]);
Michal Vasko71090fc2016-05-24 16:37:28 +02001560 ret |= NC_PSPOLL_ERROR;
1561 }
Michal Vasko428087d2016-01-14 16:04:28 +01001562
1563 /* special case if term_reason was set in callback, last reply was sent (needed for <close-session> if nothing else) */
1564 if ((session->status == NC_STATUS_RUNNING) && (session->term_reason != NC_SESSION_TERM_NONE)) {
1565 session->status = NC_STATUS_INVALID;
1566 }
1567
Michal Vasko71090fc2016-05-24 16:37:28 +02001568 return ret;
Michal Vasko428087d2016-01-14 16:04:28 +01001569}
1570
Michal Vaskof9467762023-03-28 09:02:08 +02001571/**
1572 * @brief Poll a session from pspoll acquiring IO lock as needed.
1573 * Session must be running and session RPC lock held!
1574 *
1575 * @param[in] session Session to use.
1576 * @param[in] io_timeout Timeout to use for acquiring IO lock.
1577 * @param[in] now_mono Current monotonic timestamp.
1578 * @param[in,out] msg Message to fill in case of an error.
1579 * @return NC_PSPOLL_RPC if some application data are available.
1580 * @return NC_PSPOLL_TIMEOUT if a timeout elapsed.
1581 * @return NC_PSPOLL_SSH_CHANNEL if a new SSH channel has been created.
1582 * @return NC_PSPOLL_SSH_MSG if just an SSH message has been processed.
1583 * @return NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR if session has been terminated (@p msg filled).
1584 * @return NC_PSPOLL_ERROR on other fatal errors (@p msg filled).
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001585 */
1586static int
Michal Vasko131120a2018-05-29 15:44:02 +02001587nc_ps_poll_session_io(struct nc_session *session, int io_timeout, time_t now_mono, char *msg)
Michal Vasko428087d2016-01-14 16:04:28 +01001588{
Michal Vasko9a327362017-01-11 11:31:46 +01001589 struct pollfd pfd;
Michal Vasko2260f552018-06-06 10:06:12 +02001590 int r, ret = 0;
Michal Vaskob83a3fa2021-05-26 09:53:42 +02001591
Michal Vasko9a327362017-01-11 11:31:46 +01001592#ifdef NC_ENABLED_SSH
1593 struct nc_session *new;
1594#endif
Michal Vasko428087d2016-01-14 16:04:28 +01001595
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001596 /* check timeout first */
Michal Vaskodf68e7e2022-04-21 11:04:00 +02001597 if (!(session->flags & NC_SESSION_CALLHOME) && !nc_session_get_notif_status(session) && server_opts.idle_timeout &&
Michal Vaskob83a3fa2021-05-26 09:53:42 +02001598 (now_mono >= session->opts.server.last_rpc + server_opts.idle_timeout)) {
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001599 sprintf(msg, "session idle timeout elapsed");
1600 session->status = NC_STATUS_INVALID;
1601 session->term_reason = NC_SESSION_TERM_TIMEOUT;
1602 return NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR;
1603 }
1604
Michal Vasko131120a2018-05-29 15:44:02 +02001605 r = nc_session_io_lock(session, io_timeout, __func__);
1606 if (r < 0) {
1607 sprintf(msg, "session IO lock failed to be acquired");
1608 return NC_PSPOLL_ERROR;
1609 } else if (!r) {
1610 return NC_PSPOLL_TIMEOUT;
1611 }
1612
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001613 switch (session->ti_type) {
1614#ifdef NC_ENABLED_SSH
1615 case NC_TI_LIBSSH:
1616 r = ssh_channel_poll_timeout(session->ti.libssh.channel, 0, 0);
Michal Vasko8dcaa882017-10-19 14:28:42 +02001617 if (r == SSH_EOF) {
1618 sprintf(msg, "SSH channel unexpected EOF");
1619 session->status = NC_STATUS_INVALID;
1620 session->term_reason = NC_SESSION_TERM_DROPPED;
1621 ret = NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR;
1622 } else if (r == SSH_ERROR) {
1623 sprintf(msg, "SSH channel poll error (%s)", ssh_get_error(session->ti.libssh.session));
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001624 session->status = NC_STATUS_INVALID;
1625 session->term_reason = NC_SESSION_TERM_OTHER;
1626 ret = NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR;
Michal Vasko8dcaa882017-10-19 14:28:42 +02001627 } else if (!r) {
1628 if (session->flags & NC_SESSION_SSH_NEW_MSG) {
1629 /* new SSH message */
1630 session->flags &= ~NC_SESSION_SSH_NEW_MSG;
1631 if (session->ti.libssh.next) {
1632 for (new = session->ti.libssh.next; new != session; new = new->ti.libssh.next) {
Michal Vaskob83a3fa2021-05-26 09:53:42 +02001633 if ((new->status == NC_STATUS_STARTING) && new->ti.libssh.channel &&
1634 (new->flags & NC_SESSION_SSH_SUBSYS_NETCONF)) {
Michal Vasko8dcaa882017-10-19 14:28:42 +02001635 /* new NETCONF SSH channel */
1636 ret = NC_PSPOLL_SSH_CHANNEL;
1637 break;
1638 }
1639 }
1640 if (new != session) {
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001641 break;
1642 }
1643 }
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001644
Michal Vasko8dcaa882017-10-19 14:28:42 +02001645 /* just some SSH message */
1646 ret = NC_PSPOLL_SSH_MSG;
1647 } else {
1648 ret = NC_PSPOLL_TIMEOUT;
1649 }
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001650 } else {
1651 /* we have some application data */
1652 ret = NC_PSPOLL_RPC;
1653 }
1654 break;
1655#endif
1656#ifdef NC_ENABLED_TLS
1657 case NC_TI_OPENSSL:
1658 r = SSL_pending(session->ti.tls);
1659 if (!r) {
1660 /* no data pending in the SSL buffer, poll fd */
1661 pfd.fd = SSL_get_rfd(session->ti.tls);
1662 if (pfd.fd < 0) {
1663 sprintf(msg, "internal error (%s:%d)", __FILE__, __LINE__);
1664 ret = NC_PSPOLL_ERROR;
1665 break;
1666 }
1667 pfd.events = POLLIN;
1668 pfd.revents = 0;
1669 r = poll(&pfd, 1, 0);
1670
1671 if ((r < 0) && (errno != EINTR)) {
1672 sprintf(msg, "poll failed (%s)", strerror(errno));
1673 session->status = NC_STATUS_INVALID;
1674 ret = NC_PSPOLL_ERROR;
1675 } else if (r > 0) {
1676 if (pfd.revents & (POLLHUP | POLLNVAL)) {
1677 sprintf(msg, "communication socket unexpectedly closed");
1678 session->status = NC_STATUS_INVALID;
1679 session->term_reason = NC_SESSION_TERM_DROPPED;
1680 ret = NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR;
1681 } else if (pfd.revents & POLLERR) {
1682 sprintf(msg, "communication socket error");
1683 session->status = NC_STATUS_INVALID;
1684 session->term_reason = NC_SESSION_TERM_OTHER;
1685 ret = NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR;
1686 } else {
1687 ret = NC_PSPOLL_RPC;
1688 }
1689 } else {
1690 ret = NC_PSPOLL_TIMEOUT;
1691 }
1692 } else {
1693 ret = NC_PSPOLL_RPC;
1694 }
1695 break;
1696#endif
1697 case NC_TI_FD:
Olivier Matzac7fa2f2018-10-11 10:02:04 +02001698 case NC_TI_UNIX:
1699 pfd.fd = (session->ti_type == NC_TI_FD) ? session->ti.fd.in : session->ti.unixsock.sock;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001700 pfd.events = POLLIN;
1701 pfd.revents = 0;
1702 r = poll(&pfd, 1, 0);
1703
1704 if ((r < 0) && (errno != EINTR)) {
1705 sprintf(msg, "poll failed (%s)", strerror(errno));
1706 session->status = NC_STATUS_INVALID;
1707 ret = NC_PSPOLL_ERROR;
1708 } else if (r > 0) {
1709 if (pfd.revents & (POLLHUP | POLLNVAL)) {
1710 sprintf(msg, "communication socket unexpectedly closed");
1711 session->status = NC_STATUS_INVALID;
1712 session->term_reason = NC_SESSION_TERM_DROPPED;
1713 ret = NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR;
1714 } else if (pfd.revents & POLLERR) {
1715 sprintf(msg, "communication socket error");
1716 session->status = NC_STATUS_INVALID;
1717 session->term_reason = NC_SESSION_TERM_OTHER;
1718 ret = NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR;
1719 } else {
1720 ret = NC_PSPOLL_RPC;
1721 }
1722 } else {
1723 ret = NC_PSPOLL_TIMEOUT;
1724 }
1725 break;
1726 case NC_TI_NONE:
1727 sprintf(msg, "internal error (%s:%d)", __FILE__, __LINE__);
1728 ret = NC_PSPOLL_ERROR;
1729 break;
1730 }
1731
Michal Vasko131120a2018-05-29 15:44:02 +02001732 nc_session_io_unlock(session, __func__);
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001733 return ret;
1734}
1735
1736API int
1737nc_ps_poll(struct nc_pollsession *ps, int timeout, struct nc_session **session)
1738{
Michal Vasko443faa02022-10-20 09:09:03 +02001739 int ret = NC_PSPOLL_ERROR, r;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001740 uint8_t q_id;
1741 uint16_t i, j;
1742 char msg[256];
1743 struct timespec ts_timeout, ts_cur;
1744 struct nc_session *cur_session;
fanchanghu3d4e7212017-08-09 09:42:30 +08001745 struct nc_ps_session *cur_ps_session;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001746 struct nc_server_rpc *rpc = NULL;
1747
Michal Vasko30a5d6b2017-02-15 14:29:39 +01001748 if (!ps) {
Michal Vasko45e53ae2016-04-07 11:46:03 +02001749 ERRARG("ps");
Michal Vasko71090fc2016-05-24 16:37:28 +02001750 return NC_PSPOLL_ERROR;
Michal Vasko428087d2016-01-14 16:04:28 +01001751 }
1752
Michal Vaskoade892d2017-02-22 13:40:35 +01001753 /* PS LOCK */
Michal Vasko26043172016-07-26 14:08:59 +02001754 if (nc_ps_lock(ps, &q_id, __func__)) {
Michal Vasko71090fc2016-05-24 16:37:28 +02001755 return NC_PSPOLL_ERROR;
Michal Vaskobe86fe32016-04-07 10:43:03 +02001756 }
Michal Vasko48a63ed2016-03-01 09:48:21 +01001757
Michal Vaskoade892d2017-02-22 13:40:35 +01001758 if (!ps->session_count) {
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001759 nc_ps_unlock(ps, q_id, __func__);
1760 return NC_PSPOLL_NOSESSIONS;
Michal Vaskoade892d2017-02-22 13:40:35 +01001761 }
Michal Vaskobd8ef262016-01-20 11:09:27 +01001762
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001763 /* fill timespecs */
Michal Vaskod8a74192023-02-06 15:51:50 +01001764 nc_timeouttime_get(&ts_cur, 0);
Michal Vasko36c7be82017-02-22 13:37:59 +01001765 if (timeout > -1) {
Michal Vaskod8a74192023-02-06 15:51:50 +01001766 nc_timeouttime_get(&ts_timeout, timeout);
Michal Vasko36c7be82017-02-22 13:37:59 +01001767 }
1768
1769 /* poll all the sessions one-by-one */
Michal Vasko9a327362017-01-11 11:31:46 +01001770 do {
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001771 /* loop from i to j once (all sessions) */
Michal Vasko9a327362017-01-11 11:31:46 +01001772 if (ps->last_event_session == ps->session_count - 1) {
1773 i = j = 0;
1774 } else {
1775 i = j = ps->last_event_session + 1;
Michal Vaskobd8ef262016-01-20 11:09:27 +01001776 }
Michal Vasko9a327362017-01-11 11:31:46 +01001777 do {
fanchanghu3d4e7212017-08-09 09:42:30 +08001778 cur_ps_session = ps->sessions[i];
1779 cur_session = cur_ps_session->session;
Michal Vasko428087d2016-01-14 16:04:28 +01001780
Michal Vasko131120a2018-05-29 15:44:02 +02001781 /* SESSION RPC LOCK */
1782 r = nc_session_rpc_lock(cur_session, 0, __func__);
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001783 if (r == -1) {
1784 ret = NC_PSPOLL_ERROR;
1785 } else if (r == 1) {
1786 /* no one else is currently working with the session, so we can, otherwise skip it */
Michal Vaskoc97cb162017-10-16 12:10:23 +02001787 switch (cur_ps_session->state) {
1788 case NC_PS_STATE_NONE:
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001789 if (cur_session->status == NC_STATUS_RUNNING) {
1790 /* session is fine, work with it */
fanchanghu3d4e7212017-08-09 09:42:30 +08001791 cur_ps_session->state = NC_PS_STATE_BUSY;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001792
Michal Vasko131120a2018-05-29 15:44:02 +02001793 ret = nc_ps_poll_session_io(cur_session, NC_SESSION_LOCK_TIMEOUT, ts_cur.tv_sec, msg);
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001794 switch (ret) {
1795 case NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR:
Michal Vasko05532772021-06-03 12:12:38 +02001796 ERR(cur_session, "%s.", msg);
fanchanghu3d4e7212017-08-09 09:42:30 +08001797 cur_ps_session->state = NC_PS_STATE_INVALID;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001798 break;
1799 case NC_PSPOLL_ERROR:
Michal Vasko05532772021-06-03 12:12:38 +02001800 ERR(cur_session, "%s.", msg);
fanchanghu3d4e7212017-08-09 09:42:30 +08001801 cur_ps_session->state = NC_PS_STATE_NONE;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001802 break;
1803 case NC_PSPOLL_TIMEOUT:
1804#ifdef NC_ENABLED_SSH
1805 case NC_PSPOLL_SSH_CHANNEL:
1806 case NC_PSPOLL_SSH_MSG:
1807#endif
fanchanghu3d4e7212017-08-09 09:42:30 +08001808 cur_ps_session->state = NC_PS_STATE_NONE;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001809 break;
1810 case NC_PSPOLL_RPC:
1811 /* let's keep the state busy, we are not done with this session */
1812 break;
1813 }
1814 } else {
1815 /* session is not fine, let the caller know */
1816 ret = NC_PSPOLL_SESSION_TERM;
1817 if (cur_session->term_reason != NC_SESSION_TERM_CLOSED) {
1818 ret |= NC_PSPOLL_SESSION_ERROR;
1819 }
fanchanghu3d4e7212017-08-09 09:42:30 +08001820 cur_ps_session->state = NC_PS_STATE_INVALID;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001821 }
Michal Vaskoc97cb162017-10-16 12:10:23 +02001822 break;
1823 case NC_PS_STATE_BUSY:
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001824 /* it definitely should not be busy because we have the lock */
1825 ERRINT;
Michal Vasko4992d802017-08-09 12:20:01 +02001826 ret = NC_PSPOLL_ERROR;
Michal Vaskoc97cb162017-10-16 12:10:23 +02001827 break;
1828 case NC_PS_STATE_INVALID:
1829 /* we got it locked, but it will be freed, let it be */
1830 ret = NC_PSPOLL_TIMEOUT;
1831 break;
Michal Vasko428087d2016-01-14 16:04:28 +01001832 }
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001833
Michal Vasko131120a2018-05-29 15:44:02 +02001834 /* keep RPC lock in this one case */
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001835 if (ret != NC_PSPOLL_RPC) {
Michal Vasko131120a2018-05-29 15:44:02 +02001836 /* SESSION RPC UNLOCK */
1837 nc_session_rpc_unlock(cur_session, NC_SESSION_LOCK_TIMEOUT, __func__);
Michal Vaskoade892d2017-02-22 13:40:35 +01001838 }
Michal Vaskoade892d2017-02-22 13:40:35 +01001839 } else {
1840 /* timeout */
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001841 ret = NC_PSPOLL_TIMEOUT;
Michal Vasko428087d2016-01-14 16:04:28 +01001842 }
Michal Vasko428087d2016-01-14 16:04:28 +01001843
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001844 /* something happened */
1845 if (ret != NC_PSPOLL_TIMEOUT) {
1846 break;
1847 }
1848
Michal Vasko9a327362017-01-11 11:31:46 +01001849 if (i == ps->session_count - 1) {
1850 i = 0;
1851 } else {
1852 ++i;
1853 }
1854 } while (i != j);
1855
Michal Vaskoade892d2017-02-22 13:40:35 +01001856 /* no event, no session remains locked */
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001857 if (ret == NC_PSPOLL_TIMEOUT) {
Michal Vasko9a327362017-01-11 11:31:46 +01001858 usleep(NC_TIMEOUT_STEP);
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001859
Michal Vaskod8a74192023-02-06 15:51:50 +01001860 if ((timeout > -1) && (nc_timeouttime_cur_diff(&ts_timeout) < 1)) {
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001861 /* final timeout */
1862 break;
Michal Vasko9a327362017-01-11 11:31:46 +01001863 }
Michal Vasko428087d2016-01-14 16:04:28 +01001864 }
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001865 } while (ret == NC_PSPOLL_TIMEOUT);
Michal Vasko428087d2016-01-14 16:04:28 +01001866
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001867 /* do we want to return the session? */
1868 switch (ret) {
1869 case NC_PSPOLL_RPC:
1870 case NC_PSPOLL_SESSION_TERM:
1871 case NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR:
1872#ifdef NC_ENABLED_SSH
1873 case NC_PSPOLL_SSH_CHANNEL:
1874 case NC_PSPOLL_SSH_MSG:
1875#endif
1876 if (session) {
1877 *session = cur_session;
1878 }
1879 ps->last_event_session = i;
1880 break;
1881 default:
1882 break;
Michal Vasko71090fc2016-05-24 16:37:28 +02001883 }
Michal Vasko428087d2016-01-14 16:04:28 +01001884
Michal Vaskoade892d2017-02-22 13:40:35 +01001885 /* PS UNLOCK */
1886 nc_ps_unlock(ps, q_id, __func__);
Michal Vasko428087d2016-01-14 16:04:28 +01001887
Michal Vasko131120a2018-05-29 15:44:02 +02001888 /* we have some data available and the session is RPC locked (but not IO locked) */
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001889 if (ret == NC_PSPOLL_RPC) {
Michal Vasko131120a2018-05-29 15:44:02 +02001890 ret = nc_server_recv_rpc_io(cur_session, timeout, &rpc);
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001891 if (ret & (NC_PSPOLL_ERROR | NC_PSPOLL_BAD_RPC)) {
1892 if (cur_session->status != NC_STATUS_RUNNING) {
1893 ret |= NC_PSPOLL_SESSION_TERM | NC_PSPOLL_SESSION_ERROR;
fanchanghu3d4e7212017-08-09 09:42:30 +08001894 cur_ps_session->state = NC_PS_STATE_INVALID;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001895 } else {
fanchanghu3d4e7212017-08-09 09:42:30 +08001896 cur_ps_session->state = NC_PS_STATE_NONE;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001897 }
1898 } else {
Michal Vasko9fb42272017-10-05 13:50:05 +02001899 cur_session->opts.server.last_rpc = ts_cur.tv_sec;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001900
Michal Vasko7f1ee932018-10-11 09:41:42 +02001901 /* process RPC */
Michal Vasko131120a2018-05-29 15:44:02 +02001902 ret |= nc_server_send_reply_io(cur_session, timeout, rpc);
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001903 if (cur_session->status != NC_STATUS_RUNNING) {
1904 ret |= NC_PSPOLL_SESSION_TERM;
1905 if (!(cur_session->term_reason & (NC_SESSION_TERM_CLOSED | NC_SESSION_TERM_KILLED))) {
1906 ret |= NC_PSPOLL_SESSION_ERROR;
1907 }
fanchanghu3d4e7212017-08-09 09:42:30 +08001908 cur_ps_session->state = NC_PS_STATE_INVALID;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001909 } else {
fanchanghu3d4e7212017-08-09 09:42:30 +08001910 cur_ps_session->state = NC_PS_STATE_NONE;
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001911 }
Michal Vasko428087d2016-01-14 16:04:28 +01001912 }
Michal Vasko77367452021-02-16 16:32:18 +01001913 nc_server_rpc_free(rpc);
Michal Vaskoefbc5e32017-05-26 14:02:16 +02001914
Michal Vasko131120a2018-05-29 15:44:02 +02001915 /* SESSION RPC UNLOCK */
1916 nc_session_rpc_unlock(cur_session, NC_SESSION_LOCK_TIMEOUT, __func__);
Michal Vasko428087d2016-01-14 16:04:28 +01001917 }
1918
Michal Vasko48a63ed2016-03-01 09:48:21 +01001919 return ret;
Michal Vasko428087d2016-01-14 16:04:28 +01001920}
1921
Michal Vaskod09eae62016-02-01 10:32:52 +01001922API void
Michal Vaskoe1a64ec2016-03-01 12:21:58 +01001923nc_ps_clear(struct nc_pollsession *ps, int all, void (*data_free)(void *))
Michal Vaskod09eae62016-02-01 10:32:52 +01001924{
Michal Vaskob30b99c2016-07-26 11:35:43 +02001925 uint8_t q_id;
Michal Vaskod09eae62016-02-01 10:32:52 +01001926 uint16_t i;
1927 struct nc_session *session;
1928
Michal Vasko9a25e932016-02-01 10:36:42 +01001929 if (!ps) {
Michal Vasko45e53ae2016-04-07 11:46:03 +02001930 ERRARG("ps");
Michal Vasko9a25e932016-02-01 10:36:42 +01001931 return;
1932 }
1933
Michal Vasko48a63ed2016-03-01 09:48:21 +01001934 /* LOCK */
Michal Vasko26043172016-07-26 14:08:59 +02001935 if (nc_ps_lock(ps, &q_id, __func__)) {
Michal Vaskobe86fe32016-04-07 10:43:03 +02001936 return;
1937 }
Michal Vaskod09eae62016-02-01 10:32:52 +01001938
Michal Vasko48a63ed2016-03-01 09:48:21 +01001939 if (all) {
Radek Krejci4f8042c2016-03-03 13:11:26 +01001940 for (i = 0; i < ps->session_count; i++) {
fanchanghu3d4e7212017-08-09 09:42:30 +08001941 nc_session_free(ps->sessions[i]->session, data_free);
1942 free(ps->sessions[i]);
Michal Vasko48a63ed2016-03-01 09:48:21 +01001943 }
1944 free(ps->sessions);
1945 ps->sessions = NULL;
Michal Vasko48a63ed2016-03-01 09:48:21 +01001946 ps->session_count = 0;
Michal Vasko9a327362017-01-11 11:31:46 +01001947 ps->last_event_session = 0;
Michal Vasko48a63ed2016-03-01 09:48:21 +01001948 } else {
1949 for (i = 0; i < ps->session_count; ) {
fanchanghu3d4e7212017-08-09 09:42:30 +08001950 if (ps->sessions[i]->session->status != NC_STATUS_RUNNING) {
1951 session = ps->sessions[i]->session;
Radek Krejcid5f978f2016-03-03 13:14:45 +01001952 _nc_ps_del_session(ps, NULL, i);
Michal Vaskoe1a64ec2016-03-01 12:21:58 +01001953 nc_session_free(session, data_free);
Michal Vasko48a63ed2016-03-01 09:48:21 +01001954 continue;
1955 }
1956
1957 ++i;
1958 }
Michal Vaskod09eae62016-02-01 10:32:52 +01001959 }
Michal Vasko48a63ed2016-03-01 09:48:21 +01001960
1961 /* UNLOCK */
Michal Vasko26043172016-07-26 14:08:59 +02001962 nc_ps_unlock(ps, q_id, __func__);
Michal Vaskod09eae62016-02-01 10:32:52 +01001963}
1964
Michal Vasko5f352c52019-07-10 16:12:06 +02001965static int
apropp-molex4e903c32020-04-20 03:06:58 -04001966nc_get_uid(int sock, uid_t *uid)
1967{
Michal Vaskod3910912020-04-20 09:12:49 +02001968 int ret;
apropp-molex4e903c32020-04-20 03:06:58 -04001969
Michal Vaskod3910912020-04-20 09:12:49 +02001970#ifdef SO_PEERCRED
1971 struct ucred ucred;
1972 socklen_t len;
Michal Vasko292c5542023-02-01 14:33:17 +01001973
Michal Vaskod3910912020-04-20 09:12:49 +02001974 len = sizeof(ucred);
1975 ret = getsockopt(sock, SOL_SOCKET, SO_PEERCRED, &ucred, &len);
1976 if (!ret) {
1977 *uid = ucred.uid;
1978 }
1979#else
1980 ret = getpeereid(sock, uid, NULL);
1981#endif
1982
1983 if (ret < 0) {
Michal Vasko05532772021-06-03 12:12:38 +02001984 ERR(NULL, "Failed to get credentials from unix socket (%s).", strerror(errno));
Michal Vaskod3910912020-04-20 09:12:49 +02001985 return -1;
1986 }
apropp-molex4e903c32020-04-20 03:06:58 -04001987 return 0;
1988}
1989
1990static int
Michal Vasko5f352c52019-07-10 16:12:06 +02001991nc_accept_unix(struct nc_session *session, int sock)
1992{
Michal Vaskob83a3fa2021-05-26 09:53:42 +02001993#if defined (SO_PEERCRED) || defined (HAVE_GETPEEREID)
Jan Kundrát6aa0eeb2021-10-08 21:10:05 +02001994 struct passwd *pw, pw_buf;
Michal Vasko5f352c52019-07-10 16:12:06 +02001995 char *username;
Michal Vasko292c5542023-02-01 14:33:17 +01001996
Michal Vasko5f352c52019-07-10 16:12:06 +02001997 session->ti_type = NC_TI_UNIX;
Michal Vasko143aa142021-10-01 15:31:48 +02001998 uid_t uid = 0;
Jan Kundrát6aa0eeb2021-10-08 21:10:05 +02001999 char *buf = NULL;
2000 size_t buf_len = 0;
Michal Vasko5f352c52019-07-10 16:12:06 +02002001
Michal Vaskod3910912020-04-20 09:12:49 +02002002 if (nc_get_uid(sock, &uid)) {
2003 close(sock);
Michal Vasko5f352c52019-07-10 16:12:06 +02002004 return -1;
2005 }
2006
Jan Kundrát6aa0eeb2021-10-08 21:10:05 +02002007 pw = nc_getpwuid(uid, &pw_buf, &buf, &buf_len);
Michal Vasko5f352c52019-07-10 16:12:06 +02002008 if (pw == NULL) {
Michal Vasko05532772021-06-03 12:12:38 +02002009 ERR(NULL, "Failed to find username for uid=%u (%s).\n", uid, strerror(errno));
Michal Vasko5f352c52019-07-10 16:12:06 +02002010 close(sock);
2011 return -1;
2012 }
2013
2014 username = strdup(pw->pw_name);
Michal Vaskoccd2dd02021-10-11 09:13:01 +02002015 free(buf);
Michal Vasko5f352c52019-07-10 16:12:06 +02002016 if (username == NULL) {
2017 ERRMEM;
2018 close(sock);
2019 return -1;
2020 }
Michal Vasko93224072021-11-09 12:14:28 +01002021 session->username = username;
Michal Vasko5f352c52019-07-10 16:12:06 +02002022
2023 session->ti.unixsock.sock = sock;
2024
2025 return 1;
Claus Klein22091912020-01-20 13:45:47 +01002026#else
2027 return -1;
2028#endif
Michal Vasko5f352c52019-07-10 16:12:06 +02002029}
2030
Michal Vaskoe2713da2016-08-22 16:06:40 +02002031API int
Michal Vasko2e6defd2016-10-07 15:48:15 +02002032nc_server_add_endpt(const char *name, NC_TRANSPORT_IMPL ti)
Michal Vasko9e036d52016-01-08 10:49:26 +01002033{
Michal Vasko3031aae2016-01-27 16:07:18 +01002034 uint16_t i;
Michal Vaskoade892d2017-02-22 13:40:35 +01002035 int ret = 0;
Michal Vasko9e036d52016-01-08 10:49:26 +01002036
Michal Vasko45e53ae2016-04-07 11:46:03 +02002037 if (!name) {
2038 ERRARG("name");
2039 return -1;
Michal Vasko9e036d52016-01-08 10:49:26 +01002040 }
2041
Michal Vaskoade892d2017-02-22 13:40:35 +01002042 /* BIND LOCK */
2043 pthread_mutex_lock(&server_opts.bind_lock);
2044
2045 /* ENDPT WRITE LOCK */
Michal Vasko2e6defd2016-10-07 15:48:15 +02002046 pthread_rwlock_wrlock(&server_opts.endpt_lock);
Michal Vasko3031aae2016-01-27 16:07:18 +01002047
2048 /* check name uniqueness */
2049 for (i = 0; i < server_opts.endpt_count; ++i) {
Michal Vaskoe2713da2016-08-22 16:06:40 +02002050 if (!strcmp(server_opts.endpts[i].name, name)) {
Michal Vasko05532772021-06-03 12:12:38 +02002051 ERR(NULL, "Endpoint \"%s\" already exists.", name);
Michal Vaskoade892d2017-02-22 13:40:35 +01002052 ret = -1;
2053 goto cleanup;
Michal Vasko3031aae2016-01-27 16:07:18 +01002054 }
2055 }
2056
Michal Vasko93224072021-11-09 12:14:28 +01002057 server_opts.endpts = nc_realloc(server_opts.endpts, (server_opts.endpt_count + 1) * sizeof *server_opts.endpts);
Michal Vaskoe2713da2016-08-22 16:06:40 +02002058 if (!server_opts.endpts) {
Michal Vasko4eb3c312016-03-01 14:09:37 +01002059 ERRMEM;
Michal Vaskoade892d2017-02-22 13:40:35 +01002060 ret = -1;
2061 goto cleanup;
Michal Vaskoe2713da2016-08-22 16:06:40 +02002062 }
Michal Vasko93224072021-11-09 12:14:28 +01002063 memset(&server_opts.endpts[server_opts.endpt_count], 0, sizeof *server_opts.endpts);
2064 ++server_opts.endpt_count;
2065
2066 server_opts.endpts[server_opts.endpt_count - 1].name = strdup(name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002067 server_opts.endpts[server_opts.endpt_count - 1].ti = ti;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002068 server_opts.endpts[server_opts.endpt_count - 1].ka.idle_time = 1;
2069 server_opts.endpts[server_opts.endpt_count - 1].ka.max_probes = 10;
2070 server_opts.endpts[server_opts.endpt_count - 1].ka.probe_interval = 5;
Michal Vaskoe2713da2016-08-22 16:06:40 +02002071
Michal Vaskoe2713da2016-08-22 16:06:40 +02002072 server_opts.binds = nc_realloc(server_opts.binds, server_opts.endpt_count * sizeof *server_opts.binds);
Michal Vaskoe2713da2016-08-22 16:06:40 +02002073 if (!server_opts.binds) {
2074 ERRMEM;
Michal Vaskoade892d2017-02-22 13:40:35 +01002075 ret = -1;
2076 goto cleanup;
Michal Vasko4eb3c312016-03-01 14:09:37 +01002077 }
Michal Vasko9e036d52016-01-08 10:49:26 +01002078
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002079 memset(&server_opts.binds[server_opts.endpt_count - 1], 0, sizeof *server_opts.binds);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002080 server_opts.binds[server_opts.endpt_count - 1].sock = -1;
2081
2082 switch (ti) {
Radek Krejci53691be2016-02-22 13:58:37 +01002083#ifdef NC_ENABLED_SSH
Michal Vasko2e6defd2016-10-07 15:48:15 +02002084 case NC_TI_LIBSSH:
2085 server_opts.endpts[server_opts.endpt_count - 1].opts.ssh = calloc(1, sizeof(struct nc_server_ssh_opts));
2086 if (!server_opts.endpts[server_opts.endpt_count - 1].opts.ssh) {
2087 ERRMEM;
Michal Vaskoade892d2017-02-22 13:40:35 +01002088 ret = -1;
2089 goto cleanup;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002090 }
2091 server_opts.endpts[server_opts.endpt_count - 1].opts.ssh->auth_methods =
roman41a11e42022-06-22 09:27:08 +02002092 NC_SSH_AUTH_PUBLICKEY | NC_SSH_AUTH_PASSWORD;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002093 server_opts.endpts[server_opts.endpt_count - 1].opts.ssh->auth_attempts = 3;
Michal Vaskocbad4c52019-06-27 16:30:35 +02002094 server_opts.endpts[server_opts.endpt_count - 1].opts.ssh->auth_timeout = 30;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002095 break;
Michal Vaskoe2713da2016-08-22 16:06:40 +02002096#endif
Michal Vaskoe2713da2016-08-22 16:06:40 +02002097#ifdef NC_ENABLED_TLS
Michal Vasko2e6defd2016-10-07 15:48:15 +02002098 case NC_TI_OPENSSL:
2099 server_opts.endpts[server_opts.endpt_count - 1].opts.tls = calloc(1, sizeof(struct nc_server_tls_opts));
2100 if (!server_opts.endpts[server_opts.endpt_count - 1].opts.tls) {
2101 ERRMEM;
Michal Vaskoade892d2017-02-22 13:40:35 +01002102 ret = -1;
2103 goto cleanup;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002104 }
2105 break;
Michal Vaskoe2713da2016-08-22 16:06:40 +02002106#endif
Olivier Matzac7fa2f2018-10-11 10:02:04 +02002107 case NC_TI_UNIX:
2108 server_opts.endpts[server_opts.endpt_count - 1].opts.unixsock = calloc(1, sizeof(struct nc_server_unix_opts));
2109 if (!server_opts.endpts[server_opts.endpt_count - 1].opts.unixsock) {
2110 ERRMEM;
2111 ret = -1;
2112 goto cleanup;
2113 }
2114 server_opts.endpts[server_opts.endpt_count - 1].opts.unixsock->mode = (mode_t)-1;
2115 server_opts.endpts[server_opts.endpt_count - 1].opts.unixsock->uid = (uid_t)-1;
2116 server_opts.endpts[server_opts.endpt_count - 1].opts.unixsock->gid = (gid_t)-1;
2117 break;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002118 default:
2119 ERRINT;
Michal Vaskoade892d2017-02-22 13:40:35 +01002120 ret = -1;
2121 goto cleanup;
Michal Vaskoe2713da2016-08-22 16:06:40 +02002122 }
Michal Vaskoe2713da2016-08-22 16:06:40 +02002123
Michal Vaskoade892d2017-02-22 13:40:35 +01002124cleanup:
2125 /* ENDPT UNLOCK */
Michal Vasko2e6defd2016-10-07 15:48:15 +02002126 pthread_rwlock_unlock(&server_opts.endpt_lock);
Michal Vasko9e036d52016-01-08 10:49:26 +01002127
Michal Vaskoade892d2017-02-22 13:40:35 +01002128 /* BIND UNLOCK */
2129 pthread_mutex_unlock(&server_opts.bind_lock);
Michal Vaskob48aa812016-01-18 14:13:09 +01002130
Michal Vaskoade892d2017-02-22 13:40:35 +01002131 return ret;
Michal Vasko9e036d52016-01-08 10:49:26 +01002132}
2133
Olivier Matzac7fa2f2018-10-11 10:02:04 +02002134API int
Michal Vasko59050372016-11-22 14:33:55 +01002135nc_server_del_endpt(const char *name, NC_TRANSPORT_IMPL ti)
Michal Vasko9e036d52016-01-08 10:49:26 +01002136{
2137 uint32_t i;
2138 int ret = -1;
2139
Michal Vaskoade892d2017-02-22 13:40:35 +01002140 /* BIND LOCK */
2141 pthread_mutex_lock(&server_opts.bind_lock);
Michal Vaskob48aa812016-01-18 14:13:09 +01002142
Michal Vaskoade892d2017-02-22 13:40:35 +01002143 /* ENDPT WRITE LOCK */
Michal Vasko2e6defd2016-10-07 15:48:15 +02002144 pthread_rwlock_wrlock(&server_opts.endpt_lock);
Michal Vasko9e036d52016-01-08 10:49:26 +01002145
Michal Vasko59050372016-11-22 14:33:55 +01002146 if (!name && !ti) {
Michal Vaskoe2713da2016-08-22 16:06:40 +02002147 /* remove all endpoints */
Michal Vasko3031aae2016-01-27 16:07:18 +01002148 for (i = 0; i < server_opts.endpt_count; ++i) {
Michal Vasko93224072021-11-09 12:14:28 +01002149 free(server_opts.endpts[i].name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002150 switch (server_opts.endpts[i].ti) {
Radek Krejci53691be2016-02-22 13:58:37 +01002151#ifdef NC_ENABLED_SSH
Michal Vasko2e6defd2016-10-07 15:48:15 +02002152 case NC_TI_LIBSSH:
2153 nc_server_ssh_clear_opts(server_opts.endpts[i].opts.ssh);
2154 free(server_opts.endpts[i].opts.ssh);
2155 break;
Michal Vasko3031aae2016-01-27 16:07:18 +01002156#endif
Radek Krejci53691be2016-02-22 13:58:37 +01002157#ifdef NC_ENABLED_TLS
Michal Vasko2e6defd2016-10-07 15:48:15 +02002158 case NC_TI_OPENSSL:
2159 nc_server_tls_clear_opts(server_opts.endpts[i].opts.tls);
2160 free(server_opts.endpts[i].opts.tls);
2161 break;
Michal Vasko3031aae2016-01-27 16:07:18 +01002162#endif
Olivier Matzac7fa2f2018-10-11 10:02:04 +02002163 case NC_TI_UNIX:
2164 free(server_opts.endpts[i].opts.unixsock);
2165 break;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002166 default:
2167 ERRINT;
2168 /* won't get here ...*/
2169 break;
2170 }
Michal Vasko9e036d52016-01-08 10:49:26 +01002171 ret = 0;
2172 }
Michal Vasko3031aae2016-01-27 16:07:18 +01002173 free(server_opts.endpts);
2174 server_opts.endpts = NULL;
Michal Vaskoe2713da2016-08-22 16:06:40 +02002175
2176 /* remove all binds */
2177 for (i = 0; i < server_opts.endpt_count; ++i) {
Michal Vasko93224072021-11-09 12:14:28 +01002178 free(server_opts.binds[i].address);
Michal Vaskoe2713da2016-08-22 16:06:40 +02002179 if (server_opts.binds[i].sock > -1) {
2180 close(server_opts.binds[i].sock);
2181 }
2182 }
Michal Vaskoe2713da2016-08-22 16:06:40 +02002183 free(server_opts.binds);
2184 server_opts.binds = NULL;
2185
Michal Vasko3031aae2016-01-27 16:07:18 +01002186 server_opts.endpt_count = 0;
2187
Michal Vasko1a38c862016-01-15 15:50:07 +01002188 } else {
Michal Vasko59050372016-11-22 14:33:55 +01002189 /* remove one endpoint with bind(s) or all endpoints using one transport protocol */
Michal Vasko3031aae2016-01-27 16:07:18 +01002190 for (i = 0; i < server_opts.endpt_count; ++i) {
Michal Vasko59050372016-11-22 14:33:55 +01002191 if ((name && !strcmp(server_opts.endpts[i].name, name)) || (!name && (server_opts.endpts[i].ti == ti))) {
Michal Vaskoe2713da2016-08-22 16:06:40 +02002192 /* remove endpt */
Michal Vasko93224072021-11-09 12:14:28 +01002193 free(server_opts.endpts[i].name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002194 switch (server_opts.endpts[i].ti) {
Radek Krejci53691be2016-02-22 13:58:37 +01002195#ifdef NC_ENABLED_SSH
Michal Vasko2e6defd2016-10-07 15:48:15 +02002196 case NC_TI_LIBSSH:
2197 nc_server_ssh_clear_opts(server_opts.endpts[i].opts.ssh);
2198 free(server_opts.endpts[i].opts.ssh);
2199 break;
Michal Vasko3031aae2016-01-27 16:07:18 +01002200#endif
Radek Krejci53691be2016-02-22 13:58:37 +01002201#ifdef NC_ENABLED_TLS
Michal Vasko2e6defd2016-10-07 15:48:15 +02002202 case NC_TI_OPENSSL:
2203 nc_server_tls_clear_opts(server_opts.endpts[i].opts.tls);
2204 free(server_opts.endpts[i].opts.tls);
2205 break;
Michal Vasko3031aae2016-01-27 16:07:18 +01002206#endif
Olivier Matzac7fa2f2018-10-11 10:02:04 +02002207 case NC_TI_UNIX:
2208 free(server_opts.endpts[i].opts.unixsock);
2209 break;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002210 default:
2211 ERRINT;
2212 break;
2213 }
Michal Vasko1a38c862016-01-15 15:50:07 +01002214
Michal Vaskoe2713da2016-08-22 16:06:40 +02002215 /* remove bind(s) */
Michal Vasko93224072021-11-09 12:14:28 +01002216 free(server_opts.binds[i].address);
Michal Vaskoe2713da2016-08-22 16:06:40 +02002217 if (server_opts.binds[i].sock > -1) {
2218 close(server_opts.binds[i].sock);
2219 }
2220
2221 /* move last endpt and bind(s) to the empty space */
Michal Vasko3031aae2016-01-27 16:07:18 +01002222 --server_opts.endpt_count;
Michal Vasko9ed67a72016-10-13 15:00:51 +02002223 if (!server_opts.endpt_count) {
Michal Vaskoc0256492016-02-02 12:19:06 +01002224 free(server_opts.binds);
2225 server_opts.binds = NULL;
2226 free(server_opts.endpts);
2227 server_opts.endpts = NULL;
Michal Vasko9ed67a72016-10-13 15:00:51 +02002228 } else if (i < server_opts.endpt_count) {
2229 memcpy(&server_opts.binds[i], &server_opts.binds[server_opts.endpt_count], sizeof *server_opts.binds);
2230 memcpy(&server_opts.endpts[i], &server_opts.endpts[server_opts.endpt_count], sizeof *server_opts.endpts);
Michal Vaskoc0256492016-02-02 12:19:06 +01002231 }
Michal Vasko1a38c862016-01-15 15:50:07 +01002232
2233 ret = 0;
Michal Vasko59050372016-11-22 14:33:55 +01002234 if (name) {
2235 break;
2236 }
Michal Vasko1a38c862016-01-15 15:50:07 +01002237 }
2238 }
Michal Vasko9e036d52016-01-08 10:49:26 +01002239 }
2240
Michal Vaskoade892d2017-02-22 13:40:35 +01002241 /* ENDPT UNLOCK */
Michal Vasko2e6defd2016-10-07 15:48:15 +02002242 pthread_rwlock_unlock(&server_opts.endpt_lock);
Michal Vaskob48aa812016-01-18 14:13:09 +01002243
Michal Vaskoade892d2017-02-22 13:40:35 +01002244 /* BIND UNLOCK */
2245 pthread_mutex_unlock(&server_opts.bind_lock);
Michal Vasko9e036d52016-01-08 10:49:26 +01002246
2247 return ret;
2248}
2249
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002250API int
2251nc_server_endpt_count(void)
2252{
2253 return server_opts.endpt_count;
2254}
2255
Michal Vasko1b5973e2020-01-30 16:05:46 +01002256API int
2257nc_server_is_endpt(const char *name)
2258{
2259 uint16_t i;
2260 int found = 0;
2261
Michal Vaskofb1724b2020-01-31 11:02:00 +01002262 if (!name) {
2263 return found;
2264 }
2265
Michal Vasko1b5973e2020-01-30 16:05:46 +01002266 /* ENDPT READ LOCK */
2267 pthread_rwlock_rdlock(&server_opts.endpt_lock);
2268
2269 /* check name uniqueness */
2270 for (i = 0; i < server_opts.endpt_count; ++i) {
2271 if (!strcmp(server_opts.endpts[i].name, name)) {
2272 found = 1;
2273 break;
2274 }
2275 }
2276
2277 /* ENDPT UNLOCK */
2278 pthread_rwlock_unlock(&server_opts.endpt_lock);
2279
2280 return found;
2281}
2282
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002283int
2284nc_server_endpt_set_address_port(const char *endpt_name, const char *address, uint16_t port)
2285{
2286 struct nc_endpt *endpt;
2287 struct nc_bind *bind = NULL;
2288 uint16_t i;
2289 int sock = -1, set_addr, ret = 0;
2290
2291 if (!endpt_name) {
2292 ERRARG("endpt_name");
2293 return -1;
2294 } else if ((!address && !port) || (address && port)) {
2295 ERRARG("address and port");
2296 return -1;
2297 }
2298
2299 if (address) {
2300 set_addr = 1;
2301 } else {
2302 set_addr = 0;
2303 }
2304
2305 /* BIND LOCK */
2306 pthread_mutex_lock(&server_opts.bind_lock);
2307
2308 /* ENDPT LOCK */
2309 endpt = nc_server_endpt_lock_get(endpt_name, 0, &i);
2310 if (!endpt) {
2311 /* BIND UNLOCK */
2312 pthread_mutex_unlock(&server_opts.bind_lock);
2313 return -1;
2314 }
2315
2316 bind = &server_opts.binds[i];
2317
2318 if (set_addr) {
2319 port = bind->port;
2320 } else {
2321 address = bind->address;
2322 }
2323
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002324 if (!set_addr && (endpt->ti == NC_TI_UNIX)) {
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002325 ret = -1;
2326 goto cleanup;
2327 }
2328
2329 /* we have all the information we need to create a listening socket */
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002330 if (address && (port || (endpt->ti == NC_TI_UNIX))) {
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002331 /* create new socket, close the old one */
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002332 if (endpt->ti == NC_TI_UNIX) {
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002333 sock = nc_sock_listen_unix(address, endpt->opts.unixsock);
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002334 } else {
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002335 sock = nc_sock_listen_inet(address, port, &endpt->ka);
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002336 }
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002337 if (sock == -1) {
2338 ret = -1;
2339 goto cleanup;
2340 }
2341
2342 if (bind->sock > -1) {
2343 close(bind->sock);
2344 }
2345 bind->sock = sock;
2346 } /* else we are just setting address or port */
2347
2348 if (set_addr) {
Michal Vasko93224072021-11-09 12:14:28 +01002349 free(bind->address);
2350 bind->address = strdup(address);
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002351 } else {
2352 bind->port = port;
2353 }
2354
2355 if (sock > -1) {
Michal Vasko946cacb2020-08-12 11:18:08 +02002356 switch (endpt->ti) {
2357 case NC_TI_UNIX:
Michal Vasko05532772021-06-03 12:12:38 +02002358 VRB(NULL, "Listening on %s for UNIX connections.", address);
Michal Vasko946cacb2020-08-12 11:18:08 +02002359 break;
2360#ifdef NC_ENABLED_SSH
2361 case NC_TI_LIBSSH:
Michal Vasko05532772021-06-03 12:12:38 +02002362 VRB(NULL, "Listening on %s:%u for SSH connections.", address, port);
Michal Vasko946cacb2020-08-12 11:18:08 +02002363 break;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002364#endif
Michal Vasko946cacb2020-08-12 11:18:08 +02002365#ifdef NC_ENABLED_TLS
2366 case NC_TI_OPENSSL:
Michal Vasko05532772021-06-03 12:12:38 +02002367 VRB(NULL, "Listening on %s:%u for TLS connections.", address, port);
Michal Vasko946cacb2020-08-12 11:18:08 +02002368 break;
2369#endif
2370 default:
2371 ERRINT;
2372 break;
2373 }
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002374 }
2375
2376cleanup:
2377 /* ENDPT UNLOCK */
2378 pthread_rwlock_unlock(&server_opts.endpt_lock);
2379
2380 /* BIND UNLOCK */
2381 pthread_mutex_unlock(&server_opts.bind_lock);
2382
2383 return ret;
2384}
2385
2386API int
2387nc_server_endpt_set_address(const char *endpt_name, const char *address)
2388{
2389 return nc_server_endpt_set_address_port(endpt_name, address, 0);
2390}
2391
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002392#if defined (NC_ENABLED_SSH) || defined (NC_ENABLED_TLS)
Michal Vasko946cacb2020-08-12 11:18:08 +02002393
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002394API int
2395nc_server_endpt_set_port(const char *endpt_name, uint16_t port)
2396{
2397 return nc_server_endpt_set_address_port(endpt_name, NULL, port);
2398}
2399
Michal Vasko946cacb2020-08-12 11:18:08 +02002400#endif
2401
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002402API int
2403nc_server_endpt_set_perms(const char *endpt_name, mode_t mode, uid_t uid, gid_t gid)
2404{
2405 struct nc_endpt *endpt;
2406 uint16_t i;
2407 int ret = 0;
2408
2409 if (!endpt_name) {
2410 ERRARG("endpt_name");
2411 return -1;
2412 } else if (mode == 0) {
2413 ERRARG("mode");
2414 return -1;
2415 }
2416
2417 /* ENDPT LOCK */
2418 endpt = nc_server_endpt_lock_get(endpt_name, 0, &i);
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002419 if (!endpt) {
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002420 return -1;
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002421 }
Michal Vaskoe49a15f2019-05-27 14:18:36 +02002422
2423 if (endpt->ti != NC_TI_UNIX) {
2424 ret = -1;
2425 goto cleanup;
2426 }
2427
2428 endpt->opts.unixsock->mode = mode;
2429 endpt->opts.unixsock->uid = uid;
2430 endpt->opts.unixsock->gid = gid;
2431
2432cleanup:
2433 /* ENDPT UNLOCK */
2434 pthread_rwlock_unlock(&server_opts.endpt_lock);
2435
2436 return ret;
2437}
2438
2439API int
2440nc_server_endpt_enable_keepalives(const char *endpt_name, int enable)
2441{
2442 struct nc_endpt *endpt;
2443 int ret = 0;
2444
2445 if (!endpt_name) {
2446 ERRARG("endpt_name");
2447 return -1;
2448 }
2449
2450 /* ENDPT LOCK */
2451 endpt = nc_server_endpt_lock_get(endpt_name, 0, NULL);
2452 if (!endpt) {
2453 return -1;
2454 }
2455
2456 endpt->ka.enabled = (enable ? 1 : 0);
2457
2458 /* ENDPT UNLOCK */
2459 pthread_rwlock_unlock(&server_opts.endpt_lock);
2460
2461 return ret;
2462}
2463
2464API int
2465nc_server_endpt_set_keepalives(const char *endpt_name, int idle_time, int max_probes, int probe_interval)
2466{
2467 struct nc_endpt *endpt;
2468 int ret = 0;
2469
2470 if (!endpt_name) {
2471 ERRARG("endpt_name");
2472 return -1;
2473 }
2474
2475 /* ENDPT LOCK */
2476 endpt = nc_server_endpt_lock_get(endpt_name, 0, NULL);
2477 if (!endpt) {
2478 return -1;
2479 }
2480
2481 if (idle_time > -1) {
2482 endpt->ka.idle_time = idle_time;
2483 }
2484 if (max_probes > -1) {
2485 endpt->ka.max_probes = max_probes;
2486 }
2487 if (probe_interval > -1) {
2488 endpt->ka.probe_interval = probe_interval;
2489 }
2490
2491 /* ENDPT UNLOCK */
2492 pthread_rwlock_unlock(&server_opts.endpt_lock);
2493
2494 return ret;
2495}
2496
Michal Vasko71090fc2016-05-24 16:37:28 +02002497API NC_MSG_TYPE
Michal Vasko93224072021-11-09 12:14:28 +01002498nc_accept(int timeout, const struct ly_ctx *ctx, struct nc_session **session)
Michal Vasko9e036d52016-01-08 10:49:26 +01002499{
Michal Vasko71090fc2016-05-24 16:37:28 +02002500 NC_MSG_TYPE msgtype;
Michal Vasko1a38c862016-01-15 15:50:07 +01002501 int sock, ret;
Michal Vasko5c2f7952016-01-22 13:16:31 +01002502 char *host = NULL;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002503 uint16_t port, bind_idx;
Michal Vasko9fb42272017-10-05 13:50:05 +02002504 struct timespec ts_cur;
Michal Vasko9e036d52016-01-08 10:49:26 +01002505
Michal Vasko93224072021-11-09 12:14:28 +01002506 if (!ctx) {
2507 ERRARG("ctx");
Michal Vasko71090fc2016-05-24 16:37:28 +02002508 return NC_MSG_ERROR;
Michal Vasko45e53ae2016-04-07 11:46:03 +02002509 } else if (!session) {
2510 ERRARG("session");
Michal Vasko71090fc2016-05-24 16:37:28 +02002511 return NC_MSG_ERROR;
Michal Vasko9e036d52016-01-08 10:49:26 +01002512 }
2513
Michal Vasko93224072021-11-09 12:14:28 +01002514 /* init ctx as needed */
2515 nc_server_init_ctx(ctx);
2516
Michal Vaskoade892d2017-02-22 13:40:35 +01002517 /* BIND LOCK */
2518 pthread_mutex_lock(&server_opts.bind_lock);
Michal Vasko51e514d2016-02-02 15:51:52 +01002519
2520 if (!server_opts.endpt_count) {
Michal Vasko05532772021-06-03 12:12:38 +02002521 ERR(NULL, "No endpoints to accept sessions on.");
Michal Vaskoade892d2017-02-22 13:40:35 +01002522 /* BIND UNLOCK */
2523 pthread_mutex_unlock(&server_opts.bind_lock);
Michal Vasko71090fc2016-05-24 16:37:28 +02002524 return NC_MSG_ERROR;
Michal Vasko51e514d2016-02-02 15:51:52 +01002525 }
Michal Vaskob48aa812016-01-18 14:13:09 +01002526
Michal Vaskoe2713da2016-08-22 16:06:40 +02002527 ret = nc_sock_accept_binds(server_opts.binds, server_opts.endpt_count, timeout, &host, &port, &bind_idx);
Michal Vasko50456e82016-02-02 12:16:08 +01002528 if (ret < 1) {
Michal Vaskoade892d2017-02-22 13:40:35 +01002529 /* BIND UNLOCK */
2530 pthread_mutex_unlock(&server_opts.bind_lock);
Michal Vaskob737d752016-02-09 09:01:27 +01002531 free(host);
Michal Vasko5e203472016-05-30 15:27:58 +02002532 if (!ret) {
2533 return NC_MSG_WOULDBLOCK;
2534 }
Michal Vasko71090fc2016-05-24 16:37:28 +02002535 return NC_MSG_ERROR;
Michal Vasko9e036d52016-01-08 10:49:26 +01002536 }
Michal Vaskoade892d2017-02-22 13:40:35 +01002537
2538 /* switch bind_lock for endpt_lock, so that another thread can accept another session */
2539 /* ENDPT READ LOCK */
2540 pthread_rwlock_rdlock(&server_opts.endpt_lock);
2541
2542 /* BIND UNLOCK */
2543 pthread_mutex_unlock(&server_opts.bind_lock);
2544
Michal Vaskob48aa812016-01-18 14:13:09 +01002545 sock = ret;
Michal Vasko9e036d52016-01-08 10:49:26 +01002546
Michal Vasko131120a2018-05-29 15:44:02 +02002547 *session = nc_new_session(NC_SERVER, 0);
Michal Vasko686aa312016-01-21 15:58:18 +01002548 if (!(*session)) {
Michal Vasko9e036d52016-01-08 10:49:26 +01002549 ERRMEM;
Michal Vaskoc14e3c82016-01-11 16:14:30 +01002550 close(sock);
Michal Vasko5c2f7952016-01-22 13:16:31 +01002551 free(host);
Michal Vasko71090fc2016-05-24 16:37:28 +02002552 msgtype = NC_MSG_ERROR;
2553 goto cleanup;
Michal Vasko9e036d52016-01-08 10:49:26 +01002554 }
Michal Vasko1a38c862016-01-15 15:50:07 +01002555 (*session)->status = NC_STATUS_STARTING;
Michal Vasko93224072021-11-09 12:14:28 +01002556 (*session)->ctx = (struct ly_ctx *)ctx;
Michal Vasko1a38c862016-01-15 15:50:07 +01002557 (*session)->flags = NC_SESSION_SHAREDCTX;
Michal Vasko93224072021-11-09 12:14:28 +01002558 (*session)->host = host;
Michal Vasko1a38c862016-01-15 15:50:07 +01002559 (*session)->port = port;
Michal Vasko9e036d52016-01-08 10:49:26 +01002560
Michal Vaskoc14e3c82016-01-11 16:14:30 +01002561 /* sock gets assigned to session or closed */
Radek Krejci53691be2016-02-22 13:58:37 +01002562#ifdef NC_ENABLED_SSH
Michal Vasko2e6defd2016-10-07 15:48:15 +02002563 if (server_opts.endpts[bind_idx].ti == NC_TI_LIBSSH) {
2564 (*session)->data = server_opts.endpts[bind_idx].opts.ssh;
Michal Vaskob70c8b82017-03-17 09:09:29 +01002565 ret = nc_accept_ssh_session(*session, sock, NC_TRANSPORT_TIMEOUT);
Michal Vasko71090fc2016-05-24 16:37:28 +02002566 if (ret < 0) {
2567 msgtype = NC_MSG_ERROR;
2568 goto cleanup;
2569 } else if (!ret) {
2570 msgtype = NC_MSG_WOULDBLOCK;
2571 goto cleanup;
Michal Vasko9e036d52016-01-08 10:49:26 +01002572 }
Michal Vasko3d865d22016-01-28 16:00:53 +01002573 } else
2574#endif
Radek Krejci53691be2016-02-22 13:58:37 +01002575#ifdef NC_ENABLED_TLS
Michal Vasko2e6defd2016-10-07 15:48:15 +02002576 if (server_opts.endpts[bind_idx].ti == NC_TI_OPENSSL) {
2577 (*session)->data = server_opts.endpts[bind_idx].opts.tls;
Michal Vaskob70c8b82017-03-17 09:09:29 +01002578 ret = nc_accept_tls_session(*session, sock, NC_TRANSPORT_TIMEOUT);
Michal Vasko71090fc2016-05-24 16:37:28 +02002579 if (ret < 0) {
2580 msgtype = NC_MSG_ERROR;
2581 goto cleanup;
2582 } else if (!ret) {
2583 msgtype = NC_MSG_WOULDBLOCK;
2584 goto cleanup;
Michal Vasko9e036d52016-01-08 10:49:26 +01002585 }
Michal Vasko3d865d22016-01-28 16:00:53 +01002586 } else
2587#endif
Olivier Matzac7fa2f2018-10-11 10:02:04 +02002588 if (server_opts.endpts[bind_idx].ti == NC_TI_UNIX) {
2589 (*session)->data = server_opts.endpts[bind_idx].opts.unixsock;
2590 ret = nc_accept_unix(*session, sock);
2591 if (ret < 0) {
2592 msgtype = NC_MSG_ERROR;
2593 goto cleanup;
2594 }
2595 } else {
Michal Vasko9e036d52016-01-08 10:49:26 +01002596 ERRINT;
Michal Vaskoc14e3c82016-01-11 16:14:30 +01002597 close(sock);
Michal Vasko71090fc2016-05-24 16:37:28 +02002598 msgtype = NC_MSG_ERROR;
2599 goto cleanup;
Michal Vasko9e036d52016-01-08 10:49:26 +01002600 }
2601
Michal Vasko2cc4c682016-03-01 09:16:48 +01002602 (*session)->data = NULL;
2603
Michal Vaskoade892d2017-02-22 13:40:35 +01002604 /* ENDPT UNLOCK */
Michal Vasko2e6defd2016-10-07 15:48:15 +02002605 pthread_rwlock_unlock(&server_opts.endpt_lock);
Michal Vasko3031aae2016-01-27 16:07:18 +01002606
Michal Vaskob48aa812016-01-18 14:13:09 +01002607 /* assign new SID atomically */
Michal Vasko5bd4a3f2021-06-17 16:40:10 +02002608 (*session)->id = ATOMIC_INC_RELAXED(server_opts.new_session_id);
Michal Vaskob48aa812016-01-18 14:13:09 +01002609
Michal Vasko9e036d52016-01-08 10:49:26 +01002610 /* NETCONF handshake */
Michal Vasko131120a2018-05-29 15:44:02 +02002611 msgtype = nc_handshake_io(*session);
Michal Vasko71090fc2016-05-24 16:37:28 +02002612 if (msgtype != NC_MSG_HELLO) {
Michal Vaskoe1a64ec2016-03-01 12:21:58 +01002613 nc_session_free(*session, NULL);
Michal Vasko3031aae2016-01-27 16:07:18 +01002614 *session = NULL;
Michal Vasko71090fc2016-05-24 16:37:28 +02002615 return msgtype;
Michal Vasko9e036d52016-01-08 10:49:26 +01002616 }
Michal Vasko9fb42272017-10-05 13:50:05 +02002617
Michal Vaskod8a74192023-02-06 15:51:50 +01002618 nc_timeouttime_get(&ts_cur, 0);
Michal Vasko9fb42272017-10-05 13:50:05 +02002619 (*session)->opts.server.last_rpc = ts_cur.tv_sec;
Michal Vaskod8a74192023-02-06 15:51:50 +01002620 nc_realtime_get(&ts_cur);
Michal Vasko9fb42272017-10-05 13:50:05 +02002621 (*session)->opts.server.session_start = ts_cur.tv_sec;
Michal Vasko1a38c862016-01-15 15:50:07 +01002622 (*session)->status = NC_STATUS_RUNNING;
Michal Vasko9e036d52016-01-08 10:49:26 +01002623
Michal Vasko71090fc2016-05-24 16:37:28 +02002624 return msgtype;
Michal Vasko9e036d52016-01-08 10:49:26 +01002625
Michal Vasko71090fc2016-05-24 16:37:28 +02002626cleanup:
Michal Vaskoade892d2017-02-22 13:40:35 +01002627 /* ENDPT UNLOCK */
Michal Vasko2e6defd2016-10-07 15:48:15 +02002628 pthread_rwlock_unlock(&server_opts.endpt_lock);
Michal Vasko3031aae2016-01-27 16:07:18 +01002629
Michal Vaskoe1a64ec2016-03-01 12:21:58 +01002630 nc_session_free(*session, NULL);
Michal Vasko1a38c862016-01-15 15:50:07 +01002631 *session = NULL;
Michal Vasko71090fc2016-05-24 16:37:28 +02002632 return msgtype;
Michal Vasko9e036d52016-01-08 10:49:26 +01002633}
2634
Michal Vaskob83a3fa2021-05-26 09:53:42 +02002635#if defined (NC_ENABLED_SSH) || defined (NC_ENABLED_TLS)
Michal Vasko946cacb2020-08-12 11:18:08 +02002636
Michal Vaskoadf30f02019-06-24 09:34:47 +02002637/* client is expected to be locked */
2638static int
2639_nc_server_ch_client_del_endpt(struct nc_ch_client *client, const char *endpt_name, NC_TRANSPORT_IMPL ti)
Michal Vasko2e6defd2016-10-07 15:48:15 +02002640{
2641 uint16_t i;
Michal Vaskoadf30f02019-06-24 09:34:47 +02002642 int ret = -1;
2643
2644 if (!endpt_name) {
2645 /* remove all endpoints */
2646 for (i = 0; i < client->ch_endpt_count; ++i) {
Michal Vasko93224072021-11-09 12:14:28 +01002647 free(client->ch_endpts[i].name);
2648 free(client->ch_endpts[i].address);
Michal Vaskoadf30f02019-06-24 09:34:47 +02002649 if (client->ch_endpts[i].sock_pending != -1) {
2650 close(client->ch_endpts[i].sock_pending);
2651 }
2652 switch (client->ch_endpts[i].ti) {
2653#ifdef NC_ENABLED_SSH
2654 case NC_TI_LIBSSH:
2655 nc_server_ssh_clear_opts(client->ch_endpts[i].opts.ssh);
2656 free(client->ch_endpts[i].opts.ssh);
2657 break;
2658#endif
2659#ifdef NC_ENABLED_TLS
2660 case NC_TI_OPENSSL:
2661 nc_server_tls_clear_opts(client->ch_endpts[i].opts.tls);
2662 free(client->ch_endpts[i].opts.tls);
2663 break;
2664#endif
2665 default:
2666 ERRINT;
2667 /* won't get here ...*/
2668 break;
2669 }
2670 }
2671 free(client->ch_endpts);
2672 client->ch_endpts = NULL;
2673 client->ch_endpt_count = 0;
2674
2675 ret = 0;
2676 } else {
2677 for (i = 0; i < client->ch_endpt_count; ++i) {
2678 if (!strcmp(client->ch_endpts[i].name, endpt_name) && (!ti || (ti == client->ch_endpts[i].ti))) {
Michal Vasko93224072021-11-09 12:14:28 +01002679 free(client->ch_endpts[i].name);
2680 free(client->ch_endpts[i].address);
Michal Vaskoadf30f02019-06-24 09:34:47 +02002681 if (client->ch_endpts[i].sock_pending != -1) {
2682 close(client->ch_endpts[i].sock_pending);
2683 }
2684 switch (client->ch_endpts[i].ti) {
2685#ifdef NC_ENABLED_SSH
2686 case NC_TI_LIBSSH:
2687 nc_server_ssh_clear_opts(client->ch_endpts[i].opts.ssh);
2688 free(client->ch_endpts[i].opts.ssh);
2689 break;
2690#endif
2691#ifdef NC_ENABLED_TLS
2692 case NC_TI_OPENSSL:
2693 nc_server_tls_clear_opts(client->ch_endpts[i].opts.tls);
2694 free(client->ch_endpts[i].opts.tls);
2695 break;
2696#endif
2697 default:
2698 ERRINT;
2699 /* won't get here ...*/
2700 break;
2701 }
2702
2703 /* move last endpoint to the empty space */
2704 --client->ch_endpt_count;
2705 if (i < client->ch_endpt_count) {
2706 memcpy(&client->ch_endpts[i], &client->ch_endpts[client->ch_endpt_count], sizeof *client->ch_endpts);
2707 } else if (!server_opts.ch_client_count) {
2708 free(server_opts.ch_clients);
2709 server_opts.ch_clients = NULL;
2710 }
2711
2712 ret = 0;
2713 break;
2714 }
2715 }
2716 }
2717
2718 return ret;
2719}
2720
2721API int
2722nc_server_ch_add_client(const char *name)
2723{
2724 uint16_t i;
2725 struct nc_ch_client *client;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002726
2727 if (!name) {
2728 ERRARG("name");
2729 return -1;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002730 }
2731
2732 /* WRITE LOCK */
2733 pthread_rwlock_wrlock(&server_opts.ch_client_lock);
2734
2735 /* check name uniqueness */
2736 for (i = 0; i < server_opts.ch_client_count; ++i) {
2737 if (!strcmp(server_opts.ch_clients[i].name, name)) {
Michal Vasko05532772021-06-03 12:12:38 +02002738 ERR(NULL, "Call Home client \"%s\" already exists.", name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002739 /* WRITE UNLOCK */
2740 pthread_rwlock_unlock(&server_opts.ch_client_lock);
2741 return -1;
2742 }
2743 }
2744
2745 ++server_opts.ch_client_count;
2746 server_opts.ch_clients = nc_realloc(server_opts.ch_clients, server_opts.ch_client_count * sizeof *server_opts.ch_clients);
2747 if (!server_opts.ch_clients) {
2748 ERRMEM;
2749 /* WRITE UNLOCK */
2750 pthread_rwlock_unlock(&server_opts.ch_client_lock);
2751 return -1;
2752 }
Michal Vaskoadf30f02019-06-24 09:34:47 +02002753 client = &server_opts.ch_clients[server_opts.ch_client_count - 1];
Michal Vasko2e6defd2016-10-07 15:48:15 +02002754
Michal Vasko93224072021-11-09 12:14:28 +01002755 client->name = strdup(name);
Michal Vasko5bd4a3f2021-06-17 16:40:10 +02002756 client->id = ATOMIC_INC_RELAXED(server_opts.new_client_id);
Michal Vaskoadf30f02019-06-24 09:34:47 +02002757 client->ch_endpts = NULL;
2758 client->ch_endpt_count = 0;
2759 client->conn_type = 0;
Michal Vaskoc4bc5812016-10-13 10:59:36 +02002760
Michal Vasko2e6defd2016-10-07 15:48:15 +02002761 /* set CH default options */
Michal Vaskoadf30f02019-06-24 09:34:47 +02002762 client->start_with = NC_CH_FIRST_LISTED;
2763 client->max_attempts = 3;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002764
Michal Vaskoadf30f02019-06-24 09:34:47 +02002765 pthread_mutex_init(&client->lock, NULL);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002766
2767 /* WRITE UNLOCK */
2768 pthread_rwlock_unlock(&server_opts.ch_client_lock);
2769
2770 return 0;
2771}
2772
2773API int
Michal Vaskoadf30f02019-06-24 09:34:47 +02002774nc_server_ch_del_client(const char *name)
Michal Vasko2e6defd2016-10-07 15:48:15 +02002775{
Michal Vaskoadf30f02019-06-24 09:34:47 +02002776 uint16_t i;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002777 int ret = -1;
2778
2779 /* WRITE LOCK */
2780 pthread_rwlock_wrlock(&server_opts.ch_client_lock);
2781
Michal Vaskoadf30f02019-06-24 09:34:47 +02002782 if (!name) {
2783 /* remove all CH clients with endpoints */
Michal Vasko2e6defd2016-10-07 15:48:15 +02002784 for (i = 0; i < server_opts.ch_client_count; ++i) {
Michal Vasko93224072021-11-09 12:14:28 +01002785 free(server_opts.ch_clients[i].name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002786
2787 /* remove all endpoints */
Michal Vaskoadf30f02019-06-24 09:34:47 +02002788 _nc_server_ch_client_del_endpt(&server_opts.ch_clients[i], NULL, 0);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002789
2790 pthread_mutex_destroy(&server_opts.ch_clients[i].lock);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002791 ret = 0;
2792 }
2793 free(server_opts.ch_clients);
2794 server_opts.ch_clients = NULL;
2795
2796 server_opts.ch_client_count = 0;
2797
2798 } else {
Michal Vaskoadf30f02019-06-24 09:34:47 +02002799 /* remove one client with endpoints */
Michal Vasko2e6defd2016-10-07 15:48:15 +02002800 for (i = 0; i < server_opts.ch_client_count; ++i) {
Michal Vaskoadf30f02019-06-24 09:34:47 +02002801 if (!strcmp(server_opts.ch_clients[i].name, name)) {
Michal Vasko93224072021-11-09 12:14:28 +01002802 free(server_opts.ch_clients[i].name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002803
Michal Vasko2e6defd2016-10-07 15:48:15 +02002804 /* remove all endpoints */
Michal Vaskoadf30f02019-06-24 09:34:47 +02002805 _nc_server_ch_client_del_endpt(&server_opts.ch_clients[i], NULL, 0);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002806
2807 pthread_mutex_destroy(&server_opts.ch_clients[i].lock);
2808
2809 /* move last client and endpoint(s) to the empty space */
2810 --server_opts.ch_client_count;
2811 if (i < server_opts.ch_client_count) {
2812 memcpy(&server_opts.ch_clients[i], &server_opts.ch_clients[server_opts.ch_client_count],
Michal Vaskoadf30f02019-06-24 09:34:47 +02002813 sizeof *server_opts.ch_clients);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002814 } else if (!server_opts.ch_client_count) {
2815 free(server_opts.ch_clients);
2816 server_opts.ch_clients = NULL;
2817 }
2818
2819 ret = 0;
Michal Vaskoadf30f02019-06-24 09:34:47 +02002820 break;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002821 }
2822 }
2823 }
2824
2825 /* WRITE UNLOCK */
2826 pthread_rwlock_unlock(&server_opts.ch_client_lock);
2827
2828 return ret;
2829}
2830
2831API int
Michal Vaskofb1724b2020-01-31 11:02:00 +01002832nc_server_ch_is_client(const char *name)
2833{
2834 uint16_t i;
2835 int found = 0;
2836
2837 if (!name) {
2838 return found;
2839 }
2840
2841 /* READ LOCK */
2842 pthread_rwlock_rdlock(&server_opts.ch_client_lock);
2843
2844 /* check name uniqueness */
2845 for (i = 0; i < server_opts.ch_client_count; ++i) {
2846 if (!strcmp(server_opts.ch_clients[i].name, name)) {
2847 found = 1;
2848 break;
2849 }
2850 }
2851
2852 /* UNLOCK */
2853 pthread_rwlock_unlock(&server_opts.ch_client_lock);
2854
2855 return found;
2856}
2857
2858API int
Michal Vaskoadf30f02019-06-24 09:34:47 +02002859nc_server_ch_client_add_endpt(const char *client_name, const char *endpt_name, NC_TRANSPORT_IMPL ti)
Michal Vasko2e6defd2016-10-07 15:48:15 +02002860{
2861 uint16_t i;
2862 struct nc_ch_client *client;
Michal Vaskoadf30f02019-06-24 09:34:47 +02002863 struct nc_ch_endpt *endpt;
2864 int ret = -1;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002865
2866 if (!client_name) {
2867 ERRARG("client_name");
2868 return -1;
2869 } else if (!endpt_name) {
2870 ERRARG("endpt_name");
2871 return -1;
Michal Vaskoadf30f02019-06-24 09:34:47 +02002872 } else if (!ti) {
2873 ERRARG("ti");
2874 return -1;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002875 }
2876
2877 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02002878 nc_server_ch_client_lock(client_name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002879 if (!client) {
2880 return -1;
2881 }
2882
2883 for (i = 0; i < client->ch_endpt_count; ++i) {
2884 if (!strcmp(client->ch_endpts[i].name, endpt_name)) {
Michal Vasko05532772021-06-03 12:12:38 +02002885 ERR(NULL, "Call Home client \"%s\" endpoint \"%s\" already exists.", client_name, endpt_name);
Michal Vaskoadf30f02019-06-24 09:34:47 +02002886 goto cleanup;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002887 }
2888 }
2889
2890 ++client->ch_endpt_count;
2891 client->ch_endpts = realloc(client->ch_endpts, client->ch_endpt_count * sizeof *client->ch_endpts);
2892 if (!client->ch_endpts) {
2893 ERRMEM;
Michal Vaskoadf30f02019-06-24 09:34:47 +02002894 goto cleanup;
2895 }
2896 endpt = &client->ch_endpts[client->ch_endpt_count - 1];
2897
2898 memset(endpt, 0, sizeof *client->ch_endpts);
Michal Vasko93224072021-11-09 12:14:28 +01002899 endpt->name = strdup(endpt_name);
Michal Vaskoadf30f02019-06-24 09:34:47 +02002900 endpt->ti = ti;
2901 endpt->sock_pending = -1;
2902 endpt->ka.idle_time = 1;
2903 endpt->ka.max_probes = 10;
2904 endpt->ka.probe_interval = 5;
2905
2906 switch (ti) {
2907#ifdef NC_ENABLED_SSH
2908 case NC_TI_LIBSSH:
2909 endpt->opts.ssh = calloc(1, sizeof(struct nc_server_ssh_opts));
2910 if (!endpt->opts.ssh) {
2911 ERRMEM;
2912 goto cleanup;
2913 }
roman41a11e42022-06-22 09:27:08 +02002914 endpt->opts.ssh->auth_methods = NC_SSH_AUTH_PUBLICKEY | NC_SSH_AUTH_PASSWORD;
Michal Vaskoadf30f02019-06-24 09:34:47 +02002915 endpt->opts.ssh->auth_attempts = 3;
Michal Vaskocbad4c52019-06-27 16:30:35 +02002916 endpt->opts.ssh->auth_timeout = 30;
Michal Vaskoadf30f02019-06-24 09:34:47 +02002917 break;
2918#endif
2919#ifdef NC_ENABLED_TLS
2920 case NC_TI_OPENSSL:
2921 endpt->opts.tls = calloc(1, sizeof(struct nc_server_tls_opts));
2922 if (!endpt->opts.tls) {
2923 ERRMEM;
2924 goto cleanup;
2925 }
2926 break;
2927#endif
2928 default:
2929 ERRINT;
2930 goto cleanup;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002931 }
2932
Michal Vaskoadf30f02019-06-24 09:34:47 +02002933 /* success */
2934 ret = 0;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002935
Michal Vaskoadf30f02019-06-24 09:34:47 +02002936cleanup:
Michal Vasko2e6defd2016-10-07 15:48:15 +02002937 /* UNLOCK */
2938 nc_server_ch_client_unlock(client);
2939
Michal Vaskoadf30f02019-06-24 09:34:47 +02002940 return ret;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002941}
2942
2943API int
Michal Vaskoadf30f02019-06-24 09:34:47 +02002944nc_server_ch_client_del_endpt(const char *client_name, const char *endpt_name, NC_TRANSPORT_IMPL ti)
Michal Vasko2e6defd2016-10-07 15:48:15 +02002945{
Michal Vaskoadf30f02019-06-24 09:34:47 +02002946 int ret;
Michal Vasko2e6defd2016-10-07 15:48:15 +02002947 struct nc_ch_client *client;
2948
2949 if (!client_name) {
2950 ERRARG("client_name");
2951 return -1;
2952 }
2953
2954 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02002955 nc_server_ch_client_lock(client_name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002956 if (!client) {
2957 return -1;
2958 }
2959
Michal Vaskoadf30f02019-06-24 09:34:47 +02002960 ret = _nc_server_ch_client_del_endpt(client, endpt_name, ti);
Michal Vasko2e6defd2016-10-07 15:48:15 +02002961
2962 /* UNLOCK */
2963 nc_server_ch_client_unlock(client);
2964
2965 return ret;
2966}
2967
2968API int
Michal Vaskofb1724b2020-01-31 11:02:00 +01002969nc_server_ch_client_is_endpt(const char *client_name, const char *endpt_name)
2970{
2971 uint16_t i;
2972 struct nc_ch_client *client = NULL;
2973 int found = 0;
2974
2975 if (!client_name || !endpt_name) {
2976 return found;
2977 }
2978
2979 /* READ LOCK */
2980 pthread_rwlock_rdlock(&server_opts.ch_client_lock);
2981
2982 for (i = 0; i < server_opts.ch_client_count; ++i) {
2983 if (!strcmp(server_opts.ch_clients[i].name, client_name)) {
2984 client = &server_opts.ch_clients[i];
2985 break;
2986 }
2987 }
2988
2989 if (!client) {
2990 goto cleanup;
2991 }
2992
2993 for (i = 0; i < client->ch_endpt_count; ++i) {
2994 if (!strcmp(client->ch_endpts[i].name, endpt_name)) {
2995 found = 1;
2996 break;
2997 }
2998 }
2999
3000cleanup:
3001 /* UNLOCK */
3002 pthread_rwlock_unlock(&server_opts.ch_client_lock);
3003 return found;
3004}
3005
3006API int
Michal Vasko2e6defd2016-10-07 15:48:15 +02003007nc_server_ch_client_endpt_set_address(const char *client_name, const char *endpt_name, const char *address)
3008{
Michal Vasko2e6defd2016-10-07 15:48:15 +02003009 struct nc_ch_client *client;
Michal Vaskoadf30f02019-06-24 09:34:47 +02003010 struct nc_ch_endpt *endpt;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003011
3012 if (!client_name) {
3013 ERRARG("client_name");
3014 return -1;
3015 } else if (!endpt_name) {
3016 ERRARG("endpt_name");
3017 return -1;
3018 } else if (!address) {
3019 ERRARG("address");
3020 return -1;
3021 }
3022
3023 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003024 endpt = nc_server_ch_client_lock(client_name, endpt_name, 0, &client);
3025 if (!endpt) {
Michal Vasko2e6defd2016-10-07 15:48:15 +02003026 return -1;
3027 }
3028
Michal Vasko93224072021-11-09 12:14:28 +01003029 free(endpt->address);
3030 endpt->address = strdup(address);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003031
3032 /* UNLOCK */
3033 nc_server_ch_client_unlock(client);
3034
Michal Vaskoadf30f02019-06-24 09:34:47 +02003035 return 0;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003036}
3037
3038API int
3039nc_server_ch_client_endpt_set_port(const char *client_name, const char *endpt_name, uint16_t port)
3040{
Michal Vasko2e6defd2016-10-07 15:48:15 +02003041 struct nc_ch_client *client;
Michal Vaskoadf30f02019-06-24 09:34:47 +02003042 struct nc_ch_endpt *endpt;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003043
3044 if (!client_name) {
3045 ERRARG("client_name");
3046 return -1;
3047 } else if (!endpt_name) {
3048 ERRARG("endpt_name");
3049 return -1;
3050 } else if (!port) {
3051 ERRARG("port");
3052 return -1;
3053 }
3054
3055 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003056 endpt = nc_server_ch_client_lock(client_name, endpt_name, 0, &client);
3057 if (!endpt) {
Michal Vasko2e6defd2016-10-07 15:48:15 +02003058 return -1;
3059 }
3060
Michal Vaskoadf30f02019-06-24 09:34:47 +02003061 endpt->port = port;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003062
3063 /* UNLOCK */
3064 nc_server_ch_client_unlock(client);
3065
ravsz5c5a4422020-03-31 15:53:21 +02003066 return 0;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003067}
3068
3069API int
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003070nc_server_ch_client_endpt_enable_keepalives(const char *client_name, const char *endpt_name, int enable)
3071{
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003072 struct nc_ch_client *client;
Michal Vaskoadf30f02019-06-24 09:34:47 +02003073 struct nc_ch_endpt *endpt;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003074
3075 if (!client_name) {
3076 ERRARG("client_name");
3077 return -1;
3078 } else if (!endpt_name) {
3079 ERRARG("endpt_name");
3080 return -1;
3081 }
3082
3083 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003084 endpt = nc_server_ch_client_lock(client_name, endpt_name, 0, &client);
3085 if (!endpt) {
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003086 return -1;
3087 }
3088
Michal Vaskoadf30f02019-06-24 09:34:47 +02003089 endpt->ka.enabled = (enable ? 1 : 0);
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003090
3091 /* UNLOCK */
3092 nc_server_ch_client_unlock(client);
3093
Michal Vasko9af829a2019-09-12 13:50:00 +02003094 return 0;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003095}
3096
3097API int
3098nc_server_ch_client_endpt_set_keepalives(const char *client_name, const char *endpt_name, int idle_time, int max_probes,
3099 int probe_interval)
3100{
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003101 struct nc_ch_client *client;
Michal Vaskoadf30f02019-06-24 09:34:47 +02003102 struct nc_ch_endpt *endpt;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003103
3104 if (!client_name) {
3105 ERRARG("client_name");
3106 return -1;
3107 } else if (!endpt_name) {
3108 ERRARG("endpt_name");
3109 return -1;
3110 }
3111
3112 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003113 endpt = nc_server_ch_client_lock(client_name, endpt_name, 0, &client);
3114 if (!endpt) {
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003115 return -1;
3116 }
3117
Michal Vaskoadf30f02019-06-24 09:34:47 +02003118 if (idle_time > -1) {
3119 endpt->ka.idle_time = idle_time;
3120 }
3121 if (max_probes > -1) {
3122 endpt->ka.max_probes = max_probes;
3123 }
3124 if (probe_interval > -1) {
3125 endpt->ka.probe_interval = probe_interval;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003126 }
3127
3128 /* UNLOCK */
3129 nc_server_ch_client_unlock(client);
3130
Michal Vasko9af829a2019-09-12 13:50:00 +02003131 return 0;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003132}
3133
3134API int
Michal Vasko2e6defd2016-10-07 15:48:15 +02003135nc_server_ch_client_set_conn_type(const char *client_name, NC_CH_CONN_TYPE conn_type)
3136{
3137 struct nc_ch_client *client;
3138
3139 if (!client_name) {
3140 ERRARG("client_name");
3141 return -1;
3142 } else if (!conn_type) {
3143 ERRARG("conn_type");
3144 return -1;
3145 }
3146
3147 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003148 nc_server_ch_client_lock(client_name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003149 if (!client) {
3150 return -1;
3151 }
3152
3153 if (client->conn_type != conn_type) {
3154 client->conn_type = conn_type;
3155
3156 /* set default options */
3157 switch (conn_type) {
3158 case NC_CH_PERSIST:
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003159 /* no options */
Michal Vasko2e6defd2016-10-07 15:48:15 +02003160 break;
3161 case NC_CH_PERIOD:
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003162 client->conn.period.period = 60;
3163 client->conn.period.anchor_time = 0;
3164 client->conn.period.idle_timeout = 120;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003165 break;
3166 default:
3167 ERRINT;
3168 break;
3169 }
3170 }
3171
3172 /* UNLOCK */
3173 nc_server_ch_client_unlock(client);
3174
3175 return 0;
3176}
3177
3178API int
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003179nc_server_ch_client_periodic_set_period(const char *client_name, uint16_t period)
3180{
3181 struct nc_ch_client *client;
3182
3183 if (!client_name) {
3184 ERRARG("client_name");
3185 return -1;
3186 } else if (!period) {
3187 ERRARG("period");
3188 return -1;
3189 }
3190
3191 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003192 nc_server_ch_client_lock(client_name, NULL, 0, &client);
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003193 if (!client) {
3194 return -1;
3195 }
3196
3197 if (client->conn_type != NC_CH_PERIOD) {
Michal Vasko05532772021-06-03 12:12:38 +02003198 ERR(NULL, "Call Home client \"%s\" is not of periodic connection type.", client_name);
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003199 /* UNLOCK */
3200 nc_server_ch_client_unlock(client);
3201 return -1;
3202 }
3203
3204 client->conn.period.period = period;
3205
3206 /* UNLOCK */
3207 nc_server_ch_client_unlock(client);
3208
3209 return 0;
3210}
3211
3212API int
3213nc_server_ch_client_periodic_set_anchor_time(const char *client_name, time_t anchor_time)
Michal Vasko2e6defd2016-10-07 15:48:15 +02003214{
3215 struct nc_ch_client *client;
3216
3217 if (!client_name) {
3218 ERRARG("client_name");
3219 return -1;
3220 }
3221
3222 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003223 nc_server_ch_client_lock(client_name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003224 if (!client) {
3225 return -1;
3226 }
3227
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003228 if (client->conn_type != NC_CH_PERIOD) {
Michal Vasko05532772021-06-03 12:12:38 +02003229 ERR(NULL, "Call Home client \"%s\" is not of periodic connection type.", client_name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003230 /* UNLOCK */
3231 nc_server_ch_client_unlock(client);
3232 return -1;
3233 }
3234
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003235 client->conn.period.anchor_time = anchor_time;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003236
3237 /* UNLOCK */
3238 nc_server_ch_client_unlock(client);
3239
3240 return 0;
3241}
3242
3243API int
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003244nc_server_ch_client_periodic_set_idle_timeout(const char *client_name, uint16_t idle_timeout)
Michal Vasko2e6defd2016-10-07 15:48:15 +02003245{
3246 struct nc_ch_client *client;
3247
3248 if (!client_name) {
3249 ERRARG("client_name");
3250 return -1;
3251 }
3252
3253 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003254 nc_server_ch_client_lock(client_name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003255 if (!client) {
3256 return -1;
3257 }
3258
3259 if (client->conn_type != NC_CH_PERIOD) {
Michal Vasko05532772021-06-03 12:12:38 +02003260 ERR(NULL, "Call Home client \"%s\" is not of periodic connection type.", client_name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003261 /* UNLOCK */
3262 nc_server_ch_client_unlock(client);
3263 return -1;
3264 }
3265
3266 client->conn.period.idle_timeout = idle_timeout;
3267
3268 /* UNLOCK */
3269 nc_server_ch_client_unlock(client);
3270
3271 return 0;
3272}
3273
3274API int
Michal Vasko2e6defd2016-10-07 15:48:15 +02003275nc_server_ch_client_set_start_with(const char *client_name, NC_CH_START_WITH start_with)
3276{
3277 struct nc_ch_client *client;
3278
3279 if (!client_name) {
3280 ERRARG("client_name");
3281 return -1;
3282 }
3283
3284 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003285 nc_server_ch_client_lock(client_name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003286 if (!client) {
3287 return -1;
3288 }
3289
3290 client->start_with = start_with;
3291
3292 /* UNLOCK */
3293 nc_server_ch_client_unlock(client);
3294
3295 return 0;
3296}
3297
3298API int
3299nc_server_ch_client_set_max_attempts(const char *client_name, uint8_t max_attempts)
3300{
3301 struct nc_ch_client *client;
3302
3303 if (!client_name) {
3304 ERRARG("client_name");
3305 return -1;
3306 } else if (!max_attempts) {
3307 ERRARG("max_attempts");
3308 return -1;
3309 }
3310
3311 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003312 nc_server_ch_client_lock(client_name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003313 if (!client) {
3314 return -1;
3315 }
3316
3317 client->max_attempts = max_attempts;
3318
3319 /* UNLOCK */
3320 nc_server_ch_client_unlock(client);
3321
3322 return 0;
3323}
3324
Michal Vasko056f53c2022-10-21 13:38:15 +02003325/**
3326 * @brief Create a connection for an endpoint.
3327 *
3328 * Client lock is expected to be held.
3329 *
3330 * @param[in] endpt Endpoint to use.
3331 * @param[in] acquire_ctx_cb Callback for acquiring the libyang context.
3332 * @param[in] release_ctx_cb Callback for releasing the libyang context.
3333 * @param[in] ctx_cb_data Context callbacks data.
3334 * @param[out] session Created NC session.
3335 * @return NC_MSG values.
3336 */
Michal Vasko2e6defd2016-10-07 15:48:15 +02003337static NC_MSG_TYPE
Michal Vasko58bac1c2022-03-24 15:25:26 +01003338nc_connect_ch_endpt(struct nc_ch_endpt *endpt, nc_server_ch_session_acquire_ctx_cb acquire_ctx_cb,
3339 nc_server_ch_session_release_ctx_cb release_ctx_cb, void *ctx_cb_data, struct nc_session **session)
Michal Vaskob05053d2016-01-22 16:12:06 +01003340{
Michal Vasko71090fc2016-05-24 16:37:28 +02003341 NC_MSG_TYPE msgtype;
Michal Vasko58bac1c2022-03-24 15:25:26 +01003342 const struct ly_ctx *ctx = NULL;
Michal Vaskob05053d2016-01-22 16:12:06 +01003343 int sock, ret;
Michal Vasko9fb42272017-10-05 13:50:05 +02003344 struct timespec ts_cur;
Michal Vasko66032bc2019-01-22 15:03:12 +01003345 char *ip_host;
Michal Vaskob05053d2016-01-22 16:12:06 +01003346
Michal Vasko056f53c2022-10-21 13:38:15 +02003347 sock = nc_sock_connect(endpt->address, endpt->port, NC_CH_CONNECT_TIMEOUT, &endpt->ka, &endpt->sock_pending, &ip_host);
Michal Vaskoc61c4492016-01-25 11:13:34 +01003348 if (sock < 0) {
Michal Vasko71090fc2016-05-24 16:37:28 +02003349 return NC_MSG_ERROR;
Michal Vaskob05053d2016-01-22 16:12:06 +01003350 }
3351
Michal Vasko93224072021-11-09 12:14:28 +01003352 /* acquire context */
3353 ctx = acquire_ctx_cb(ctx_cb_data);
3354 if (!ctx) {
3355 ERR(NULL, "Failed to acquire context for a new Call Home session.");
3356 close(sock);
3357 free(ip_host);
3358 return NC_MSG_ERROR;
3359 }
3360
3361 /* create session */
Michal Vasko131120a2018-05-29 15:44:02 +02003362 *session = nc_new_session(NC_SERVER, 0);
Michal Vaskob05053d2016-01-22 16:12:06 +01003363 if (!(*session)) {
3364 ERRMEM;
3365 close(sock);
Michal Vasko66032bc2019-01-22 15:03:12 +01003366 free(ip_host);
Michal Vaskod530d362022-09-07 10:07:57 +02003367 msgtype = NC_MSG_ERROR;
3368 goto fail;
Michal Vaskob05053d2016-01-22 16:12:06 +01003369 }
3370 (*session)->status = NC_STATUS_STARTING;
Michal Vasko93224072021-11-09 12:14:28 +01003371 (*session)->ctx = (struct ly_ctx *)ctx;
Michal Vaskodc96bb92023-03-28 08:52:48 +02003372 (*session)->flags = NC_SESSION_SHAREDCTX | NC_SESSION_CALLHOME;
Michal Vasko93224072021-11-09 12:14:28 +01003373 (*session)->host = ip_host;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003374 (*session)->port = endpt->port;
Michal Vaskob05053d2016-01-22 16:12:06 +01003375
Michal Vaskob05053d2016-01-22 16:12:06 +01003376 /* sock gets assigned to session or closed */
Radek Krejci53691be2016-02-22 13:58:37 +01003377#ifdef NC_ENABLED_SSH
Michal Vaskoadf30f02019-06-24 09:34:47 +02003378 if (endpt->ti == NC_TI_LIBSSH) {
3379 (*session)->data = endpt->opts.ssh;
Michal Vasko0190bc32016-03-02 15:47:49 +01003380 ret = nc_accept_ssh_session(*session, sock, NC_TRANSPORT_TIMEOUT);
Michal Vasko2cc4c682016-03-01 09:16:48 +01003381 (*session)->data = NULL;
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +01003382
Michal Vasko71090fc2016-05-24 16:37:28 +02003383 if (ret < 0) {
3384 msgtype = NC_MSG_ERROR;
3385 goto fail;
3386 } else if (!ret) {
3387 msgtype = NC_MSG_WOULDBLOCK;
Michal Vaskob05053d2016-01-22 16:12:06 +01003388 goto fail;
3389 }
Michal Vasko3d865d22016-01-28 16:00:53 +01003390 } else
3391#endif
Radek Krejci53691be2016-02-22 13:58:37 +01003392#ifdef NC_ENABLED_TLS
Michal Vaskoadf30f02019-06-24 09:34:47 +02003393 if (endpt->ti == NC_TI_OPENSSL) {
3394 (*session)->data = endpt->opts.tls;
Michal Vasko0190bc32016-03-02 15:47:49 +01003395 ret = nc_accept_tls_session(*session, sock, NC_TRANSPORT_TIMEOUT);
Michal Vasko2cc4c682016-03-01 09:16:48 +01003396 (*session)->data = NULL;
Michal Vaskoc6b9c7b2016-01-28 11:10:08 +01003397
Michal Vasko71090fc2016-05-24 16:37:28 +02003398 if (ret < 0) {
3399 msgtype = NC_MSG_ERROR;
3400 goto fail;
3401 } else if (!ret) {
3402 msgtype = NC_MSG_WOULDBLOCK;
Michal Vaskob05053d2016-01-22 16:12:06 +01003403 goto fail;
3404 }
Michal Vasko3d865d22016-01-28 16:00:53 +01003405 } else
3406#endif
3407 {
Michal Vaskob05053d2016-01-22 16:12:06 +01003408 ERRINT;
3409 close(sock);
Michal Vasko71090fc2016-05-24 16:37:28 +02003410 msgtype = NC_MSG_ERROR;
Michal Vaskob05053d2016-01-22 16:12:06 +01003411 goto fail;
3412 }
3413
3414 /* assign new SID atomically */
Michal Vasko5bd4a3f2021-06-17 16:40:10 +02003415 (*session)->id = ATOMIC_INC_RELAXED(server_opts.new_session_id);
Michal Vaskob05053d2016-01-22 16:12:06 +01003416
3417 /* NETCONF handshake */
Michal Vasko131120a2018-05-29 15:44:02 +02003418 msgtype = nc_handshake_io(*session);
Michal Vasko71090fc2016-05-24 16:37:28 +02003419 if (msgtype != NC_MSG_HELLO) {
Michal Vaskob05053d2016-01-22 16:12:06 +01003420 goto fail;
3421 }
Michal Vasko9fb42272017-10-05 13:50:05 +02003422
Michal Vaskod8a74192023-02-06 15:51:50 +01003423 nc_timeouttime_get(&ts_cur, 0);
Michal Vasko9fb42272017-10-05 13:50:05 +02003424 (*session)->opts.server.last_rpc = ts_cur.tv_sec;
Michal Vaskod8a74192023-02-06 15:51:50 +01003425 nc_realtime_get(&ts_cur);
Michal Vasko9fb42272017-10-05 13:50:05 +02003426 (*session)->opts.server.session_start = ts_cur.tv_sec;
Michal Vaskob05053d2016-01-22 16:12:06 +01003427 (*session)->status = NC_STATUS_RUNNING;
3428
Michal Vasko71090fc2016-05-24 16:37:28 +02003429 return msgtype;
Michal Vaskob05053d2016-01-22 16:12:06 +01003430
3431fail:
Michal Vaskoe1a64ec2016-03-01 12:21:58 +01003432 nc_session_free(*session, NULL);
Michal Vaskob05053d2016-01-22 16:12:06 +01003433 *session = NULL;
Michal Vasko58bac1c2022-03-24 15:25:26 +01003434 if (ctx) {
3435 release_ctx_cb(ctx_cb_data);
3436 }
Michal Vasko71090fc2016-05-24 16:37:28 +02003437 return msgtype;
Michal Vaskob05053d2016-01-22 16:12:06 +01003438}
3439
Michal Vasko2e6defd2016-10-07 15:48:15 +02003440struct nc_ch_client_thread_arg {
3441 char *client_name;
Michal Vasko93224072021-11-09 12:14:28 +01003442 nc_server_ch_session_acquire_ctx_cb acquire_ctx_cb;
3443 nc_server_ch_session_release_ctx_cb release_ctx_cb;
3444 void *ctx_cb_data;
3445 nc_server_ch_new_session_cb new_session_cb;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003446};
3447
3448static struct nc_ch_client *
3449nc_server_ch_client_with_endpt_lock(const char *name)
3450{
3451 struct nc_ch_client *client;
3452
3453 while (1) {
3454 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003455 nc_server_ch_client_lock(name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003456 if (!client) {
3457 return NULL;
3458 }
3459 if (client->ch_endpt_count) {
3460 return client;
3461 }
3462 /* no endpoints defined yet */
3463
3464 /* UNLOCK */
3465 nc_server_ch_client_unlock(client);
3466
3467 usleep(NC_CH_NO_ENDPT_WAIT * 1000);
3468 }
3469
3470 return NULL;
3471}
3472
3473static int
3474nc_server_ch_client_thread_session_cond_wait(struct nc_session *session, struct nc_ch_client_thread_arg *data)
3475{
Michal Vasko3f05a092018-03-13 10:39:49 +01003476 int ret = 0, r;
Michal Vaskoc4bc5812016-10-13 10:59:36 +02003477 uint32_t idle_timeout;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003478 struct timespec ts;
3479 struct nc_ch_client *client;
3480
Michal Vasko2e6defd2016-10-07 15:48:15 +02003481 /* CH LOCK */
Michal Vaskoacf98472021-02-04 15:33:57 +01003482 pthread_mutex_lock(&session->opts.server.ch_lock);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003483
Michal Vaskofeccb312022-03-24 15:24:59 +01003484 session->flags |= NC_SESSION_CH_THREAD;
Michal Vasko0db3db52021-03-03 10:45:42 +01003485
Michal Vasko2e6defd2016-10-07 15:48:15 +02003486 /* give the session to the user */
Michal Vasko93224072021-11-09 12:14:28 +01003487 if (data->new_session_cb(data->client_name, session)) {
Michal Vaskof1c26c22021-04-12 16:34:33 +02003488 /* something is wrong, free the session */
Michal Vaskofeccb312022-03-24 15:24:59 +01003489 session->flags &= ~NC_SESSION_CH_THREAD;
Michal Vaskof1c26c22021-04-12 16:34:33 +02003490
3491 /* CH UNLOCK */
3492 pthread_mutex_unlock(&session->opts.server.ch_lock);
3493
Michal Vasko77d56d72022-09-07 10:30:48 +02003494 /* session terminated, free it and release its context */
Michal Vaskof1c26c22021-04-12 16:34:33 +02003495 nc_session_free(session, NULL);
Michal Vasko58bac1c2022-03-24 15:25:26 +01003496 data->release_ctx_cb(data->ctx_cb_data);
3497 return ret;
Michal Vaskof1c26c22021-04-12 16:34:33 +02003498 }
Michal Vasko2e6defd2016-10-07 15:48:15 +02003499
3500 do {
Michal Vaskod8a74192023-02-06 15:51:50 +01003501 nc_timeouttime_get(&ts, NC_CH_NO_ENDPT_WAIT);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003502
Michal Vasko0db3db52021-03-03 10:45:42 +01003503 /* CH COND WAIT */
Michal Vaskod8a74192023-02-06 15:51:50 +01003504 r = pthread_cond_clockwait(&session->opts.server.ch_cond, &session->opts.server.ch_lock, COMPAT_CLOCK_ID, &ts);
Michal Vasko3f05a092018-03-13 10:39:49 +01003505 if (!r) {
3506 /* we were woken up, something probably happened */
3507 if (session->status != NC_STATUS_RUNNING) {
3508 break;
3509 }
3510 } else if (r != ETIMEDOUT) {
Michal Vasko05532772021-06-03 12:12:38 +02003511 ERR(session, "Pthread condition timedwait failed (%s).", strerror(r));
Michal Vasko3f05a092018-03-13 10:39:49 +01003512 ret = -1;
3513 break;
Michal Vasko2e39ed92018-03-12 13:51:44 +01003514 }
3515
Michal Vasko2e6defd2016-10-07 15:48:15 +02003516 /* check whether the client was not removed */
3517 /* LOCK */
Michal Vaskoadf30f02019-06-24 09:34:47 +02003518 nc_server_ch_client_lock(data->client_name, NULL, 0, &client);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003519 if (!client) {
3520 /* client was removed, finish thread */
Michal Vasko05532772021-06-03 12:12:38 +02003521 VRB(session, "Call Home client \"%s\" removed, but an established session will not be terminated.",
Michal Vaskoadf30f02019-06-24 09:34:47 +02003522 data->client_name);
Michal Vasko3f05a092018-03-13 10:39:49 +01003523 ret = 1;
3524 break;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003525 }
3526
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003527 if (client->conn_type == NC_CH_PERIOD) {
Michal Vaskoc4bc5812016-10-13 10:59:36 +02003528 idle_timeout = client->conn.period.idle_timeout;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003529 } else {
3530 idle_timeout = 0;
Michal Vaskoc4bc5812016-10-13 10:59:36 +02003531 }
3532
Michal Vaskod8a74192023-02-06 15:51:50 +01003533 nc_timeouttime_get(&ts, 0);
Michal Vaskodf68e7e2022-04-21 11:04:00 +02003534 if (!nc_session_get_notif_status(session) && idle_timeout && (ts.tv_sec >= session->opts.server.last_rpc + idle_timeout)) {
Michal Vasko05532772021-06-03 12:12:38 +02003535 VRB(session, "Call Home client \"%s\": session idle timeout elapsed.", client->name);
Michal Vaskoc4bc5812016-10-13 10:59:36 +02003536 session->status = NC_STATUS_INVALID;
3537 session->term_reason = NC_SESSION_TERM_TIMEOUT;
3538 }
Michal Vasko2e6defd2016-10-07 15:48:15 +02003539
3540 /* UNLOCK */
3541 nc_server_ch_client_unlock(client);
3542
3543 } while (session->status == NC_STATUS_RUNNING);
3544
Michal Vaskofeccb312022-03-24 15:24:59 +01003545 /* signal to nc_session_free() that CH thread is terminating */
3546 session->flags &= ~NC_SESSION_CH_THREAD;
3547 pthread_cond_signal(&session->opts.server.ch_cond);
Michal Vasko0db3db52021-03-03 10:45:42 +01003548
Michal Vasko27377422018-03-15 08:59:35 +01003549 /* CH UNLOCK */
Michal Vaskoacf98472021-02-04 15:33:57 +01003550 pthread_mutex_unlock(&session->opts.server.ch_lock);
Michal Vasko27377422018-03-15 08:59:35 +01003551
Michal Vasko3f05a092018-03-13 10:39:49 +01003552 return ret;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003553}
3554
3555static void *
3556nc_ch_client_thread(void *arg)
3557{
3558 struct nc_ch_client_thread_arg *data = (struct nc_ch_client_thread_arg *)arg;
3559 NC_MSG_TYPE msgtype;
3560 uint8_t cur_attempts = 0;
Peter Feiged05f2252018-09-03 08:09:47 +00003561 uint16_t next_endpt_index;
Michal Vasko9550cf12017-03-21 15:33:58 +01003562 char *cur_endpt_name = NULL;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003563 struct nc_ch_endpt *cur_endpt;
3564 struct nc_session *session;
3565 struct nc_ch_client *client;
Michal Vasko5d0a1e32022-09-07 07:46:36 +02003566 uint32_t client_id, reconnect_in;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003567
3568 /* LOCK */
3569 client = nc_server_ch_client_with_endpt_lock(data->client_name);
3570 if (!client) {
3571 goto cleanup;
3572 }
Andrew Langefeld6ed922d2018-09-12 14:08:32 -05003573 client_id = client->id;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003574
3575 cur_endpt = &client->ch_endpts[0];
3576 cur_endpt_name = strdup(cur_endpt->name);
3577
3578 while (1) {
Michal Vasko056f53c2022-10-21 13:38:15 +02003579 if (!cur_attempts) {
3580 VRB(NULL, "Call Home client \"%s\" endpoint \"%s\" connecting...", data->client_name, cur_endpt_name);
3581 }
Michal Vasko58bac1c2022-03-24 15:25:26 +01003582 msgtype = nc_connect_ch_endpt(cur_endpt, data->acquire_ctx_cb, data->release_ctx_cb, data->ctx_cb_data, &session);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003583
3584 if (msgtype == NC_MSG_HELLO) {
3585 /* UNLOCK */
3586 nc_server_ch_client_unlock(client);
3587
Michal Vasko05532772021-06-03 12:12:38 +02003588 VRB(NULL, "Call Home client \"%s\" session %u established.", data->client_name, session->id);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003589 if (nc_server_ch_client_thread_session_cond_wait(session, data)) {
3590 goto cleanup;
3591 }
Michal Vasko05532772021-06-03 12:12:38 +02003592 VRB(NULL, "Call Home client \"%s\" session terminated.", data->client_name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003593
3594 /* LOCK */
3595 client = nc_server_ch_client_with_endpt_lock(data->client_name);
3596 if (!client) {
3597 goto cleanup;
3598 }
Andrew Langefeld6ed922d2018-09-12 14:08:32 -05003599 if (client->id != client_id) {
3600 nc_server_ch_client_unlock(client);
3601 goto cleanup;
3602 }
Michal Vasko2e6defd2016-10-07 15:48:15 +02003603
3604 /* session changed status -> it was disconnected for whatever reason,
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003605 * persistent connection immediately tries to reconnect, periodic connects at specific times */
Michal Vasko2e6defd2016-10-07 15:48:15 +02003606 if (client->conn_type == NC_CH_PERIOD) {
Michal Vasko18e1fa02021-11-29 09:02:05 +01003607 if (client->conn.period.anchor_time) {
3608 /* anchored */
3609 reconnect_in = (time(NULL) - client->conn.period.anchor_time) % (client->conn.period.period * 60);
3610 } else {
3611 /* fixed timeout */
3612 reconnect_in = client->conn.period.period * 60;
3613 }
3614
Michal Vasko2e6defd2016-10-07 15:48:15 +02003615 /* UNLOCK */
3616 nc_server_ch_client_unlock(client);
3617
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003618 /* sleep until we should reconnect TODO wake up sometimes to check for new notifications */
Michal Vasko75b836e2022-09-07 07:51:45 +02003619 VRB(NULL, "Call Home client \"%s\" reconnecting in %" PRIu32 " seconds.", data->client_name, reconnect_in);
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003620 sleep(reconnect_in);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003621
3622 /* LOCK */
3623 client = nc_server_ch_client_with_endpt_lock(data->client_name);
3624 if (!client) {
3625 goto cleanup;
3626 }
Andrew Langefeld6ed922d2018-09-12 14:08:32 -05003627 if (client->id != client_id) {
3628 nc_server_ch_client_unlock(client);
3629 goto cleanup;
3630 }
Michal Vasko2e6defd2016-10-07 15:48:15 +02003631 }
3632
3633 /* set next endpoint to try */
3634 if (client->start_with == NC_CH_FIRST_LISTED) {
Peter Feiged05f2252018-09-03 08:09:47 +00003635 next_endpt_index = 0;
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003636 } else if (client->start_with == NC_CH_LAST_CONNECTED) {
Peter Feiged05f2252018-09-03 08:09:47 +00003637 /* we keep the current one but due to unlock/lock we have to find it again */
3638 for (next_endpt_index = 0; next_endpt_index < client->ch_endpt_count; ++next_endpt_index) {
3639 if (!strcmp(client->ch_endpts[next_endpt_index].name, cur_endpt_name)) {
3640 break;
3641 }
3642 }
3643 if (next_endpt_index >= client->ch_endpt_count) {
3644 /* endpoint was removed, start with the first one */
3645 next_endpt_index = 0;
3646 }
Michal Vaskoe49a15f2019-05-27 14:18:36 +02003647 } else {
3648 /* just get a random index */
3649 next_endpt_index = rand() % client->ch_endpt_count;
Peter Feiged05f2252018-09-03 08:09:47 +00003650 }
3651
Michal Vasko2e6defd2016-10-07 15:48:15 +02003652 } else {
Michal Vasko6bb116b2016-10-26 13:53:46 +02003653 /* UNLOCK */
3654 nc_server_ch_client_unlock(client);
3655
Michal Vasko2e6defd2016-10-07 15:48:15 +02003656 /* session was not created */
Michal Vaskod8951c72022-09-26 09:39:29 +02003657 sleep(NC_CH_ENDPT_BACKOFF_WAIT);
Michal Vaskoc4bc5812016-10-13 10:59:36 +02003658
Michal Vasko6bb116b2016-10-26 13:53:46 +02003659 /* LOCK */
3660 client = nc_server_ch_client_with_endpt_lock(data->client_name);
3661 if (!client) {
3662 goto cleanup;
3663 }
Andrew Langefeld6ed922d2018-09-12 14:08:32 -05003664 if (client->id != client_id) {
3665 nc_server_ch_client_unlock(client);
3666 goto cleanup;
3667 }
Michal Vasko6bb116b2016-10-26 13:53:46 +02003668
Michal Vasko2e6defd2016-10-07 15:48:15 +02003669 ++cur_attempts;
Michal Vasko4cb8de52018-04-23 14:38:07 +02003670
3671 /* try to find our endpoint again */
Peter Feiged05f2252018-09-03 08:09:47 +00003672 for (next_endpt_index = 0; next_endpt_index < client->ch_endpt_count; ++next_endpt_index) {
3673 if (!strcmp(client->ch_endpts[next_endpt_index].name, cur_endpt_name)) {
Michal Vasko4cb8de52018-04-23 14:38:07 +02003674 break;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003675 }
Michal Vasko4cb8de52018-04-23 14:38:07 +02003676 }
3677
Peter Feiged05f2252018-09-03 08:09:47 +00003678 if (next_endpt_index >= client->ch_endpt_count) {
Michal Vasko4cb8de52018-04-23 14:38:07 +02003679 /* endpoint was removed, start with the first one */
Michal Vasko056f53c2022-10-21 13:38:15 +02003680 VRB(NULL, "Call Home client \"%s\" endpoint \"%s\" removed.", data->client_name, cur_endpt_name);
Peter Feiged05f2252018-09-03 08:09:47 +00003681 next_endpt_index = 0;
Michal Vasko4cb8de52018-04-23 14:38:07 +02003682 cur_attempts = 0;
3683 } else if (cur_attempts == client->max_attempts) {
3684 /* we have tried to connect to this endpoint enough times */
Michal Vasko056f53c2022-10-21 13:38:15 +02003685 VRB(NULL, "Call Home client \"%s\" endpoint \"%s\" failed connection attempt limit %" PRIu8 " reached.",
3686 data->client_name, cur_endpt_name, client->max_attempts);
3687
3688 /* clear a pending socket, if any */
3689 cur_endpt = &client->ch_endpts[next_endpt_index];
3690 if (cur_endpt->sock_pending > -1) {
3691 close(cur_endpt->sock_pending);
3692 cur_endpt->sock_pending = -1;
3693 }
3694
Peter Feiged05f2252018-09-03 08:09:47 +00003695 if (next_endpt_index < client->ch_endpt_count - 1) {
Michal Vasko2e6defd2016-10-07 15:48:15 +02003696 /* just go to the next endpoint */
Peter Feiged05f2252018-09-03 08:09:47 +00003697 ++next_endpt_index;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003698 } else {
Michal Vasko4cb8de52018-04-23 14:38:07 +02003699 /* cur_endpoint is the last, start with the first one */
Michal Vasko2a225342018-09-05 08:38:34 +02003700 next_endpt_index = 0;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003701 }
Michal Vasko2e6defd2016-10-07 15:48:15 +02003702 cur_attempts = 0;
3703 } /* else we keep the current one */
3704 }
Peter Feiged05f2252018-09-03 08:09:47 +00003705
3706 cur_endpt = &client->ch_endpts[next_endpt_index];
3707 free(cur_endpt_name);
3708 cur_endpt_name = strdup(cur_endpt->name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003709 }
3710
3711cleanup:
Michal Vasko05532772021-06-03 12:12:38 +02003712 VRB(NULL, "Call Home client \"%s\" thread exit.", data->client_name);
Michal Vasko9550cf12017-03-21 15:33:58 +01003713 free(cur_endpt_name);
Michal Vasko2e6defd2016-10-07 15:48:15 +02003714 free(data->client_name);
3715 free(data);
3716 return NULL;
3717}
3718
3719API int
Michal Vasko93224072021-11-09 12:14:28 +01003720nc_connect_ch_client_dispatch(const char *client_name, nc_server_ch_session_acquire_ctx_cb acquire_ctx_cb,
3721 nc_server_ch_session_release_ctx_cb release_ctx_cb, void *ctx_cb_data, nc_server_ch_new_session_cb new_session_cb)
Michal Vasko3f05a092018-03-13 10:39:49 +01003722{
Michal Vasko2e6defd2016-10-07 15:48:15 +02003723 int ret;
3724 pthread_t tid;
3725 struct nc_ch_client_thread_arg *arg;
3726
3727 if (!client_name) {
3728 ERRARG("client_name");
3729 return -1;
Michal Vasko93224072021-11-09 12:14:28 +01003730 } else if (!acquire_ctx_cb) {
3731 ERRARG("acquire_ctx_cb");
3732 return -1;
3733 } else if (!release_ctx_cb) {
3734 ERRARG("release_ctx_cb");
3735 return -1;
3736 } else if (!new_session_cb) {
3737 ERRARG("new_session_cb");
Michal Vasko2e6defd2016-10-07 15:48:15 +02003738 return -1;
3739 }
3740
3741 arg = malloc(sizeof *arg);
3742 if (!arg) {
3743 ERRMEM;
3744 return -1;
3745 }
3746 arg->client_name = strdup(client_name);
3747 if (!arg->client_name) {
3748 ERRMEM;
3749 free(arg);
3750 return -1;
3751 }
Michal Vasko93224072021-11-09 12:14:28 +01003752 arg->acquire_ctx_cb = acquire_ctx_cb;
3753 arg->release_ctx_cb = release_ctx_cb;
3754 arg->ctx_cb_data = ctx_cb_data;
3755 arg->new_session_cb = new_session_cb;
Michal Vasko2e6defd2016-10-07 15:48:15 +02003756
3757 ret = pthread_create(&tid, NULL, nc_ch_client_thread, arg);
3758 if (ret) {
Michal Vasko05532772021-06-03 12:12:38 +02003759 ERR(NULL, "Creating a new thread failed (%s).", strerror(ret));
Michal Vasko2e6defd2016-10-07 15:48:15 +02003760 free(arg->client_name);
3761 free(arg);
3762 return -1;
3763 }
3764 /* the thread now manages arg */
3765
3766 pthread_detach(tid);
3767
3768 return 0;
3769}
3770
Radek Krejci53691be2016-02-22 13:58:37 +01003771#endif /* NC_ENABLED_SSH || NC_ENABLED_TLS */
Michal Vaskof8352352016-05-24 09:11:36 +02003772
Michal Vaskoc45ebd32016-05-25 11:17:36 +02003773API time_t
3774nc_session_get_start_time(const struct nc_session *session)
Michal Vaskof8352352016-05-24 09:11:36 +02003775{
Michal Vasko2e6defd2016-10-07 15:48:15 +02003776 if (!session || (session->side != NC_SERVER)) {
Michal Vaskof8352352016-05-24 09:11:36 +02003777 ERRARG("session");
Michal Vaskoc45ebd32016-05-25 11:17:36 +02003778 return 0;
Michal Vaskof8352352016-05-24 09:11:36 +02003779 }
3780
Michal Vasko2e6defd2016-10-07 15:48:15 +02003781 return session->opts.server.session_start;
Michal Vaskof8352352016-05-24 09:11:36 +02003782}
Michal Vasko3486a7c2017-03-03 13:28:07 +01003783
3784API void
Michal Vasko71dbd772021-03-23 14:08:37 +01003785nc_session_inc_notif_status(struct nc_session *session)
Michal Vasko3486a7c2017-03-03 13:28:07 +01003786{
3787 if (!session || (session->side != NC_SERVER)) {
3788 ERRARG("session");
3789 return;
3790 }
3791
Michal Vaskodf68e7e2022-04-21 11:04:00 +02003792 /* NTF STATUS LOCK */
3793 pthread_mutex_lock(&session->opts.server.ntf_status_lock);
3794
Michal Vasko71dbd772021-03-23 14:08:37 +01003795 ++session->opts.server.ntf_status;
Michal Vaskodf68e7e2022-04-21 11:04:00 +02003796
3797 /* NTF STATUS UNLOCK */
3798 pthread_mutex_unlock(&session->opts.server.ntf_status_lock);
Michal Vasko71dbd772021-03-23 14:08:37 +01003799}
3800
3801API void
3802nc_session_dec_notif_status(struct nc_session *session)
3803{
3804 if (!session || (session->side != NC_SERVER)) {
3805 ERRARG("session");
3806 return;
3807 }
3808
Michal Vaskodf68e7e2022-04-21 11:04:00 +02003809 /* NTF STATUS LOCK */
3810 pthread_mutex_lock(&session->opts.server.ntf_status_lock);
3811
Michal Vasko71dbd772021-03-23 14:08:37 +01003812 if (session->opts.server.ntf_status) {
3813 --session->opts.server.ntf_status;
3814 }
Michal Vaskodf68e7e2022-04-21 11:04:00 +02003815
3816 /* NTF STATUS UNLOCK */
3817 pthread_mutex_unlock(&session->opts.server.ntf_status_lock);
Michal Vasko3486a7c2017-03-03 13:28:07 +01003818}
3819
3820API int
3821nc_session_get_notif_status(const struct nc_session *session)
3822{
Michal Vaskodf68e7e2022-04-21 11:04:00 +02003823 uint32_t ntf_status;
3824
Michal Vasko3486a7c2017-03-03 13:28:07 +01003825 if (!session || (session->side != NC_SERVER)) {
3826 ERRARG("session");
3827 return 0;
3828 }
3829
Michal Vaskodf68e7e2022-04-21 11:04:00 +02003830 /* NTF STATUS LOCK */
3831 pthread_mutex_lock(&((struct nc_session *)session)->opts.server.ntf_status_lock);
3832
3833 ntf_status = session->opts.server.ntf_status;
3834
3835 /* NTF STATUS UNLOCK */
3836 pthread_mutex_unlock(&((struct nc_session *)session)->opts.server.ntf_status_lock);
3837
3838 return ntf_status;
Michal Vasko086311b2016-01-08 09:53:11 +01003839}