/* @(#) $Id: manage_agents.c,v 1.26 2007/07/26 01:32:02 dcid Exp $ */

/* Copyright (C) 2003-2007 Daniel B. Cid <dcid@ossec.net>
 * All rights reserved.
 *
 * This program is a free software; you can redistribute it
 * and/or modify it under the terms of the GNU General Public
 * License (version 3) as published by the FSF - Free Software
 * Foundation.
 *
 * License details at the LICENSE file included with OSSEC or 
 * online at: http://www.ossec.net/en/licensing.html
 */


/* Manage agents tool
 * Add/extract and remove agents from a server.
 */


#include "manage_agents.h"
#include "os_crypto/md5/md5_op.h"



/* Global internal variables */



/* chomp: remove spaces, new lines, etc from a string */
char *chomp(char *str)
{
    char *tmp_str;
    int size = 0;

    /* Removing spaces from the beginning */
    while(*str == ' ' || *str == '\t')
        str++;
    
    
    /* Removing any trailing new lines or \r */
    do
    {
        tmp_str = strchr(str, '\n');
        if(tmp_str)
        {
            *tmp_str = '\0';
            continue;
        }

        tmp_str = strchr(str, '\r');
        if(tmp_str)
        {
            *tmp_str = '\0';
        }
    }while(tmp_str != NULL);

    
    /* Removing spaces at the end of the string */
    tmp_str = str;
    size = strlen(str)-1;
    
    while((size >= 0) && (tmp_str[size] == ' ' || tmp_str[size] == '\t'))
    {
        tmp_str[size] = '\0';
        size--;
    }
    
    return(str);
}



/* Add an agent */
int add_agent()
{
    int i = 1;
    FILE *fp;
    char str1[STR_SIZE +1];
    char str2[STR_SIZE +1];
    
    os_md5 md1;
    os_md5 md2;
    
    char *user_input;
    char *_name;
    char *_id;
    char *_ip;

    char name[FILE_SIZE +1];
    char id[FILE_SIZE +1];
    char ip[FILE_SIZE +1];
    os_ip *c_ip;


    /* Allocating for c_ip */
    os_calloc(1, sizeof(os_ip), c_ip);
    
    
    /* Checking if we can open the auth_file */
    fp = fopen(AUTH_FILE,"a");
    if(!fp)
    {
        ErrorExit(FOPEN_ERROR, ARGV0, AUTH_FILE);
    }
    fclose(fp);

    #ifndef WIN32
    chmod(AUTH_FILE, 0440);
    #endif
    
    /* Setting time 2 */
    time2 = time(0);

    
    /* Source is time1+ time2 +pid + ppid */
    #ifndef WIN32
    srand(time2 + time1 + getpid() + getppid());
    #else
    srand(time2 + time1 + getpid());
    #endif
    rand1 = rand();

    
    /* Zeroing strings */
    memset(str1,'\0', STR_SIZE +1);
    memset(str2,'\0', STR_SIZE +1);


    printf(ADD_NEW);

    
    /* Getting the name */
    memset(name, '\0', FILE_SIZE +1);

    do
    {
        printf(ADD_NAME);
        fflush(stdout);
        _name = read_from_user();

        if(strcmp(_name, QUIT) == 0)
            return(0);

        strncpy(name, _name, FILE_SIZE -1);

        /* check the name */
        if(!OS_IsValidName(name))
            printf(INVALID_NAME,name);

        /* Search for name  -- no duplicates */
        if(NameExist(name))
            printf(ADD_ERROR_NAME, name);

    } while(NameExist(name) || !OS_IsValidName(name));


    /* Getting IP */
    memset(ip, '\0', FILE_SIZE +1);

    do
    {
      printf(ADD_IP);
      fflush(stdout);
    
      _ip = read_from_user();
      
      /* quit */
      if(strcmp(_ip, QUIT) == 0)
          return(0);
                              
      strncpy(ip, _ip, FILE_SIZE -1);
      
      if(!OS_IsValidIP(ip, c_ip))
      {
          printf(IP_ERROR, ip);
          _ip = NULL;
      }

    } while(!_ip);
   
    
    do
    {
        /* Default ID */
        i = 1024;
        snprintf(id, 8, "%03d", i);
        while(!IDExist(id))
        {
            i--;
            snprintf(id, 8, "%03d", i);

            /* No key present, use id 0 */
            if(i <= 0)
            {
                i = 0;
                break;
            }
        }
        snprintf(id, 8, "%03d", i+1);

        /* Getting ID */
        printf(ADD_ID, id);
        fflush(stdout);

        _id = read_from_user();



        /* quit */
        if(strcmp(_id, QUIT) == 0)
            return(0);


        if(_id[0] != '\0')
        {
            strncpy(id, _id, FILE_SIZE -1);
        }

        if(!OS_IsValidID(id))
            printf(INVALID_ID, id);

        /* Search for ID KEY  -- no duplicates */
        if(IDExist(id))
            printf(ADD_ERROR_ID, id);

    } while(IDExist(id) || !OS_IsValidID(id));
    
    

    printf(AGENT_INFO, id, name, ip);
    fflush(stdout);

    do
    {
      printf(ADD_CONFIRM);
      user_input = read_from_user();
   
      /* If user accepts to add */ 
      if(user_input[0] == 'y' || user_input[0] == 'Y')
      {
        time3 = time(0);
        rand2 = rand();

        fp = fopen(AUTH_FILE,"a");
        if(!fp)
        {
            ErrorExit(FOPEN_ERROR, ARGV0, KEYS_FILE);
        }
        #ifndef WIN32
        chmod(AUTH_FILE, 0440);
        #endif
                
        
        /* Random 1: Time took to write the agent information.
         * Random 2: Time took to choose the action.
         * Random 3: All of this + time + pid
         * Random 4: Md5 all of this + the name, key and ip
         * Random 5: Final key
         */
        
        snprintf(str1, STR_SIZE, "%d%s%d",time3-time2, name, rand1);
        snprintf(str2, STR_SIZE, "%d%s%s%d", time2-time1, ip, id, rand2);

        OS_MD5_Str(str1, md1);
        OS_MD5_Str(str2, md2);

        snprintf(str1, STR_SIZE, "%s%d%d%d",md1,(int)getpid(),rand(), time3);
        OS_MD5_Str(str1, md1);

        fprintf(fp,"%s %s %s %s%s\n",id, name, c_ip->ip, md1,md2);

        fclose(fp);

        printf(AGENT_ADD);
        restart_necessary = 1;
        break;
      }
      else if(user_input[0] == 'n' || user_input[0] == 'N')
      {
        printf(ADD_NOT);
        break;
      }

    } while(1);

    return(0);
}


