configparser.c

Go to the documentation of this file.

#include "config.h"


#include <libxml/parser.h>
#include <stdio.h>
#include <string.h>

extern int errno;
    
static xmlValidCtxt cvp;    
static inline int parser_init()
{
    cvp.userData = (void*)stderr;
    cvp.error = (xmlValidityErrorFunc) fprintf;
    cvp.warning = (xmlValidityWarningFunc) fprintf;
    return 1;
}

static inline void parser_destroy()
{
    xmlCleanupParser();
}

static inline void quote_trim_dup(str *dest, char *src)
{
    int i=0;
    dest->s=0;
    dest->len=0;
    if (!src) return;
    dest->len = strlen(src);
    if (src[0]=='\"') {i++;dest->len--;}
    if (src[dest->len-1]=='\"') {dest->len--;}

    dest->s = shm_malloc(dest->len+1);
    if (!dest->s) {
        LOG_NO_MEM("shm",dest->len);
        dest->len=0;
        return;
    }
    memcpy(dest->s,src+i,dest->len);
    dest->s[dest->len]=0;
}

dp_config* parse_dp_config(char* filename)
{
    FILE *f=0;
    dp_config *x=0;
    xmlDocPtr doc=0;
    xmlNodePtr root=0,child=0,nephew=0;
    xmlChar *xc=0;
    int k;
    routing_entry *re,*rei;
    routing_realm *rr,*rri;

    parser_init();

    if (!filename){
        LOG(L_ERR,"ERROR:parse_dp_config(): filename parameter is null\n");
        goto error;
    }
    f = fopen(filename,"r");
    if (!f){
        LOG(L_ERR,"ERROR:parse_dp_config(): Error opening <%s> file > %s\n",filename,strerror(errno));
        goto error;
    }
    fclose(f);
    
    
    x = new_dp_config();
    if (!f){
        LOG(L_ERR,"ERROR:parse_dp_config(): Error opening <%s> file > %s\n",filename,strerror(errno));
        goto error;
    }

    doc = xmlParseFile(filename);
    if (!doc){
        LOG(L_ERR,"ERR:parse_dp_config():  This is not a valid XML file <%s>\n",
            filename);
        goto error;
    }

    root = xmlDocGetRootElement(doc);
    if (!root){
        LOG(L_ERR,"ERR:parse_dp_config():  Empty XML <%s>\n",filename);
        goto error;
    }

    k = xmlStrlen(root->name);
    if (k>12) k = 12;
    if (strncasecmp((char*)root->name,"DiameterPeer",k)!=0){
        LOG(L_ERR,"ERR:parse_dp_config(): XML Root is not <DiameterPeer>\n");
        goto error;
    }

    xc = xmlGetProp(root,(xmlChar*)"FQDN");
    quote_trim_dup(&(x->fqdn),(char*)xc);
    quote_trim_dup(&(x->identity),(char*)xc);

    xc = xmlGetProp(root,(xmlChar*)"Realm");
    quote_trim_dup(&(x->realm),(char*)xc);
    
    xc = xmlGetProp(root,(xmlChar*)"Vendor_Id");
    x->vendor_id = atoi((char*)xc);

    xc = xmlGetProp(root,(xmlChar*)"Product_Name");
    quote_trim_dup(&(x->product_name),(char*)xc);

    xc = xmlGetProp(root,(xmlChar*)"AcceptUnknownPeers");
    x->accept_unknown_peers = atoi((char*)xc);
    
    xc = xmlGetProp(root,(xmlChar*)"DropUnknownOnDisconnect");
    x->drop_unknown_peers = atoi((char*)xc);
    
    xc = xmlGetProp(root,(xmlChar*)"Tc");
    x->tc = atoi((char*)xc);

    xc = xmlGetProp(root,(xmlChar*)"Workers");
    x->workers = atoi((char*)xc);

    xc = xmlGetProp(root,(xmlChar*)"QueueLength");
    x->queue_length = atoi((char*)xc);

    for(child = root->children; child; child = child->next)
        if (child->type == XML_ELEMENT_NODE)
    {
        if (xmlStrlen(child->name)==4 && strncasecmp((char*)child->name,"Peer",4)==0){
            //PEER
            x->peers_cnt++;     
        }
        else if (xmlStrlen(child->name)==8 && strncasecmp((char*)child->name,"Acceptor",8)==0){
            //Acceptor
            x->acceptors_cnt++;     
        }
        else if (xmlStrlen(child->name)==4 && (strncasecmp((char*)child->name,"Auth",4)==0||
            strncasecmp((char*)child->name,"Acct",4)==0)){
            //Application
            x->applications_cnt++;      
        }   
    }
    x->peers = shm_malloc(x->peers_cnt*sizeof(peer_config));
    if (!x->peers){
        LOG_NO_MEM("shm",x->peers_cnt*sizeof(peer_config));
        goto error;
    }
    memset(x->peers,0,x->peers_cnt*sizeof(peer_config));
    x->peers_cnt=0;
    x->acceptors = shm_malloc(x->acceptors_cnt*sizeof(acceptor_config));
    if (!x->acceptors){
        LOG_NO_MEM("shm",x->acceptors_cnt*sizeof(acceptor_config));
        goto error;
    }
    memset(x->acceptors,0,x->acceptors_cnt*sizeof(acceptor_config));
    x->acceptors_cnt=0;
    x->applications = shm_malloc(x->applications_cnt*sizeof(app_config));
    if (!x->applications){
        LOG_NO_MEM("shm",x->applications_cnt*sizeof(app_config));
        goto error;
    }
    memset(x->applications,0,x->applications_cnt*sizeof(app_config));
    x->applications_cnt=0;

    for(child = root->children; child; child = child->next)
        if (child->type == XML_ELEMENT_NODE)
    {
        if (xmlStrlen(child->name)==4 && strncasecmp((char*)child->name,"Peer",4)==0){
            //PEER
            xc = xmlGetProp(child,(xmlChar*)"FQDN");
            quote_trim_dup(&(x->peers[x->peers_cnt].fqdn),(char*)xc);
            xc = xmlGetProp(child,(xmlChar*)"Realm");
            quote_trim_dup(&(x->peers[x->peers_cnt].realm),(char*)xc);          
            xc = xmlGetProp(child,(xmlChar*)"port");
            x->peers[x->peers_cnt].port = atoi((char*)xc);                      
            x->peers_cnt++;     
        }
        else if (xmlStrlen(child->name)==8 && strncasecmp((char*)child->name,"Acceptor",8)==0){
            //Acceptor
            xc = xmlGetProp(child,(xmlChar*)"bind");            
            quote_trim_dup(&(x->acceptors[x->acceptors_cnt].bind),(char*)xc);           
            xc = xmlGetProp(child,(xmlChar*)"port");
            x->acceptors[x->acceptors_cnt].port = atoi((char*)xc);                      
            x->acceptors_cnt++;     
        }
        else if (xmlStrlen(child->name)==4 && ((char*)strncasecmp((char*)child->name,"Auth",4)==0||
            strncasecmp((char*)child->name,"Acct",4)==0)){
            //Application
            xc = xmlGetProp(child,(xmlChar*)"id");          
            x->applications[x->applications_cnt].id = atoi((char*)xc);                      
            xc = xmlGetProp(child,(xmlChar*)"vendor");
            x->applications[x->applications_cnt].vendor = atoi((char*)xc);                      
            if (child->name[1]=='u'||child->name[1]=='U')
                x->applications[x->applications_cnt].type = DP_AUTHORIZATION;                       
            else
                x->applications[x->applications_cnt].type = DP_ACCOUNTING;                                      
            x->applications_cnt++;      
        }   
        else if (xmlStrlen(child->name)==12 && ((char*)strncasecmp((char*)child->name,"DefaultRoute",12)==0)){
            if (!x->r_table) {
                x->r_table = shm_malloc(sizeof(routing_table));
                memset(x->r_table,0,sizeof(routing_table));
            }
            re = new_routing_entry();
            if (re){            
                xc = xmlGetProp(child,(xmlChar*)"FQDN");
                quote_trim_dup(&(re->fqdn),(char*)xc);          
                xc = xmlGetProp(child,(xmlChar*)"metric");          
                re->metric = atoi((char*)xc);           
                
                /* add it the list in ascending order */
                if (! x->r_table->routes || re->metric <= x->r_table->routes->metric){
                    re->next = x->r_table->routes;
                    x->r_table->routes = re;
                }else{
                    for(rei=x->r_table->routes;rei;rei=rei->next)
                        if (!rei->next){
                            rei->next = re;
                            break;                      
                        }else{
                            if (re->metric <= rei->next->metric){
                                re->next = rei->next;
                                rei->next = re;
                                break;
                            }
                        }               
                }
            }                   
        }
        else if (xmlStrlen(child->name)==5 && ((char*)strncasecmp((char*)child->name,"Realm",5)==0)){
            if (!x->r_table) {
                x->r_table = shm_malloc(sizeof(routing_table));
                memset(x->r_table,0,sizeof(routing_table));
            }
            rr = new_routing_realm();
            if (rr){            
                xc = xmlGetProp(child,(xmlChar*)"name");
                quote_trim_dup(&(rr->realm),(char*)xc);         
                
                if (!x->r_table->realms) {              
                    x->r_table->realms = rr;
                }else{              
                    for(rri=x->r_table->realms;rri->next;rri=rri->next);
                    rri->next = rr;             
                }           
                for(nephew = child->children; nephew; nephew = nephew->next)
                    if (nephew->type == XML_ELEMENT_NODE){
                        if (xmlStrlen(nephew->name)==5 && ((char*)strncasecmp((char*)nephew->name,"Route",5)==0))
                        {
                            re = new_routing_entry();
                            if (re) {
                                xc = xmlGetProp(nephew,(xmlChar*)"FQDN");
                                quote_trim_dup(&(re->fqdn),(char*)xc);          
                                xc = xmlGetProp(nephew,(xmlChar*)"metric");         
                                re->metric = atoi((char*)xc);           
                                
                                /* add it the list in ascending order */
                                if (! rr->routes || re->metric <= rr->routes->metric){
                                    re->next = rr->routes;
                                    rr->routes = re;
                                }else{
                                    for(rei=rr->routes;rei;rei=rei->next)
                                        if (!rei->next){
                                            rei->next = re;
                                            break;                      
                                        }else{
                                            if (re->metric <= rei->next->metric){
                                                re->next = rei->next;
                                                rei->next = re;
                                                break;
                                            }
                                        }                   
                                }
                            }
                        }
                    }
            }       
        }
    }
    
    if (doc) xmlFreeDoc(doc);   
    parser_destroy();
    return x;
error:
    if (doc) xmlFreeDoc(doc);
    parser_destroy();
    if (x) free_dp_config(x);
    return 0;   
}

---