conf.c

Go to the documentation of this file.

/*-------------------------------------------------------------------------
 *--------------------------- RT-WMP IP INTERFACE -------------------------
 *-------------------------------------------------------------------------
 *
 * File: conf.c
 * Authors: Rubén Durán
 *          Danilo Tardioli
 *-------------------------------------------------------------------------
 *  Copyright (C) 2011, Universidad de Zaragoza, SPAIN
 *
 *  This program 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 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 *-----------------------------------------------------------------------*/

#include "config/compiler.h"
#include "core/include/global.h"
#include <linux/version.h>
#include <linux/proc_fs.h>
#include <linux/module.h>
#include <linux/kernel.h>
#include <asm/uaccess.h>
#include "conf.h"

static struct proc_dir_entry *directory;

/* Creates /proc/rt-wmp/ and all the files inside it */
int conf_init_proc(void) {
        printk(KERN_ERR "Initializing /proc files (net device)...");
        directory = get_proc_root();
        directory = proc_mkdir("port", directory);
        if (!directory) {
                remove_proc_entry("port");
                return -ENOMEM;
        }
        return 1;
}


int conf_close_proc(tpConfig *conf) {
        int i;
        for (i = 0; i < conf->udp_in.count; i++) {
                remove_proc_entry(conf->udp_in.port[i].proc_entry->name, directory);
        };

        for (i = 0; i < conf->tcp_out.count; i++) {
                remove_proc_entry(conf->tcp_out.port[i].proc_entry->name, directory);
        };

        for (i = 0; i < conf->udp_out.count; i++) {
                remove_proc_entry(conf->udp_out.port[i].proc_entry->name, directory);
        };

        for (i = 0; i < conf->tcp_in.count; i++) {
                remove_proc_entry(conf->tcp_in.port[i].proc_entry->name, directory);
        };
    remove_proc_entry(directory->name, directory->parent);
        return 1;
}

int f_port_read(char *page, char **start, off_t off, int count, int *eof,
                void *data) {
        if (off > 0) {
                *eof = 1;
                return 0;
        } else {
                tpPort * tp = (tpPort*) data;
                return sprintf(page, "%u\n", tp->priority);
        }
}



int f_port_write(struct file *f, const char __user *buff, unsigned long len, void *data ){
        char aux[32];
        tpPort * tp;
        int val, size = (len>32?32:len);
        
        if (copy_from_user(aux, buff, size)) {
          return -EFAULT;
        }
        aux[size-1]='\0';

        val = atoi(aux);
        tp = (tpPort*) data;
        tp->priority = val;

        return len;
}



void asign_port (tpPortConf* portConf, u16 number, signed char priority, tpTraffic traffic, int type) {
   if (number == 0) {             // Global configuration
      portConf->count = -1;
      portConf->port[0].number = number;
      portConf->port[0].priority = priority;
      portConf->port[0].traffic = traffic;
   } else {
      /* If there is no global configuration and space for another port,
         store the configuration for the given port number */
      if(portConf->count != -1 && portConf->count < NUM_PORTS) {
         portConf->port[portConf->count].number = number;
         portConf->port[portConf->count].priority = priority;
         portConf->port[portConf->count].traffic = traffic;
         /* PROC */
         {
                         char txt[64];
                         if (type == 1){
                                 sprintf(txt,"udp_in_%d",number);
                         }else if (type == 2){
                                 sprintf(txt,"udp_out_%d",number);
                         } if (type == 3){
                                 sprintf(txt,"tcp_in_%d",number);
                         } if (type == 4){
                                 sprintf(txt,"tcp_out_%d",number);
                         }

                 portConf->port[portConf->count].proc_entry = create_proc_entry(txt, 0666, directory);
                         if(!portConf->port[portConf->count].proc_entry){
                                //__close_proc(0);
                         }
                         portConf->port[portConf->count].proc_entry->read_proc = f_port_read;
                         portConf->port[portConf->count].proc_entry->write_proc = f_port_write;

                         portConf->port[portConf->count].proc_entry->data = (void*) &(portConf->port[portConf->count]);
                         //portConf->port[portConf->count].proc_entry->write_proc = f_CpuDelay_read;
         }
         portConf->count++;
      }
   }
}

void asignConfICMP (tpPortConfICMP* portConf, signed char priority, tpTraffic traffic) {
   if (portConf->count == 0) {
      portConf->count++;
      portConf->priority = priority;
      portConf->traffic = traffic;
   } else {
      printk(KERN_INFO "ICMP already has a configuration.");
   }
}