/* remove an agent */
int remove_agent()
{
    FILE *fp;
    char *user_input;
    char u_id[FILE_SIZE +1];
    
    u_id[FILE_SIZE] = '\0';

    if(!print_agents())
    {
        printf(NO_AGENT);
        return(0);
    }

    do
    {
      printf(REMOVE_ID);
      fflush(stdout);

      user_input = read_from_user();

      if(strcmp(user_input, QUIT) == 0)
          return(0);

      strncpy(u_id, user_input, FILE_SIZE);

      if(!IDExist(user_input))
      {
        printf(NO_ID, user_input);
      }
    } while(!IDExist(user_input));
    
    do
    {
        printf(REMOVE_CONFIRM);
        fflush(stdout);

        user_input = read_from_user();

        /* If user confirm */
        if(user_input[0] == 'y' || user_input[0] == 'Y')
        {
            /* Getting full agent name */
            char *full_name = getFullnameById(u_id);
            
            
            fp = fopen(AUTH_FILE, "r+");
            if(!fp)
            {
                ErrorExit(FOPEN_ERROR, ARGV0, AUTH_FILE);
            }
            #ifndef WIN32
            chmod(AUTH_FILE, 0440);
            #endif


            fsetpos(fp, &fp_pos);
            fprintf(fp, "#*#*#*#*#*#*#*#*#*#*#");
            fclose(fp);


            /* Remove counter for id */
            delete_agentinfo(full_name); 
            RemoveCounter(u_id);


            printf(REMOVE_DONE, u_id);
            restart_necessary = 1;
            break;
        }
        else if(user_input[0] == 'n' || user_input[0] == 'N')
        {
            printf(REMOVE_NOT);
            break;
        }

    } while(1);

    return(0);
}


int list_agents()
{
    if(!print_agents())
        printf(NO_AGENT);

    printf("\n");
    printf(PRESS_ENTER);
    read_from_user();

    return(0);

}

/* EOF */