td_unit.c

   1 /*
   2  * Copyright (c) 2004 Teodor Sigaev <teodor@sigaev.ru>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *      notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *      notice, this list of conditions and the following disclaimer in the
  12  *      documentation and/or other materials provided with the distribution.
  13  * 3. Neither the name of the author nor the names of any co-contributors
  14  *      may be used to endorse or promote products derived from this software
  15  *      without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
  18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
  21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  23  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  24  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
  25  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  26  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
  27  * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29 #include <stdlib.h>
  30 #include <string.h>
  31
  32 #include "tlog.h"
  33 #include "connection.h"
  34 #include "top.h"
  35 #include "tmalloc.h"
  36
  37
  38 MassiveUnit*
  39 allocUnits(int n ) {
  40         MassiveUnit *ptr = NULL;
  41
  42         ptr = (MassiveUnit*)malloc( MUHDRSZ + n * sizeof(UnitInfo) );
  43         if ( !ptr )
  44                 tlog(TL_CRIT|TL_EXIT, "allocUnits: no memoru for %d units", n);
  45         ptr->maxnumber = n;
  46         ptr->number=0;
  47
  48         return ptr;
  49 }
  50
  51 static int
  52 cmpUnit(const void *a, const void *b) {
  53         if ( ((UnitInfo*)a)->ip.s_addr == ((UnitInfo*)b)->ip.s_addr )
  54                 return 0;
  55         return ( ((UnitInfo*)a)->ip.s_addr > ((UnitInfo*)b)->ip.s_addr ) ? 1 : -1;
  56 }
  57
  58 void
  59 fillUnit( MassiveUnit* mu, UnitInfo *unit ) {
  60         if ( mu->number==0 || ( mu->number==1 && unit->ip.s_addr == mu->units->ip.s_addr ) ) {
  61                 memcpy( mu->units, unit, sizeof(UnitInfo) );
  62                 mu->number=1;
  63         } else {
  64                 UnitInfo *ptr = bsearch(unit, mu->units, mu->number, sizeof(UnitInfo), cmpUnit);
  65                 if ( ptr )
  66                         memcpy( ptr, unit, sizeof(UnitInfo) );
  67                 else {
  68                         if ( mu->number == mu->maxnumber )
  69                                 tlog(TL_ALARM, "Can't add new unit");
  70                         else {
  71                                 ptr = mu->units + mu->number;
  72                                 memcpy( ptr, unit, sizeof(UnitInfo) );
  73                                 mu->number++;
  74                                 qsort(mu->units, mu->number, sizeof(UnitInfo), cmpUnit);
  75                         }
  76                 }
  77         }
  78 }
  79
  80