int readConfig(tpConfig *conf) {
   struct file *f;
   char proto[6], dir[5], traf[10], line[256];
   tpTraffic traffic;
   int port, priority;
   int exists, ret = 0;

   /* Initializations */
   conf->udp_in.count=0;
   conf->udp_out.count=0;
   conf->tcp_in.count=0;
   conf->tcp_out.count=0;
   conf->icmp.count=0;

   /* Open thee file */
   f = filp_open("/etc/rt-wmp/interface.cfg", O_RDONLY, 0);
   if (IS_ERR(f)) {
      printk(KERN_INFO "File /etc/rt-wmp/interface.cfg not found...\n");
      return 0;
   }

   /* Read the configuration */
   printk(KERN_INFO "Reading /etc/rt-wmp/interface.cfg...\n");
   while (fgets(line,256,f) != NULL) {
      if (line[0]<65 || line[0]>90){
         continue;
      }

      if(sscanf(line,"%s %s %d %d %s",proto,dir, &port, &priority, traf) != 5){
         printk(KERN_INFO "MALFORMED: %s\n",line);
         continue;
      }

      if (strcmp(traf, "NORMAL") == 0) {
         traffic = OTHER;
      } else if (strcmp(traf, "QoS") == 0) {
         traffic = QoS;
      } else {
         printk(KERN_INFO "Wrong type of traffic: %s\n",traf);
         continue;
      }

      exists = 0;
      if (strcmp(proto, "UDP") == 0) {
         if (strcmp(dir, "FROM") == 0) {
            asign_port(&conf->udp_in, port, priority, traffic, 1);
            exists = 1;
         } else if (strcmp(dir, "TO") == 0) {
            asign_port(&conf->udp_out, port, priority, traffic, 2);
            exists = 1;
         } else if (strcmp(dir, "BOTH") == 0) {
            asign_port(&conf->udp_in, port, priority, traffic, 1);
            asign_port(&conf->udp_out, port, priority, traffic, 2);
            exists = 1;
         } else {
            printk(KERN_INFO "*** UNKNOWN DIRECTION %s\n", dir);
         }
      } else if (strcmp(proto, "TCP") == 0) {
         if (strcmp(dir, "FROM") == 0) {
            asign_port(&conf->tcp_in, port, priority, traffic, 3);
            exists = 1;
         } else if (strcmp(dir, "TO") == 0) {
            asign_port(&conf->tcp_out, port, priority, traffic, 4);
            exists = 1;
         } else if (strcmp(dir, "BOTH") == 0) {
            asign_port(&conf->tcp_in, port, priority, traffic, 3);
            asign_port(&conf->tcp_out, port, priority, traffic, 4);
            exists = 1;
         } else {
            printk(KERN_INFO "*** UNKNOWN DIRECTION %s\n", dir);
         }
      } else if (strcmp(proto, "ICMP") == 0) {
         asignConfICMP(&conf->icmp, priority, traffic);
         exists = 1;
      } else {
         printk(KERN_INFO "*** UNKNOWN PROTOCOL %s\n", proto);
      }

      if (exists) {
         printk(KERN_INFO "READ: %s %s %d %d %s\n", proto, dir, port, priority, traf);
      }
      ret = (ret || exists);
   }
   printk(KERN_INFO "Done.\n");

   /* Close the file and return */
   filp_close(f,0);

   // Will be TRUE if one or more ports have been read

   return ret;
}

int __getPortConf (tpPortConf* portConf, u16 port, signed char *priority, tpTraffic *traffic) {
   int i, ret = 0;
   if(portConf->count == -1) {            // Same config for every port
      *priority = portConf->port[0].priority;
      *traffic = portConf->port[0].traffic;
      ret = 1;
   } else {
      for(i=0; i<portConf->count && !ret; i++){
         ret = (port == portConf->port[i].number);
         if(ret) {
            *priority = portConf->port[i].priority;
            *traffic = portConf->port[i].traffic;
         }
      }
   }
   return ret;
}

int getPortConf (tpConfig *conf, tpProto proto, tpDir dir, u16 port, signed char *priority, tpTraffic *traffic){
   int ret = 0;

   switch(proto){
      case UDP:
         if (dir == FROM) {
            ret = __getPortConf(&conf->udp_in, port, priority, traffic);
         } else {
            ret = __getPortConf(&conf->udp_out, port, priority, traffic);
         }
         break;
      case TCP:
         if (dir == FROM) {
            ret = __getPortConf(&conf->tcp_in, port, priority, traffic);
         } else {
            ret = __getPortConf(&conf->tcp_out, port, priority, traffic);
         }
         break;
   }

   return ret;
}

int getConfICMP (tpConfig* conf, signed char *priority, tpTraffic *traffic) {
   int ret = 0;
   if(conf->icmp.count > 0) {
      *priority = conf->icmp.priority;
      *traffic = conf->icmp.traffic;
      ret = 1;
   }
   return ret;
}