diff options
Diffstat (limited to 'tests/bandwidth-server-many-up.c')
| -rw-r--r-- | tests/bandwidth-server-many-up.c | 143 |
1 files changed, 143 insertions, 0 deletions
diff --git a/tests/bandwidth-server-many-up.c b/tests/bandwidth-server-many-up.c new file mode 100644 index 0000000..02968c1 --- /dev/null +++ b/tests/bandwidth-server-many-up.c @@ -0,0 +1,143 @@ +/* + * Copyright © 2009-2010 Stéphane Raimbault <stephane.raimbault@gmail.com> + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +#include <stdio.h> +#include <unistd.h> +#include <string.h> +#include <stdlib.h> +#include <errno.h> +#include <signal.h> + +#include <modbus.h> + +#if defined(_WIN32) +#include <ws2tcpip.h> +#else +#include <sys/socket.h> +#include <netinet/in.h> +#include <arpa/inet.h> +#endif + +#define NB_CONNECTION 5 + +modbus_t *ctx = NULL; +int server_socket; +modbus_mapping_t *mb_mapping; + +static void close_sigint(int dummy) +{ + close(server_socket); + modbus_free(ctx); + modbus_mapping_free(mb_mapping); + + exit(dummy); +} + +int main(void) +{ + int master_socket; + int rc; + fd_set refset; + fd_set rdset; + + /* Maximum file descriptor number */ + int fdmax; + + ctx = modbus_new_tcp("127.0.0.1", 1502); + + mb_mapping = modbus_mapping_new(MODBUS_MAX_READ_BITS, 0, + MODBUS_MAX_READ_REGISTERS, 0); + if (mb_mapping == NULL) { + fprintf(stderr, "Failed to allocate the mapping: %s\n", + modbus_strerror(errno)); + modbus_free(ctx); + return -1; + } + + server_socket = modbus_tcp_listen(ctx, NB_CONNECTION); + + signal(SIGINT, close_sigint); + + /* Clear the reference set of socket */ + FD_ZERO(&refset); + /* Add the server socket */ + FD_SET(server_socket, &refset); + + /* Keep track of the max file descriptor */ + fdmax = server_socket; + + for (;;) { + rdset = refset; + if (select(fdmax+1, &rdset, NULL, NULL, NULL) == -1) { + perror("Server select() failure."); + close_sigint(1); + } + + /* Run through the existing connections looking for data to be + * read */ + for (master_socket = 0; master_socket <= fdmax; master_socket++) { + + if (FD_ISSET(master_socket, &rdset)) { + if (master_socket == server_socket) { + /* A client is asking a new connection */ + socklen_t addrlen; + struct sockaddr_in clientaddr; + int newfd; + + /* Handle new connections */ + addrlen = sizeof(clientaddr); + memset(&clientaddr, 0, sizeof(clientaddr)); + newfd = accept(server_socket, (struct sockaddr *)&clientaddr, &addrlen); + if (newfd == -1) { + perror("Server accept() error"); + } else { + FD_SET(newfd, &refset); + + if (newfd > fdmax) { + /* Keep track of the maximum */ + fdmax = newfd; + } + printf("New connection from %s:%d on socket %d\n", + inet_ntoa(clientaddr.sin_addr), clientaddr.sin_port, newfd); + } + } else { + /* An already connected master has sent a new query */ + uint8_t query[MODBUS_TCP_MAX_ADU_LENGTH]; + + modbus_set_socket(ctx, master_socket); + rc = modbus_receive(ctx, query); + if (rc != -1) { + modbus_reply(ctx, query, rc, mb_mapping); + } else { + /* Connection closed by the client, end of server */ + printf("Connection closed on socket %d\n", master_socket); + close(master_socket); + + /* Remove from reference set */ + FD_CLR(master_socket, &refset); + + if (master_socket == fdmax) { + fdmax--; + } + } + } + } + } + } + + return 0; +} |