root/trunk/src/routers_mgmt.c

/*
 * File: routers_mgmt.c
 * 
 * Authors: ANDREU Francois-Xavier 
 *
 * Copyright (C) 2005-2011 GIP RENATER 
 */

/*  This file is part of renetcol. 
 *
 *  renetcol 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 2 of the License, or
 *  (at your option) any later version.
 *
 *  renetcol 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 renetcol; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include "routers_mgmt.h"

RouterPtr notExistRouter(RouterPtr rL, unsigned long address)
{
  RouterPtr tmp = rL;
  int i = 0;
  for ( ; tmp; tmp=tmp->next) {
    i++;
    if ( tmp->IpAddress == address ){
      return tmp;
    }
  }
  return NULL;
}

RouterPtr addRouter(RouterPtr routersList, unsigned long address, unsigned short id) 
{
  RouterPtr tmp = (RouterPtr) malloc(sizeof(struct Router));
  if (tmp==NULL) {
    syslog(LOG_ERR, "ERROR in malloc in addRouter function\n");
    exit(1);
  } else {
    tmp->IpAddress = address;
    tmp->ID = id;
    tmp->tplList = NULL;
    tmp->tplOptList = NULL;
    tmp->engineList = NULL;
#if defined(MULTISAMPLING)
    tmp->sampled = 1;
#else
    tmp->sampled = 0;
#endif
    tmp->next = routersList;
    if (routersList!=NULL) { routersList->prev = tmp;}
    tmp->prev = NULL;
    return tmp;
  }
}

RouterPtr updateRouter(RouterPtr routersList, unsigned long address, 
                       unsigned long spled) 
{
  RouterPtr tmp = NULL;
  if ((tmp = notExistRouter(routersList, 
                            address))==NULL) {
    syslog(LOG_INFO,"Router Address not registered : %lu.%lu.%lu.%lu",
           (address>>24),
           (address<<8>>24),
           (address<<16>>24),
           (address<<24>>24));
    return routersList;
  } else {
    tmp->sampled = spled; 
    return tmp;
  }
  return NULL;
}

TplFlowSetPtr newRouterTplList()
{
  TplFlowSetPtr tmp = (TplFlowSetPtr) malloc(sizeof(struct TemplateFlowSet));
  tmp->sourceId = 0;
  tmp->templateFlowSetId = 99;
  tmp->fieldCount = 0;
  tmp->fieldSet = NULL;
  tmp->lastField = NULL;
  tmp->next = NULL;
  tmp->prev = NULL;
  return tmp;
}

TplFlowSetPtr deleteTplFlSet(TplFlowSetPtr ptpl)
{
  FieldPtr tmp = NULL;
  if (ptpl) {
    while ( ptpl->fieldSet != NULL ){
      if (!(ptpl->lastField->prev)) {
        freeField(ptpl->lastField);
        ptpl->lastField = NULL;
        ptpl->fieldSet = NULL;
      } else {
        tmp = ptpl->lastField->prev;
        tmp->next = NULL;
        freeField(ptpl->lastField);
        ptpl->lastField = tmp;
      }
    }
    free(ptpl);
    ptpl = NULL;
  }
  return NULL;
}

NDEEnginePtr newRouterEngineList()
{
  NDEEnginePtr tmp = (NDEEnginePtr) malloc(sizeof(struct NDEEngine));
  tmp->package_sequence = 0;
  tmp->engineId = 0;
  tmp->prev = NULL;
  tmp->next = NULL;
  return tmp;
}

/*
 * existEngId
 *
 * return:
 *    - a pointer on the EngineId or NULL
 */
NDEEnginePtr existEngId(RouterPtr cr, unsigned long sid)
{
  NDEEnginePtr tmp=cr->engineList;
  for (; tmp; tmp=tmp->next) {
    if (tmp->engineId==sid) {
      return tmp;
    }
  }
  return NULL;
}

TplOptionPtr newRouterTplOptList()
{
  TplOptionPtr tmp = (TplOptionPtr) malloc(sizeof(struct TemplateOption));
  tmp->sourceId = 0;
  tmp->templateOptionId = 66;
  tmp->length = 0;
  tmp->optionScopeLg = 0;
  tmp->optionLg = 0;
  tmp->fieldSet = NULL;
  tmp->lastField = NULL;
  tmp->next = NULL;
  tmp->prev = NULL;
  return tmp;
}

