release/src/router/udpxy/uopt.c

   1 /* @(#) option-associated functions for udpxy
   2  *
   3  * Copyright 2008-2011 Pavel V. Cherenkov (pcherenkov@gmail.com)
   4  *
   5  *  This file is part of udpxy.
   6  *
   7  *  udpxy is free software: you can redistribute it and/or modify
   8  *  it under the terms of the GNU General Public License as published by
   9  *  the Free Software Foundation, either version 3 of the License, or
  10  *  (at your option) any later version.
  11  *
  12  *  udpxy is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *  GNU General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU General Public License
  18  *  along with udpxy.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include <assert.h>
  22 #include <stdlib.h>
  23 #include <errno.h>
  24 #include <string.h>
  25
  26 #include "udpxy.h"
  27 #include "uopt.h"
  28 #include "util.h"
  29
  30
  31 static int
  32 read_http_footer (char* buf, size_t len)
  33 {
  34     const char* ev = getenv ("UDPXY_HTTP200_FTR_LN");
  35     ssize_t n = 0;
  36
  37     assert (buf && len);
  38
  39     /* if 1-line footer is specified */
  40     if (ev) {
  41         (void) strncpy (buf, ev, len-1);
  42         buf[len-1] = '\0';
  43         return 0;
  44     }
  45
  46     if (NULL != (ev = getenv ("UDPXY_HTTP200_FTR_FILE"))) {
  47         n = txtf_read (ev, buf, len, stderr);
  48     }
  49
  50     return (n<0) ? 1 : 0;
  51 }
  52
  53
  54 /* populate options with default/initial values
  55  */
  56 int
  57 init_uopt( struct udpxy_opt* uo )
  58 {
  59     int rc = 0;
  60     assert( uo );
  61
  62     uo->is_verbose      = uf_FALSE;
  63     uo->cl_tpstat       = uf_FALSE;
  64     uo->nice_incr       = 0;
  65     uo->rbuf_len        = DEFAULT_CACHE_LEN;
  66     uo->rbuf_msgs       = 1;
  67     uo->max_clients     = DEFAULT_CLIENT_COUNT;
  68     uo->rcv_tmout       = get_timeval( "UDPXY_RCV_TMOUT",
  69                                 RLY_SOCK_TIMEOUT );
  70     uo->dhold_tmout     = get_timeval( "UDPXY_DHOLD_TMOUT",
  71                                 DHOLD_TIMEOUT );
  72     uo->sr_tmout        = get_timeval ( "UDPXY_SREAD_TMOUT", SRVSOCK_TIMEOUT  );
  73     uo->sw_tmout        = get_timeval ( "UDPXY_SWRITE_TMOUT", SRVSOCK_TIMEOUT );
  74     uo->ssel_tmout      = get_timeval ( "UDPXY_SSEL_TMOUT", SSEL_TIMEOUT );
  75     uo->lq_backlog      = (int) get_sizeval ("UDPXY_LQ_BACKLOG", LQ_BACKLOG);
  76     uo->ss_rlwmark      = (int) get_sizeval ("UDPXY_SRV_RLWMARK", SRV_RLWMARK);
  77
  78     uo->nosync_sbuf     = (u_short)get_flagval( "UDPXY_SSOCKBUF_NOSYNC", 0 );
  79     uo->nosync_dbuf     = (u_short)get_flagval( "UDPXY_DSOCKBUF_NOSYNC", 0 );
  80
  81     uo->srcfile         = NULL;
  82     uo->dstfile         = NULL;
  83     uo->mcast_refresh   = DEFAULT_MCAST_REFRESH;
  84
  85     if (-1 == read_http_footer (uo->h200_ftr, sizeof(uo->h200_ftr))) {
  86         rc = -1; /* modify rc only if there is an error */
  87     }
  88     return rc;
  89 }
  90
  91
  92 /* release resources allocated for udpxy options
  93  */
  94 void
  95 free_uopt( struct udpxy_opt* uo )
  96 {
  97     assert( uo );
  98
  99     if( uo->srcfile ) {
 100         free( uo->srcfile ); uo->srcfile = NULL;
 101     }
 102
 103     if( uo->dstfile ) {
 104         free( uo->dstfile ); uo->dstfile = NULL;
 105     }
 106 }
 107
 108
 109 #ifdef UDPXREC_MOD
 110
 111 /* populate udpxrec options with default/initial values
 112  */
 113 int
 114 init_recopt( struct udpxrec_opt* ro )
 115 {
 116     int rc = 0;
 117
 118     assert( ro );
 119
 120     ro->is_verbose      = uf_FALSE;
 121     ro->nice_incr       = 0;
 122
 123     ro->bg_time = ro->end_time = 0;
 124     ro->max_fsize = 0;
 125     ro->bufsize = DEFAULT_CACHE_LEN;
 126     ro->rbuf_msgs = 1;
 127
 128     ro->mcast_addr[0]   = '\0';
 129
 130     ro->dstfile         = NULL;
 131     ro->pidfile         = NULL;
 132     ro->rec_channel[0]  = '\0';
 133     ro->rec_port = 0;
 134     ro->waitupd_sec = -1;
 135
 136     ro->nosync_sbuf  =
 137         (flag_t)get_flagval( "UDPXY_SSOCKBUF_NOSYNC", 0 );
 138     ro->nosync_dbuf =
 139         (flag_t)get_flagval( "UDPXY_DSOCKBUF_NOSYNC", 0 );
 140
 141     ro->rcv_tmout       = 0;
 142
 143     return rc;
 144 }
 145
 146
 147 /* release resources allocated for udpxy options
 148  */
 149 void
 150 free_recopt( struct udpxrec_opt* ro )
 151 {
 152     assert( ro );
 153
 154     if( ro->dstfile ) {
 155         free( ro->dstfile ); ro->dstfile = NULL;
 156     }
 157     if( ro->pidfile ) {
 158         free( ro->pidfile ); ro->pidfile = NULL;
 159     }
 160
 161     ro->rec_channel[0] = '\0';
 162 }
 163
 164
 165 /* print udpxrec options to stream
 166  */
 167 void
 168 fprint_recopt( FILE* stream, struct udpxrec_opt* ro )
 169 {
 170     assert( stream && ro );
 171
 172     if( ro->is_verbose ) {
 173         (void)fprintf( stream, "verbose=[ON] " );
 174     }
 175     if( ro->nice_incr ) {
 176         (void)fprintf( stream, "nice_incr=[%d] ", ro->nice_incr );
 177     }
 178     if( ro->bg_time > 0 ) {
 179         (void)fprintf( stream, "begin_time=[%s] ",
 180                 Zasctime( localtime(&(ro->bg_time) )) );
 181     }
 182     if( ro->end_time > 0 ) {
 183         (void)fprintf( stream, "end_time=[%s] ",
 184                 Zasctime( localtime(&(ro->end_time) )) );
 185     }
 186     if( ro->max_fsize > 0 ) {
 187         (void)fprintf( stream, "Max filesize=[%.0f] bytes ",
 188                 (double)ro->max_fsize );
 189     }
 190
 191     (void)fprintf( stream, "Buffer size=[%ld] bytes ",
 192             (long)ro->bufsize );
 193
 194     if( ro->rbuf_msgs > 0 ) {
 195         (void)fprintf( stream, "Max messages=[%ld] ",
 196                 (long)ro->rbuf_msgs );
 197     }
 198     if( ro->mcast_addr[0] ) {
 199         (void)fprintf( stream, "Multicast interface=[%s] ", ro->mcast_addr );
 200     }
 201     if( ro->rec_channel[0] ) {
 202         (void)fprintf( stream, "Channel=[%s:%d] ", ro->rec_channel, ro->rec_port );
 203     }
 204     if( ro->pidfile ) {
 205         (void)fprintf( stream, "Pidfile=[%s] ", ro->pidfile );
 206     }
 207     if( ro->dstfile ) {
 208         (void)fprintf( stream, "Destination file=[%s] ", ro->dstfile );
 209     }
 210     if( ro->waitupd_sec > 0 ) {
 211         (void)fprintf( stream, "Update-wait=[%d] sec ", ro->waitupd_sec );
 212     }
 213
 214     (void) fputs( "\n", stream );
 215     return;
 216 }
 217
 218 #endif /* UDPXREC_MOD */
 219
 220 /* set verbose output on
 221  */
 222 void
 223 set_verbose( flag_t* verbose )
 224 {
 225     assert( verbose );
 226     *verbose = uf_TRUE;
 227 }
 228
 229
 230
 231 /* __EOF__ */
 232