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*)tmalloc( MUHDRSZ + n * sizeof(UnitInfo) );
  43         ptr->maxnumber = n;
  44         ptr->number=0;
  45
  46         return ptr;
  47 }
  48
  49 static int
  50 cmpUnit(const void *a, const void *b) {
  51         if ( ((UnitInfo*)a)->ip.s_addr == ((UnitInfo*)b)->ip.s_addr )
  52                 return 0;
  53         return ( ((UnitInfo*)a)->ip.s_addr > ((UnitInfo*)b)->ip.s_addr ) ? 1 : -1;
  54 }
  55
  56 void
  57 fillUnit( MassiveUnit* mu, UnitInfo *unit ) {
  58         if ( mu->number==0 || ( mu->number==1 && unit->ip.s_addr == mu->units->ip.s_addr ) ) {
  59                 memcpy( mu->units, unit, sizeof(UnitInfo) );
  60                 mu->number=1;
  61         } else {
  62                 UnitInfo *ptr = bsearch(unit, mu->units, mu->number, sizeof(UnitInfo), cmpUnit);
  63                 if ( ptr )
  64                         memcpy( ptr, unit, sizeof(UnitInfo) );
  65                 else {
  66                         if ( mu->number == mu->maxnumber )
  67                                 tlog(TL_ALARM, "Can't add new unit");
  68                         else {
  69                                 ptr = mu->units + mu->number;
  70                                 memcpy( ptr, unit, sizeof(UnitInfo) );
  71                                 mu->number++;
  72                                 qsort(mu->units, mu->number, sizeof(UnitInfo), cmpUnit);
  73                         }
  74                 }
  75         }
  76 }
  77
  78