TplOptionPtr deleteTplOption(TplOptionPtr ptplo)
{
  FieldPtr tmp = NULL;
  if (ptplo) {
    while ( ptplo->fieldSet != NULL ){
      if (!(ptplo->lastField->prev)) {
        freeField(ptplo->lastField);
        ptplo->lastField = NULL;
        ptplo->fieldSet = NULL;
      } else {
        tmp = ptplo->lastField->prev;
        tmp->next = NULL;
        freeField(ptplo->lastField);
        ptplo->lastField = tmp;
      }
    }
    free(ptplo);
    ptplo = NULL;
  }
  return NULL; /* FIXME I don't known */
}

#ifdef IPV4AGGIDSNMP
int getSNMPIndexList(char *filename, RouterPtr routersListPtr)
{
  FILE *indexFile = NULL;
  char line[200];
  int cptLine = 0;
  char tindex[5];
  char ttoken[2];
  char tid[2];
  char tad[16];
  int index = 0;
  int i = 0;
  unsigned short n0, n1, n2, n3;
  unsigned char buffer4[4];
  unsigned long ipAddress;
  RouterPtr routerTmp = NULL;
  unsigned short snmpIndex = 0;
  unsigned short snmpCpt = 0;

  if (!(indexFile = fopen(filename, "r"))) {
    fprintf (stderr, "error during %s opening\n", filename);
    exit(1);
  }
  while ( fgets(line, 200, indexFile) != 0) {    
    cptLine++;
    if ( strspn(line, "R") == 1 ) {
      if (sscanf(line, "%s %s\n",
                 ttoken,
                 tad) == 0) {
        fprintf(stderr, "Error in file %s, line %d\n", 
                filename, cptLine);
        exit(1);
      }
      if (sscanf(tad,"%hu.%hu.%hu.%hu\n",&n0,&n1,&n2,&n3) == 0){
        fprintf(stderr, "Address error in file %s, line %d\n", 
                filename, cptLine);
        exit(1);
      }
      buffer4[3] = (unsigned char)n0;
      buffer4[2] = (unsigned char)n1;
      buffer4[1] = (unsigned char)n2;
      buffer4[0] = (unsigned char)n3;   
      ipAddress = *((unsigned long*)(&buffer4));
      if ((routerTmp = notExistRouter(routersListPtr,
                                      ipAddress))==NULL){
        fprintf(stderr, "Error in file %s, line %d : IP unknown : %lu\n",
                filename, cptLine, ipAddress);
        exit(1);
      }
      for (i=0; i<MAX_INDEX_BY_ROUTER; i++) {
        routerTmp->snmpIndexType[i] = 2;
        routerTmp->snmpIndexID[i] = 0;
      }      
      snmpCpt = 0;
    }else{
      if (snmpCpt==0){
        routerTmp->snmpIndexType[0] = 1;
        routerTmp->snmpIndexID[0] = 0;
        snmpCpt++;
      }
      if ( strspn(line, "I") == 1 ) {
        if (sscanf(line, "%s %s %s\n",
                   ttoken,
                   tindex,
                   tid) == 0) {
          fprintf(stderr, "Error in file %s, line %d\n", 
                  filename, cptLine);
          exit(1);
        }
        index = atoi(tindex);
        if ( strcmp(tid, "B") == 0 ) {
          routerTmp->snmpIndexList[index] = 0;
          routerTmp->snmpIndexID[snmpCpt] = index;
          routerTmp->snmpIndexType[snmpCpt] = 0;
/*        fprintf(stderr, "snmpCpt %hu --> 0 \n", snmpCpt); */
        } else if ( strcmp(tid, "C") == 0 ) {
          routerTmp->snmpIndexList[index] = 1;
          routerTmp->snmpIndexID[snmpCpt] = index;
          routerTmp->snmpIndexType[snmpCpt] = 1;
/*        fprintf(stderr, "snmpCpt %hu --> 1 , indexSNMP: %hu\n", snmpCpt, index); */
        } else {
          fprintf(stderr, "Error in file %s, line %d : bad code B or C \n", 
                  filename, cptLine);
          exit(1);
        }
        snmpCpt++;
      } else {
        fprintf(stderr, "Error in file %s, line %d : bad index line \n", 
                filename, cptLine);
        exit(1);
      }
    }
  }
  return 1;
}
#endif