%% options copyright owner = Dirk Krause copyright year = 2015-xxxx SPDX-License-Identifier: BSD-3-Clause %% module #include "dk4conf.h" #if DK4_HAVE_MATH_H #ifndef MATH_H_INCLUDED #if DK4_ON_WINDOWS #ifndef _USE_MATH_DEFINES #define _USE_MATH_DEFINES 1 #endif #endif #include #define MATH_H_INCLUDED 1 #endif #endif #if DK4_HAVE_IO_H #ifndef IO_H_INCLUDED #include #define IO_H_INCLUDED 1 #endif #endif #if DK4_HAVE_FCNTL_H #ifndef FCNTL_H_INCLUDED #include #define FCNTL_H_INCLUDED 1 #endif #endif #if DK4_HAVE_STRING_H #ifndef STRING_H_INCLUDED #include #define STRING_H_INCLUDED 1 #endif #endif #include "dk4app.h" #include "dk4aopt.h" #include "dk4socka.h" #include "dk4maadu.h" #include "dk4maidddu.h" #include "dk4maodd.h" #include "dk4mpl.h" #include "dk4vers.h" #include "dk4isadm.h" #include "dk4strd.h" #include "dk4strda.h" #include "dk4mem.h" #include "dk4mema.h" #include "dk4fopda.h" #include "dk4pathd.h" #include "dk4unused.h" #include "dk4time.h" #ifndef DK4WMAIN_H_INCLUDED #include "dk4wmain.h" #endif $!trace-include /** Connected peer. */ typedef struct { dk4_sockaddr_storage_t peeraddr; /**< Peer address. */ dk4_um_t num; /**< Number of client. */ size_t peerlen; /**< Length of peer address. */ dk4_socket_t sock; /**< Socket for transfer. */ } dk_send_client_t; /** Constant texts used by the program, not localized. */ static const dkChar * const dk_send_kwnl[] = { $!string-table macro=dkT # # 0 Program group name # dktools # # 1 Help text file name # dk-send.txt # # 2 String table file name # dk-send.str # # 3 Keyword unlimited for number of clients # unlimited # # 4 Path separator for Windows # \\ # # 5 File open mode: Read binary. # rb $!end }; /** Text used by the program, replaced by localized texts if found. */ static const dkChar * const dk_send_kw_def[] = { $!string-table macro=dkT # # 0 Error: Failed to set up signal handlers! # Failed to set up signal handlers! # # 1 Error: Failed to restore signal handlers! # Failed to restore signal handlers! # # 2 Name of input file just processed (default). # stdin # # 3 Progress: Accepting incoming connection requests. # Accepting incoming connection requests. # # 4 Error: Problem with peer address length! # Problem with peer address length! # # 5 Progress: Connected ... # Connected: # # 6 Debug: Rejected ... # Rejected: # # 7 Progress: Starting data transfer. # Starting data transfer. # # 8 Progress: Finished data transfer. # Finished data transfer. # # 9 ERROR: Disconnected ... # Disconnected: # # 10 11 ERROR: Invalid number of clients ...! # Invalid number of clients " "! # # 12 ERROR: Wrong number of command line arguments! # Wrong number of command line arguments! # # 13 ERROR: Aborted by INT signal! # Aborted by INT signal! # # 14 ERROR: Aborted by TERM signal! # Aborted by TERM signal! # # 15 Debug: Not a dk-send client ... # Not a dk-send client: # # 16 17 18 # Timeout too large! Timeout must not be negative! Timeout value not found! # # 19 # Stop processing as all clients are disconnected! # # 20 21 22 # Connect time: Transfer time: Transfer size: # # 23 24 # seconds. bytes. # # # $!end }; /** Short help text. */ static const dkChar * const dk_send_help_text[] = { $!text macro=dkT Send file to one or multiple recipients (dk-recv processes). dk-send [] [] Options: -n Number of clients, ``unlimited'' or 0 for no limit. -a
[/][,...] Allowed clients. -l Listen on local host ports only. -b Listen backlog (number of connection attempts to queue). -c Timeout for read attempt after establishing a new connection. -p Running as sink in a pipe (do not abort reading input if all clients disconnected unexpectedly). --help Show this short help text. --manual *** SHOW FULL MANUAL. *** --version Show version information. --license Show license information. $!end }; /** License text. */ static const dkChar * const dk_send_license_text[] = { $!text macro=dkT,preprocessor This software uses code from the following projects, either directly or as a library: dktools Dirk Krause's tools and libraries. See http://sourceforge.net/p/dktools/wiki/Home/ for more information. #if DK4_HAVE_ZLIB_H zlib Data compression library. See http://www.zlib.net/ for more information. #endif #if DK4_HAVE_BZLIB_H bzip2 Data compression program and library. See http://www.bzip.org/ for more information. #endif #if DK4_HAVE_OPENSSL_MD5_H && DK4_HAVE_OPENSSL_SHA_H && DK4_HAVE_OPENSSL_RIPEMD_H OpenSSL Cryptographic toolkit, used to build file checksums here. See http://www.openssl.org/ for more information. This product includes software developed by the OpenSSL Project for use in the OpenSSL Toolkit (http://www.openssl.org/). This product includes cryptographic software written by Eric Young (eay@cryptsoft.com). This product includes software written by Tim Hudson (tjh@cryptsoft.com). #endif All the licenses below apply to the program. Licenses for used libraries are shown as found on my Scientific Linux 6.x computer in the /usr/share/doc directory on 2015-04-01. Check the project homepages of the used libraries for additional information and/or updated license terms. DK tools and libraries license ============================== Copyright (c) 2015-2016, Dirk Krause All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. #if DK4_HAVE_ZLIB_H Zlib license ============ (C) 1995-2004 Jean-loup Gailly and Mark Adler This software is provided 'as-is', without any express or implied warranty. In no event will the authors be held liable for any damages arising from the use of this software. Permission is granted to anyone to use this software for any purpose, including commercial applications, and to alter it and redistribute it freely, subject to the following restrictions: 1. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required. 2. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software. 3. This notice may not be removed or altered from any source distribution. Jean-loup Gailly Mark Adler jloup@gzip.org madler@alumni.caltech.edu #endif #if DK4_HAVE_BZLIB_H Bzip2 and libbzip2 library license ================================== This program, "bzip2", the associated library "libbzip2", and all documentation, are copyright (C) 1996-2007 Julian R Seward. All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. 2. The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation would be appreciated but is not required. 3. Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software. 4. The name of the author may not be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. Julian Seward, jseward@bzip.org bzip2/libbzip2 version 1.0.5 of 10 December 2007 #endif #if DK4_HAVE_OPENSSL_MD5_H && DK4_HAVE_OPENSSL_SHA_H && DK4_HAVE_OPENSSL_RIPEMD_H OpenSSL license =============== LICENSE ISSUES -------------- The OpenSSL toolkit stays under a dual license, i.e. both the conditions of the OpenSSL License and the original SSLeay license apply to the toolkit. See below for the actual license texts. Actually both licenses are BSD-style Open Source licenses. In case of any license issues related to OpenSSL please contact openssl-core@openssl.org. OpenSSL License --------------- Copyright (c) 1998-2011 The OpenSSL Project. All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. 3. All advertising materials mentioning features or use of this software must display the following acknowledgment: "This product includes software developed by the OpenSSL Project for use in the OpenSSL Toolkit. (http://www.openssl.org/)" 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to endorse or promote products derived from this software without prior written permission. For written permission, please contact openssl-core@openssl.org. 5. Products derived from this software may not be called "OpenSSL" nor may "OpenSSL" appear in their names without prior written permission of the OpenSSL Project. 6. Redistributions of any form whatsoever must retain the following acknowledgment: "This product includes software developed by the OpenSSL Project for use in the OpenSSL Toolkit (http://www.openssl.org/)" THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE OpenSSL PROJECT OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. This product includes cryptographic software written by Eric Young (eay@cryptsoft.com). This product includes software written by Tim Hudson (tjh@cryptsoft.com). Original SSLeay License ----------------------- Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com) All rights reserved. This package is an SSL implementation written by Eric Young (eay@cryptsoft.com). The implementation was written so as to conform with Netscapes SSL. This library is free for commercial and non-commercial use as long as the following conditions are aheared to. The following conditions apply to all code found in this distribution, be it the RC4, RSA, lhash, DES, etc., code; not just the SSL code. The SSL documentation included with this distribution is covered by the same copyright terms except that the holder is Tim Hudson (tjh@cryptsoft.com). Copyright remains Eric Young's, and as such any Copyright notices in the code are not to be removed. If this package is used in a product, Eric Young should be given attribution as the author of the parts of the library used. This can be in the form of a textual message at program startup or in documentation (online or textual) provided with the package. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: 1. Redistributions of source code must retain the copyright notice, this list of conditions and the following disclaimer. 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. 3. All advertising materials mentioning features or use of this software must display the following acknowledgement: "This product includes cryptographic software written by Eric Young (eay@cryptsoft.com)" The word 'cryptographic' can be left out if the rouines from the library being used are not cryptographic related :-). 4. If you include any Windows specific code (or a derivative thereof) from the apps directory (application code) you must include an acknowledgement: "This product includes software written by Tim Hudson (tjh@cryptsoft.com)" THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. The licence and distribution terms for any publically available version or derivative of this code cannot be changed. i.e. this code cannot simply be copied and put under another distribution licence [including the GNU Public Licence.] #endif $!end }; /** Options for this program. */ static const dk4_option_specification_t dk_send_options[] = { { dkT('n'), dkT("number-of-clients"), DK4_OPT_ARG_STRING }, { dkT('a'), dkT("allow"), DK4_OPT_ARG_STRING }, { dkT('l'), dkT("local-only"), DK4_OPT_ARG_NONE }, { dkT('p'), dkT("pipe"), DK4_OPT_ARG_NONE }, { dkT('b'), dkT("backlog"), DK4_OPT_ARG_UNSIGNED }, { dkT('c'), dkT("connect-timeout"), DK4_OPT_ARG_DOUBLE } }; /** Buffer for sending. Initial value: 1460 */ #if DK4_SIZEOF_SIZE_T > 2 static char dk_send_buf[1048576]; #else static char dk_send_buf[16384]; #endif /** Error report for counting number of bytes. */ static dk4_er_t ernb; /** Application structure for diagnostics. */ static dk4_app_t *app = NULL; /** Localized texts or default texts. */ static const dkChar * const *dk_send_msg = dk_send_kw_def; /** Collection of allowed clients. */ static dk4_sto_t *s_allowed = NULL; /** Iterator through allowed clients collection. */ static dk4_sto_it_t *i_allowed = NULL; /** Number of expected clients. */ static dk4_um_t numclients = (dk4_um_t)1UL; /** Number of bytes transmitted. */ static dk4_um_t numbytes = (dk4_um_t)0UL; /** Number of elements in the dk_send_msg array. */ static size_t dk_send_sz_msg = sizeof(dk_send_kw_def)/sizeof(DK4_PDKCHAR) - 1; /** Number of elements in the dk_send_options array. */ static size_t dk_send_sz_options = sizeof(dk_send_options)/sizeof(dk4_option_specification_t); /** Timestamp of first client connection. */ static dk4_time_t time_stamp_1 = (dk4_time_t)0UL; /** Timestamp start of transmission. */ static dk4_time_t time_stamp_2 = (dk4_time_t)0UL; /** Timestamp end of transmission. */ static dk4_time_t time_stamp_3 = (dk4_time_t)0UL; /** Seconds for read timeout. */ static long dk_send_timeout_secs = 0L; /** Microseconds for read timeout. */ static long dk_send_timeout_usecs = 0L; /** Exit status code returned by the program. */ static int exval = EXIT_FAILURE; /** Flag: Running as pipe, avoid SIGPIPE. */ static int is_pipe = 0; /** Flag: Allow only connections from localhost. */ static int local_only = 0; /** Number of outstanding connection requests to queue. */ static int backlog = 10; #ifdef SIGPIPE /** Indicator: SIGPIPE signal received. */ static DK4_VOLATILE dk4_sig_atomic_t sig_had_pipe = 0; #endif /** Indicator: SIGINT signal received. */ static DK4_VOLATILE dk4_sig_atomic_t sig_had_int = 0; /** Indicator: SIGTERM signal received. */ static DK4_VOLATILE dk4_sig_atomic_t sig_had_term = 0; /** Pass a volatile pointer to an atomic integer. This function is necessary as some compilers mis-optimize direct access to volatile variables (at least if you believe one of the coding standards). @param ptr Address of atomic integer variable. @return The unmodified pointer. */ static DK4_VOLATILE dk4_sig_atomic_t * sig_pass_pointer(DK4_VOLATILE dk4_sig_atomic_t *ptr) { return ptr; } #ifdef SIGPIPE /** Handler for SIGPIPE signal. @param signo Signal number (always SIGPIPE, ignored). */ static void sig_handler_pipe(int DK4_ARG_UNUSED(signo) ) { DK4_UNUSED_ARG(signo) *sig_pass_pointer(&sig_had_pipe) = 1; } #endif /** Handler for SIGINT signal. @param signo Signal number (always SIGINT, ignored). */ static void sig_handler_int(int DK4_ARG_UNUSED(signo) ) { DK4_UNUSED_ARG(signo) *sig_pass_pointer(&sig_had_int) = 1; } /** Handler for SIGTERM signal. @param signo Signal number (always SIGTERM, ignored). */ static void sig_handler_term(int DK4_ARG_UNUSED(signo) ) { DK4_UNUSED_ARG(signo) *sig_pass_pointer(&sig_had_term) = 1; } /** Read value from volatile atomic type. This function is necessary as some compilers mis-optimize direct access to volatile variables (at least if you believe one of the coding standards). @param ap Pointer to volatile atomic variable. @return Contents of the variable. */ static dk4_sig_atomic_t sig_read_atomic(DK4_VOLATILE dk4_sig_atomic_t *ap) { return (*ap); } /** Check whether we can continue or if a signal was received. @param check_pipe Flag: Check for occured SIGPIPE signal too. @return 1 if the program can continue, 0 if a signal was received. */ static int sig_can_continue( #ifdef SIGPIPE int check_pipe #else int DK4_ARG_UNUSED(check_pipe) #endif ) { int back = 1; #ifndef SIGPIPE DK4_UNUSED_ARG(check_pipe) #else if (0 != check_pipe) { if (0 != sig_read_atomic(&sig_had_pipe)) { back = 0; } } #endif if (0 != sig_read_atomic(&sig_had_int )) { back = 0; } if (0 != sig_read_atomic(&sig_had_term)) { back = 0; } return back; } /** Check whether we can continue. @return 1 for yes, 0 for no. */ static int dk_send_can_continue(void) { int back = 1; $? "+ dk_send_can_continue" if (0 == sig_can_continue(0)) { back = 0; } $? "- dk_send_can_continue %d", back return back; } /** Compare two connected clients. @param l Left client. @param r Right client. @param cr Comparison criteria (ignored). @return 1 if l>r, 0 if l=r, -1 if lnum > pr->num) { back = 1; } else { if (pl->num < pr->num) { back = -1; } else { if (pl->sock > pr->sock) { back = 1; } else { if (pl->sock < pr->sock) { back = -1; } } } } } else { back = 1; } } else { if (NULL != r) { back= -1; } } return back; } /** Obtain information about allowed clients from string. @param argcpy Copy of string, can be modified. @return 1 on success, 0 on error. */ static int dk_send_get_allowed_clients(dkChar *argcpy) { dk4_allowed_peer_t peer; dk4_er_t er; dk4_allowed_peer_t *npeer = NULL; dkChar *pc = NULL; dkChar *pn = NULL; int back = 0; int res = 0; $? "+ dk_send_get_allowed_clients \"%!ds\"", TR_STR(argcpy) s_allowed = dk4sto_open(NULL); if (NULL != s_allowed) { dk4sto_set_comp(s_allowed, dk4socket_allowed_peer_compare, 0); i_allowed = dk4sto_it_open(s_allowed, NULL); if (NULL != i_allowed) { back = 1; pc = argcpy; do { pn = dk4str_chr(pc, dkT(',')); if (NULL != pn) { *(pn++) = dkT('\0'); } DK4_MEMRES(&peer, sizeof(peer)); dk4error_init(&er); res = dk4socket_get_allowed_peer(&peer, pc, &er); if (DK4_SOCKET_RESULT_SUCCESS == res) { npeer = dk4mem_new_app(dk4_allowed_peer_t,1,app); if (NULL != npeer) { DK4_MEMCPY(npeer, &peer, sizeof(peer)); if (0 == dk4sto_add(s_allowed, npeer, NULL)) { /* ERROR: Memory */ dk4app_log_base1(app, DK4_LL_ERROR, 90); dk4mem_free(npeer); npeer = NULL; back = 0; } } else { /* ERROR: Memory, already reported */ back = 0; } } else { /* ERROR: Failed to obtain peer information from string */ dk4socket_log_simple_3( app, DK4_LL_ERROR, 141, 142, pc, &er ); back = 0; } pc = pn; } while ((NULL != pc) && (1 == back)); } else { /* ERROR: Memory */ dk4app_log_base1(app, DK4_LL_ERROR, 90); } } else { /* ERROR: Memory */ dk4app_log_base1(app, DK4_LL_ERROR, 90); } $? "- dk_send_get_allowed_clients %d", back return back; } static int dk_send_check_peer_address(dk4_sockaddr_storage_t *sasto, size_t soalen) { struct sockaddr *psoa; int back = 0; if (NULL != sasto) { psoa = (struct sockaddr *)sasto; switch (psoa->sa_family) { case AF_INET : { if (sizeof(struct sockaddr_in) == soalen) { back = 1; } } break; #if DK4_HAVE_STRUCT_SOCKADDR_IN6 && defined(AF_INET6) case AF_INET6 : { if (sizeof(struct sockaddr_in6) == soalen) { back = 1; } } break; #endif } } return back; } /** Log statistical data (connect wait time, transfer time, transfer size). */ static void log_statistics(void) { dkChar b1[8*sizeof(dk4_um_t)]; /* Connect wait time */ dkChar b2[8*sizeof(dk4_um_t)]; /* Transfer time */ dkChar b3[8*sizeof(dk4_um_t)]; /* Transfer size */ dkChar b4[8*sizeof(dk4_um_t)]; /* Space before number */ const dkChar *msg[5]; size_t sl1; size_t sl2; size_t sl3; size_t slm; size_t i; if ((time_stamp_2 >= time_stamp_1) && (time_stamp_3 >= time_stamp_2)) { dk4ma_write_decimal_unsigned( b1, DK4_SIZEOF(b1,dkChar), (time_stamp_2 - time_stamp_1), 0, &ernb ); dk4ma_write_decimal_unsigned( b2, DK4_SIZEOF(b2,dkChar), (time_stamp_3 - time_stamp_2), 0, &ernb ); dk4ma_write_decimal_unsigned( b3, DK4_SIZEOF(b3,dkChar), numbytes, 0, &ernb ); if (DK4_E_NONE == ernb.ec) { sl1 = dk4str_len(b1); sl2 = dk4str_len(b2); sl3 = dk4str_len(b3); slm = sl1; if (sl2 > slm) slm = sl2; if (sl3 > slm) slm = sl3; /* Number of connection seconds */ for (i = 0; i < (slm - sl1); i++) { b4[i] = dkT(' '); } b4[(slm - sl1)] = dkT('\0'); msg[0] = dk_send_msg[20]; msg[1] = b4; msg[2] = b1; msg[3] = dk_send_msg[23]; msg[4] = NULL; dk4app_log_msg(app, DK4_LL_INFO, msg, 4); /* Number of transfer seconds */ for (i = 0; i < (slm - sl2); i++) { b4[i] = dkT(' '); } b4[(slm - sl2)] = dkT('\0'); msg[0] = dk_send_msg[21]; msg[1] = b4; msg[2] = b2; msg[3] = dk_send_msg[23]; msg[4] = NULL; dk4app_log_msg(app, DK4_LL_INFO, msg, 4); /* Number of transferred bytes */ for (i = 0; i < (slm - sl3); i++) { b4[i] = dkT(' '); } b4[(slm - sl3)] = dkT('\0'); msg[0] = dk_send_msg[22]; msg[1] = b4; msg[2] = b3; msg[3] = dk_send_msg[24]; msg[4] = NULL; dk4app_log_msg(app, DK4_LL_INFO, msg, 4); } } } /** Accept incoming connections and transfer data. @param portname Port name to listen. @param infile Input file to read data from. */ static void dk_send_do_networking( const dkChar *portname, FILE *infile ) { dkChar peerbuf[1072]; const dkChar *msgparts[4]; fd_set rfds; dk4_sockaddr_storage_t sasto; struct sockaddr *psoa = NULL; dk4_socket_set_t *listeners = NULL; dk4_sto_t *s_clients = NULL; dk4_sto_it_t *i_clients = NULL; dk_send_client_t *pcl = NULL; char *bufstart; /* Start of remaining buf */ dk4_um_t numcl = (dk4_um_t)0UL; dk4_socket_t max = 0; /* Maximum socket number */ dk4_socket_t ncsock; /* New client socket */ size_t i; /* Traverse socket set */ size_t soalen; /* Socket address length */ size_t rdbytes; /* Bytes read from file */ size_t wrbytes; /* Bytes written to socket */ size_t wrpart; /* Bytes written in this part */ int cc; /* Flag: Can continue in loop */ int ccwr; /* Flag: Continue send loop */ int res; /* Operation result */ int allowed; /* Client from allowed peer */ int anya; /* Flag: Recipients alive */ unsigned char tra; $? "+ dk_send_do_networking" /* Create storage and iterator for connected clients. */ s_clients = dk4sto_open(NULL); if (NULL == s_clients) { /* ERROR: Memory */ dk4app_log_base1(app, DK4_LL_ERROR, 90); goto finished; } dk4sto_set_comp(s_clients, dk_send_compare_clients, 0); i_clients = dk4sto_it_open(s_clients, NULL); if (NULL == i_clients) { /* ERROR: Memory */ dk4app_log_base1(app, DK4_LL_ERROR, 90); goto finished; } /* Create listener socket set. */ listeners = dk4socket_set_service_app(portname, backlog, local_only, app); if (NULL == listeners) { goto finished; } /* Accept connection requests. */ /* PROGRESS: Accepting incoming connection requests */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_PROGRESS, 3); cc = 1; for (i = 0; i < listeners->szUsed; i++) { if (0 == i) { max = (listeners->pSockets)[i]; } else { if (max < (listeners->pSockets)[i]) { max = (listeners->pSockets)[i]; } } } dk4time_get(&time_stamp_1); while (1 == cc) { if (0 != dk_send_can_continue()) { /* Prepare read file descriptor set */ FD_ZERO(&rfds); for (i = 0; i < listeners->szUsed; i++) { FD_SET( (listeners->pSockets)[i], &rfds ); } res = select( (int)(max+1), &rfds, NULL, NULL, NULL ); if (0 < res) { for (i = 0; i < listeners->szUsed; i++) { if (FD_ISSET((listeners->pSockets)[i], &rfds)) { /* Accept connection request */ soalen = sizeof(sasto); ncsock = dk4socket_accept_app( (listeners->pSockets)[i],(struct sockaddr *)(&sasto),&soalen,app ); if (INVALID_SOCKET != ncsock) { allowed = 1; if ((NULL != s_allowed) && (NULL != i_allowed)) { $? ". check" /* Check peer address */ if (0 != dk_send_check_peer_address(&sasto, soalen)) { if (NULL == dk4sto_it_find_like(i_allowed, &sasto, 1)) { allowed = 0; $? "! not found" } else { $? ". found" } } else { $? "! address length" /* ERROR: Peer address length problem */ dk4app_log_1(app,dk_send_msg,dk_send_sz_msg,DK4_LL_ERROR,4); allowed = 0; } } else { $? ". no check" } if (1 == allowed) { pcl = dk4mem_new_app(dk_send_client_t,1,app); if (NULL != pcl) { pcl->num = numcl; pcl->sock = ncsock; DK4_MEMCPY( &(pcl->peeraddr), &sasto, sizeof(dk4_sockaddr_storage_t) ); pcl->peerlen = soalen; if (0 != dk4sto_add(s_clients, pcl, NULL)) { tra = 0x00; soalen = sizeof(tra); res = dk4socket_recv_app( ncsock, &tra, &soalen, 0, dk_send_timeout_secs, dk_send_timeout_usecs, app ); if (DK4_SOCKET_RESULT_SUCCESS == res) { if (0 < soalen) { numcl++; if (0x00 != tra) { cc = 0; i = listeners->szUsed; } if ((dk4_um_t)0UL < numclients) { if (numcl >= numclients) { cc = 0; i = listeners->szUsed; } } /* Simple message to stdout - accepted */ if (0 != dk4app_log_do(app, DK4_LL_PROGRESS)) { $? ". p" psoa = (struct sockaddr *)(&sasto); res = dk4socket_addr_port_to_text( peerbuf, DK4_SIZEOF(peerbuf,dkChar), psoa->sa_family, (const struct sockaddr *)(&sasto), pcl->peerlen, NULL ); if (DK4_SOCKET_RESULT_SUCCESS == res) { $? ". r" msgparts[0] = dk_send_msg[5]; msgparts[1] = peerbuf; dk4app_log_msg( app, DK4_LL_PROGRESS, msgparts, 2 ); $? ". dk4socket_addr_port_to_text" } else { $? "! dk4socket_addr_port_to_text" } } else { $? ". no progress" } } else { /* ERROR: Nothing was read from socket */ if (0 != dk4app_log_do(app, DK4_LL_DEBUG)) { psoa = (struct sockaddr *)(&sasto); res = dk4socket_addr_port_to_text( peerbuf, DK4_SIZEOF(peerbuf,dkChar), psoa->sa_family, (const struct sockaddr *)(&sasto), pcl->peerlen, NULL ); if (DK4_SOCKET_RESULT_SUCCESS == res) { msgparts[0] = dk_send_msg[15]; msgparts[1] = peerbuf; dk4app_log_msg( app, DK4_LL_DEBUG, msgparts, 2 ); } } dk4socket_close(pcl->sock, NULL); if (0 != dk4sto_remove(s_clients, pcl, NULL)) { dk4mem_free(pcl); } } } else { /* ERROR: Failed to read from socket */ if (0 != dk4app_log_do(app, DK4_LL_DEBUG)) { psoa = (struct sockaddr *)(&sasto); res = dk4socket_addr_port_to_text( peerbuf, DK4_SIZEOF(peerbuf,dkChar), psoa->sa_family, (const struct sockaddr *)(&sasto), pcl->peerlen, NULL ); if (DK4_SOCKET_RESULT_SUCCESS == res) { msgparts[0] = dk_send_msg[15]; msgparts[1] = peerbuf; dk4app_log_msg( app, DK4_LL_DEBUG, msgparts, 2 ); } } dk4socket_close(pcl->sock, NULL); if (0 != dk4sto_remove(s_clients, pcl, NULL)) { dk4mem_free(pcl); } } } else { /* ERROR: Memory */ dk4app_log_base1(app, DK4_LL_ERROR, 90); cc = -1; i = listeners->szUsed; dk4socket_close(ncsock, NULL); dk4mem_free(pcl); } } else { cc = -1; i = listeners->szUsed; dk4socket_close(ncsock, NULL); } } else { /* ERROR: Denied - Simple message to stdout */ if (0 != dk4app_log_do(app, DK4_LL_DEBUG)) { psoa = (struct sockaddr *)(&sasto); res = dk4socket_addr_port_to_text( peerbuf, DK4_SIZEOF(peerbuf,dkChar), psoa->sa_family, (const struct sockaddr *)(&sasto), soalen, NULL ); if (DK4_SOCKET_RESULT_SUCCESS == res) { msgparts[0] = dk_send_msg[6]; msgparts[1] = peerbuf; dk4app_log_msg( app, DK4_LL_DEBUG, msgparts, 2 ); } } dk4socket_close(ncsock, NULL); } } else { /* Accept failed */ if (0 == dk_send_can_continue()) { cc = -1; } } } } } else { if (0 == dk_send_can_continue()) { cc = -1; } } } else { cc = -1; } } if (0 != cc) { goto finished; } dk4time_get(&time_stamp_2); /* All preparations succeeded. */ exval = EXIT_SUCCESS; /* Transfer data. */ dk4socket_set_close(listeners, NULL); listeners = NULL; /* PROGRESS: Starting data transfer */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_PROGRESS, 7); cc = 1; anya = 1; while ((1 == cc) && ((0 != anya) || (0 != is_pipe))) { if (0 != dk_send_can_continue()) { rdbytes = fread(dk_send_buf, 1, sizeof(dk_send_buf), infile); if (0 < rdbytes) { $? ". fread -> %u bytes", (unsigned)rdbytes anya = 0; numbytes = dk4ma_um_add(numbytes, rdbytes, &ernb); dk4sto_it_reset(i_clients); do { pcl = (dk_send_client_t *)dk4sto_it_next(i_clients); if (NULL != pcl) { if (INVALID_SOCKET != pcl->sock) { $? ". client connection" bufstart = dk_send_buf; wrbytes = 0; ccwr = 1; do { if (0 != dk_send_can_continue()) { wrpart = rdbytes - wrbytes; $? ". wrpart=%u", (unsigned)wrpart res = dk4socket_send_app( pcl->sock, bufstart, &wrpart, 0, 0L, 0L, app ); switch (res) { case DK4_SOCKET_RESULT_SUCCESS : case DK4_SOCKET_RESULT_IN_PROGRESS : { $? ". success or partial success" wrbytes += wrpart; $? ". bytes written %u", (unsigned)wrbytes if (wrbytes >= rdbytes) { ccwr = 0; $? ". finished" } else { $? ". must continue" bufstart = &(bufstart[wrpart]); } } break; default : { $? "! failed to send" /* ERROR: Failure on socket */ ccwr = -1; exval = EXIT_FAILURE; dk4socket_close(pcl->sock, NULL); pcl->sock = INVALID_SOCKET; if (0 != dk4app_log_do(app, DK4_LL_ERROR)) { psoa = (struct sockaddr *)(&(pcl->peeraddr)); res = dk4socket_addr_port_to_text( peerbuf, DK4_SIZEOF(peerbuf,dkChar), psoa->sa_family, (const struct sockaddr *)(&(pcl->peeraddr)), pcl->peerlen, NULL ); if (DK4_SOCKET_RESULT_SUCCESS == res) { msgparts[0] = dk_send_msg[9]; msgparts[1] = peerbuf; dk4app_log_msg( app, DK4_LL_ERROR, msgparts, 2 ); } } } break; } } else { ccwr = -1; } } while (1 == ccwr); if (0 == ccwr) { anya = 1; } $? ". ccwr = %d", ccwr } } if (0 == dk_send_can_continue()) { cc = -1; pcl = NULL; } } while (NULL != pcl); if (0 != feof(infile)) { cc = 0; } } else { cc = 0; } } else { cc = -1; } } dk4time_get(&time_stamp_3); if ((1 == cc) && (0 == anya)) { /* Warning: All clients disconnected */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_WARNING, 19); } /* PROGRESS: Finished data transfer */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_PROGRESS, 8); if (0 != dk4app_log_do(app, DK4_LL_INFO)) { log_statistics(); } if (0 != cc) { exval = EXIT_FAILURE; goto finished; } /* Shut down all sockets. */ cc = 1; dk4sto_it_reset(i_clients); while (1 == cc) { if (0 != dk_send_can_continue()) { pcl = (dk_send_client_t *)dk4sto_it_next(i_clients); if (NULL != pcl) { if (INVALID_SOCKET != pcl->sock) { res = dk4socket_shutdown_app(pcl->sock, DK4_SOCKET_SHUT_WRITE, app); if (DK4_SOCKET_RESULT_SUCCESS != res) { exval = EXIT_FAILURE; dk4socket_close(pcl->sock, NULL); pcl->sock = INVALID_SOCKET; } } } else { cc = 0; } } else { cc = -1; } } /* For orderly release read from sockets until we have 0 bytes input. */ dk4sto_it_reset(i_clients); do { if (0 != dk_send_can_continue()) { pcl = (dk_send_client_t *)dk4sto_it_next(i_clients); if (NULL != pcl) { cc = 1; while (1 == cc) { rdbytes = sizeof(dk_send_buf); res = dk4socket_recv( pcl->sock, dk_send_buf, &rdbytes, 0, 0L, 0L, NULL ); if (DK4_SOCKET_RESULT_SUCCESS == res) { if (0 == rdbytes) { cc = 0; } } else { cc = 0; } if (0 == dk_send_can_continue()) { cc = 0; pcl = NULL; } } } } else { pcl = NULL; } } while (NULL != pcl); /* Finally clean up. */ finished: /* Close the connection sockets */ if ((NULL != i_clients) && (NULL != s_clients)) { dk4sto_it_reset(i_clients); do { pcl = (dk_send_client_t *)dk4sto_it_next(i_clients); if (NULL != pcl) { if (INVALID_SOCKET != pcl->sock) { dk4socket_close(pcl->sock, NULL); } } } while (NULL != pcl); } /* Close listener socket set */ if (NULL != listeners) { dk4socket_set_close(listeners, NULL); } /* Clean up connection container and iterator */ if (NULL != i_clients) { dk4sto_it_reset(i_clients); do { pcl = (dk_send_client_t *)dk4sto_it_next(i_clients); if (NULL != pcl) { dk4mem_free(pcl); } } while (NULL != pcl); dk4sto_it_close(i_clients); } if (NULL != s_clients) { dk4sto_close(s_clients); } $? "- dk_send_do_networking" } /** Process command line arguments, and continue. @param portname Port name to listen. @param infile Input file to read data from. */ static void dk_send_process_cmdline_args( const dkChar *portname, FILE *infile ) { const dkChar *argptr = NULL; const dkChar *endptr = NULL; const dkChar *argval = NULL; dkChar *argcpy = NULL; dk4_allowed_peer_t *ppeer; double timeout = 0.0; double td1 = 0.0; double td2 = 0.0; dk4_um_t um = (dk4_um_t)0UL; int res = 0; $? "+ dk_send_process_cmdline_args" /* Check whether to allow local connections only. */ if (0 != dk4app_opt_is_set_short(app, dkT('l'), NULL)) { local_only = 1; } if (0 != dk4app_opt_is_set_short(app, dkT('p'), NULL)) { is_pipe = 1; } /* Set up backlog */ if (0 != dk4app_opt_is_set_short(app, dkT('b'), NULL)) { if (0 != dk4app_opt_get_unsigned_short(&um, app, dkT('b'), NULL)) { if ((dk4_um_t)(INT_MAX) >= um) { if ((dk4_um_t)0UL < um) { backlog = (int)um; } else { backlog = 1; } } else { backlog = INT_MAX; } } else { /* ERROR: Not an unsigned number, must not happen */ goto finished; } } /* Find timeout settings. */ if (0 != dk4app_opt_is_set_short(app, dkT('c'), NULL)) { if (0 != dk4app_opt_get_double_short(&timeout, app, dkT('c'), NULL)) { if (0.0 <= timeout) { td1 = floor(timeout); td2 = timeout - td1; td2 = td2 * 1000000.0; if ((double)(LONG_MAX) >= td1) { dk_send_timeout_secs = (long)td1; dk_send_timeout_usecs = (long)td2; } else { /* ERROR: Timeout value too large! */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 16); goto finished; } } else { /* ERROR: Can not use negative timeout */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 17); goto finished; } } else { /* ERROR: Not timeout value found */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 18); goto finished; } } /* Find number of clients. */ if (0 != dk4app_opt_is_set_short(app, dkT('n'), NULL)) { argval = dk4app_opt_get_string_ptr_short(app, dkT('n'), NULL); if (dk4str_cmp(dk_send_kwnl[3], argval) == 0) { numclients = (dk4_um_t)0UL; } else { res = dk4ma_input_dk_dec_dk4_um_t( &numclients, argval, &endptr, 1, NULL ); if (0 == res) { /* ERROR: Invalid option argument */ dk4app_log_3( app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 10, 11, argval ); goto finished; } } } $? ". number of clients ok" /* Find allowed client addresses. */ if (0 != dk4app_opt_is_set_short(app, dkT('a'), NULL)) { argptr = dk4app_opt_get_string_ptr_short(app, dkT('a'), NULL); if (NULL == argptr) { /* ERROR: Failed to obtain Argument */ $? "! BUG, Failed to obtain option argument" goto finished; } argcpy = dk4str_dup_app(argptr, app); if (NULL == argcpy) { goto finished; } if (0 == dk_send_get_allowed_clients(argcpy)) { goto finished; } } $? ". remote address ok" /* Do the network transmission. */ dk_send_do_networking(portname, infile); $? ". going to release resources" /* Finally release resources. */ finished: if (NULL != argcpy) { dk4mem_free(argcpy); } if (NULL != s_allowed) { if (NULL != i_allowed) { dk4sto_it_reset(i_allowed); do { ppeer = (dk4_allowed_peer_t *)dk4sto_it_next(i_allowed); if (NULL != ppeer) { dk4mem_free(ppeer); } } while(NULL != ppeer); dk4sto_it_close(i_allowed); i_allowed = NULL; } dk4sto_close(s_allowed); s_allowed = NULL; } $? "- dk_send_process_cmdline_args" } /** Bring Winsock up before performing other steps. @param portname Port name to listen. @param infile Input file to read data from. */ static void dk_send_run_for_file( const dkChar *portname, FILE *infile ) { if (dk4socket_up_app(app)) { dk_send_process_cmdline_args(portname, infile); dk4socket_down_app(app); } } /** Run for real file name. @param portname Port name. @param filename File name, no further expansion required. */ static void dk_send_run_for_real_file_name(const dkChar *portname, const dkChar *filename) { FILE *inf; int tests = DK4_FOPEN_SC_USER; if (dk4isadmin()) { tests = DK4_FOPEN_SC_PRIVILEGED; } inf = dk4fopen_app(filename, dk_send_kwnl[5], tests, app); if (NULL != inf) { dk_send_run_for_file(portname, inf); fclose(inf); } } #if DK4_ON_WINDOWS /** Expand file name and run for file. @param portname Port name (first command line argument). @param filename File name pattern. */ static void dk_send_expand_filename(const dkChar *portname, const dkChar *filename) { dkChar buf[DK4_MAX_PATH]; dk4_dir_t *fne; const dkChar *fn; const dkChar *pt; size_t szbuf = DK4_SIZEOF(buf,dkChar); int copied = 0; fne = dk4app_fne_open(filename, app); if (NULL != fne) { fn = dk4dir_next_file(fne); if (NULL != fn) { pt = dk4dir_get_path(fne); if (NULL == dk4dir_next_file(fne)) { if (NULL != pt) { copied = 0; if (0 != dk4str_cpy_s(buf, szbuf, pt, NULL)) { if (0 != dk4str_cat_s(buf, szbuf, dk_send_kwnl[4], NULL)) { if (0 != dk4str_cat_s(buf, szbuf, fn, NULL)) { copied = 1; dk_send_run_for_real_file_name(portname, buf); } } } if (0 == copied) { /* ERROR: Path too long */ dk4app_log_base3(app, DK4_LL_ERROR, 98, 99, pt); } } else { dk_send_run_for_real_file_name(portname, fn); } } else { /* ERROR: Multiple matching files */ dk4app_log_base3(app, DK4_LL_ERROR, 100, 106, filename); } } else { /* ERROR: No such file */ dk4app_log_base3(app, DK4_LL_ERROR, 100, 107, filename); } dk4dir_close(fne); } } #endif static void dk_send_normal_run(void) { const dkChar *filename = NULL; int xargc = 0; #if DK4_ON_WINDOWS int oldmode = 0; #endif xargc = dk4app_get_argc(app); switch (xargc) { case 1: { #if DK4_ON_WINDOWS oldmode = _setmode(_fileno(stdin), _O_BINARY); #endif dk_send_run_for_file(dk4app_get_argv(app, 0), stdin); #if DK4_ON_WINDOWS _setmode(_fileno(stdin), oldmode); #endif } break; case 2: { filename = dk4app_get_argv(app, 1); #if DK4_ON_WINDOWS if (dk4path_must_expand(filename)) { dk_send_expand_filename(dk4app_get_argv(app, 0), filename); } else { #endif dk_send_run_for_real_file_name(dk4app_get_argv(app, 0), filename); #if DK4_ON_WINDOWS } #endif } break; default : { /* ERROR: Wrong number of command line arguments */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 12); } break; } } #if DK4_HAVE_SIGACTION /** Set signal handlers and run. */ static void dk_send_run_with_signal_handlers(void) { #ifdef SIGPIPE struct sigaction opipe; #endif struct sigaction oint; #ifdef SIGPIPE struct sigaction npipe; #endif struct sigaction nint; struct sigaction oterm; struct sigaction nterm; int success = 0; #ifdef SIGPIPE /* Set up signal handling for SIGPIPE. */ DK4_MEMRES(&npipe, sizeof(npipe)); npipe.sa_handler = sig_handler_pipe; npipe.sa_flags = 0; if (0 != sigemptyset(&npipe.sa_mask)) { /* ERROR: Failed to set up masked signal set for SIGPIPE */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto finished; } if (0 != sigaddset(&npipe.sa_mask, SIGPIPE)) { /* ERROR: Failed to set up masked signal set for SIGPIPE */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto finished; } if (0 != sigaction(SIGPIPE, &npipe, &opipe)) { /* ERROR: Failed to set up signal handler for SIGPIPE */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto finished; } #endif /* Set up signal handling for SIGINT. */ DK4_MEMRES(&nint, sizeof(nint)); nint.sa_handler = sig_handler_int; nint.sa_flags = 0; if (0 != sigemptyset(&nint.sa_mask)) { /* ERROR: Failed to set up masked signal set for SIGINT */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto restore_old_pipe; } if (0 != sigaddset(&nint.sa_mask, SIGINT)) { /* ERROR: Failed to set up masked signal set for SIGINT */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto restore_old_pipe; } if (0 != sigaction(SIGINT, &nint, &oint)) { /* ERROR: Failed to set up signal handler for SIGINT */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto restore_old_pipe; } /* Set up signal handling for SIGTERM */ DK4_MEMRES(&nterm, sizeof(nterm)); nterm.sa_handler = sig_handler_term; nterm.sa_flags = 0; if (0 != sigemptyset(&nterm.sa_mask)) { /* ERROR: Failed to set up masked signal set for SIGTERM */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto restore_old_int; } if (0 != sigaddset(&nterm.sa_mask, SIGTERM)) { /* ERROR: Failed to set up masked signal set for SIGTERM */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto restore_old_int; } if (0 != sigaction(SIGTERM, &nterm, &oterm)) { /* ERROR: Failed to set up signal handler for SIGTERM */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 0); goto restore_old_int; } success = 1; dk_send_normal_run(); /* Restore signal handling for SIGTERM. */ if (0 != sigaction(SIGTERM, &oterm, NULL)) { /* ERROR: Failed to restore old SIGTERM settings */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 1); success = 0; } /* Restore signal handling for SIGINT. */ restore_old_int: if (0 != sigaction(SIGINT, &oint, NULL)) { /* ERROR: Failed to restore old SIGPIPE settings */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 1); success = 0; } #ifdef SIGPIPE /* Restore signal handling for SIGPIPE. */ restore_old_pipe: if (0 != sigaction(SIGPIPE, &opipe, NULL)) { /* ERROR: Failed to restore old SIGPIPE settings */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 1); success = 0; } #endif /* Set exit status code if error occured. */ finished: if (0 == success) { exval = EXIT_FAILURE; } } #else #if DK4_HAVE_SIGSET /** Set signal handlers and run. */ static void dk_send_run_with_signal_handlers(void) { #ifdef SIGPIPE dk4_sig_handler_t *oldpipe = NULL; #endif dk4_sig_handler_t *oldint = NULL; dk4_sig_handler_t *oldterm = NULL; #ifdef SIGPIPE oldpipe = sigset(SIGPIPE, sig_handler_pipe); #endif oldint = sigset(SIGINT, sig_handler_int); oldterm = sigset(SIGTERM, sig_handler_term); dk_send_normal_run(); sigset(SIGTERM, oldterm); sigset(SIGINT, oldint); #ifdef SIGPIPE sigset(SIGPIPE, oldpipe); #endif } #else #if DK4_HAVE_SIGNAL /** Set signal handlers and run. */ static void dk_send_run_with_signal_handlers(void) { #ifdef SIGPIPE dk4_sig_handler_t *oldpipe = NULL; #endif dk4_sig_handler_t *oldint = NULL; dk4_sig_handler_t *oldterm = NULL; #ifdef SIGPIPE oldpipe = signal(SIGPIPE, sig_handler_pipe); #endif oldint = signal(SIGINT, sig_handler_int); oldterm = signal(SIGTERM, sig_handler_term); dk_send_normal_run(); signal(SIGTERM, oldterm); signal(SIGINT, oldint); #ifdef SIGPIPE signal(SIGPIPE, oldpipe); #endif } #else /** Set signal handlers and run. */ static void dk_send_run_with_signal_handlers(void) { dk_send_normal_run(); } #endif #endif #endif /** Main function. @param argc Number of command line arguments. @param argv Command line arguments array. @return 0 on success, all other values indicate errors. */ #if DK4_CHAR_SIZE > 1 int wmain(int argc, wchar_t *argv[]) #else int main(int argc, char *argv[]) #endif { $!trace-init dk-send.deb $? "+ main" dk4error_init(&ernb); app = dk4app_open_cmd( argc, argv, dk_send_options, dk_send_sz_options, dk_send_kwnl[0], DKT_VERSION_DK, dk_send_kwnl[1], dk_send_help_text, dk_send_license_text ); if (NULL != app) { dk_send_sz_msg = dk4app_string_table_size(dk_send_kw_def); dk_send_msg = dk4app_string_table(app, dk_send_kwnl[2], dk_send_kw_def); if (0 != dk4app_can_run_normally(app)) { dk_send_run_with_signal_handlers(); } else { if (0 != dk4app_help_version_license(app)) { exval = EXIT_SUCCESS; } } if (0 != sig_read_atomic(&sig_had_int)) { /* ERROR: INT signal */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 13); exval = EXIT_FAILURE; } if (0 != sig_read_atomic(&sig_had_term)) { /* ERROR: TERM signal */ dk4app_log_1(app, dk_send_msg, dk_send_sz_msg, DK4_LL_ERROR, 14); exval = EXIT_FAILURE; } #ifdef SIGPIPE if (0 != sig_read_atomic(&sig_had_pipe)) { exval = EXIT_FAILURE; } #endif dk4app_close(app); } $? "- main %d", exval $!trace-end exit(exval); return exval; }