netsukuku/src/Ntk-Console/Netsukuku-Console.c

#include "Netsukuku-Console.h"

char response[BUFFER_LENGTH];

void usage();

void clean_up();

int validity_check(char request) {
    
        if(strncmp(request,"help", (int)strlen(request))  == 0)
            return 1;
        
        else if(strncmp(request,"uptime", (int)strlen(request))  == 0)
            return 0;
        
        else if(strncmp(request,"kill", (int)strlen(request))  == 0)
            return 2;
        
        else if(strncmp(request,"version", (int)strlen(request))  == 0)
            return 3;
        
        else if(strncmp(request,"console_uptime", (int)strlen(request))  == 0)
            return 4;
        
        else if(strlen(request) > 250)
            return 5;
        
        else if(strncmp(request,"inet_connected", (int)strlen(request))  == 0)
            return 0;
        
        else if(strncmp(request,"cur_ifs", (int)strlen(request))  == 0)
            return 0;
        
        else if(strncmp(request,"cur_ifs_n", (int)strlen(request))  == 0)
            return 0;
        
        else if(strncmp(request,"cur_qspn_id", (int)strlen(request))  == 0)
            return 0;
        
        else if(strncmp(request,"cur_ip", (int)strlen(request))  == 0)
            return 0;
        
        else if(strncmp(request,"cur_node", (int)strlen(request))  == 0)
            return 0;
        
        else if(strncmp(request,"ifs", (int)strlen(request))  == 0)
            return 0;
        
        else if(strncmp(request,"ifs_n", (int)strlen(request))  == 0)
            return 0;
        
        else {
            printf("Incorrect or unreadable command, Please correct it.\n");
            return -1;
        }
    
    return -2;
    
}

void response_cleanup(char response[BUFFER_LENGTH]) {
    
    char remove = 'a';

    char* c;
    int x;
    char* pPosition;
    while(pPosition = strchr(response, 'a') != NULL)  {
        if ((c = index(response, remove)) != NULL) {
        size_t len_left = sizeof(response) - (c+1-response);
        memmove(c, c+1, len_left);
        }
    }    
    printf("Sent and received Successfully!\n The Response was: %s", response);
}

/* Sends and receives to ntkd */
void ntkd_request(char request) {

    int request_length;
    
            rc = connect(sockfd, (struct sockaddr *)&serveraddr, SUN_LEN(&serveraddr));
                if (rc < 0) {
                    perror("connect() failed");
                    exit(-1);
                }
    
    
            request_length = (int)strlen(request);
            memset(request, 'a', BUFFER_LENGTH - request_length);
            rc = send(sockfd, request, sizeof(request), 0);
            if (rc < 0) {
                perror("send() failed");
                exit(-1);
            }

            bytesReceived = 0;
            while (bytesReceived < BUFFER_LENGTH) {
                rc = recv(sockfd, & response[bytesReceived],
                   BUFFER_LENGTH - bytesReceived, 0);
                if (rc < 0) {
                    perror("recv() failed");
                    exit(-1);
                }
            else if (rc == 0) {
                printf("The server closed the connection\n");
                exit(-1);
            }

            /* Increment the number of bytes that have been received so far  */
            bytesReceived += rc;        
            }
            response_cleanup(response);
}

void opensocket(void) {
    
    sockfd = socket(AF_UNIX, SOCK_STREAM, 0);
    if (sockfd < 0) {
         perror("socket creation failed");
         exit(-1);
    }
    
    memset(&serveraddr, 0, sizeof(serveraddr));
    serveraddr.sun_family = AF_UNIX;
    strcpy(serveraddr.sun_path, SERVER_PATH);
}

void console_uptime(void) {
    
    int uptime_sec1;
    int uptime_min1;
    int uptime_hour1;
    
    int uptime_day1;
    int uptime_month1;
    int uptime_year1;
    
    time(&rawtime);
    
    timeinfo = localtime(&rawtime);
    
    uptime_sec1 = timeinfo->tm_sec;
    uptime_min1 = timeinfo->tm_min;
    uptime_hour1 = timeinfo->tm_hour;
    
    uptime_day1 = timeinfo->tm_mday;
    uptime_month1 = timeinfo->tm_mon;
    uptime_year1 = timeinfo->tm_year;
    
    uptime_sec1 -= uptime_sec;
    uptime_min1 -= uptime_min;
    uptime_hour1 -= uptime_hour;
    
    uptime_day1 -= uptime_day;
    uptime_month1 -= uptime_month;
    uptime_year1 -= uptime_year;
    
    printf("Total Uptime is: %i Year(s), %i Month(s), %i Day(s), %i Hour(s), %i Minute(s), %i Second(s)\n",uptime_year1, uptime_month1, uptime_day1, uptime_hour1, uptime_min1, uptime_sec1);
    
}

int millisleep(unsigned ms)
{
  return usleep(1000 * ms);
}

void console(char request) {
    
    if(validity_check(request) == -2)
            printf("Error: Command has not been processed!\n");
        
    if(validity_check(request) == -1)
            usage();
        
    if(strncmp(request,"quit", (int)strlen(request)) == 0) {
            clean_up();
            exit(0);
        }
    
    if(validity_check(request) == 0) {
            ntkd_request(request);
            millisleep(200);
    }
        
    if(validity_check(request) == 1)
            usage();
        
    if(validity_check(request) == 2)
            system("ntkd -k");
        
    if(validity_check(request) == 3) {
            printf("Ntk-Console Version: %s\n", VERSION_STR);
            ntkd_request(request);
        }

    if(validity_check(request) == 4)
            console_uptime();
    
    if(validity_check(request) == 5)
        printf("Error: Command longer than 250 bytes.\n");
}

int main(void) {
    
    time(&rawtime);
    
    timeinfo = localtime(&rawtime);
    
    uptime_sec = timeinfo->tm_sec;
    uptime_min = timeinfo->tm_min;
    uptime_hour = timeinfo->tm_hour;
    uptime_day = timeinfo->tm_mday;
    uptime_month = timeinfo->tm_mon;
    uptime_year = timeinfo->tm_year;
    
    opensocket();
    
    printf("This is the Netsukuku Console, Please type 'help' for more information.\n");
    
    char request;    
    
    request = (char)malloc(BUFFER_LENGTH);
    
    do {
    
    printf("\n>");
        
    fgets(request, 16, stdin);
    
    fflush(stdin);
    
    console(request);
    } while(FALSE);
    
clean_up(); 
return 0;
}

void usage(void) {
    
    	printf("Usage\n"
		" uptime    Returns the time when ntkd finished the hooking," 
					  "to get the the actual uptime just do) "
					  "time(0)-me.uptime \n"
		" help	Shows this\n"
		" kill	Kills the running instance of netsukuku with SIGINT\n\n"
		" version   Shows the running version of the ntk-console, and ntkd.\n"
		" inet_connected    If it is 1, Ntkd is connected to the Internet\n"
		"\n"
		" cur_ifs   Lists all of the interfaces in cur_ifs\n"
		" cur_ifs_n Lists the number of interfaces present in `cur_ifs'\n"
		"\n");
        printf(" cur_qspn_id   The current qspn_id we are processing. "
                                "It is cur_qspn_id[levels] big\n"
		" cur_ip    Current IP address\n"
		"\n"
		" cur_node  Current Node\n"
		" ifs   Lists all of the interfaces in server_opt.ifs\n"
                " ifs_n Lists the number of interfaces present in server_opt.ifs\n"
                " quit Exits this program\n"
		" console_uptime    Gets the uptime of this console\n");
    
}

void clean_up(void) {
    
    const int optVal = 1;
    const socklen_t optLen = sizeof(optVal);
    
    setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (void*) &optVal, optLen);
    setsockopt(sockfd1, SOL_SOCKET, SO_REUSEADDR, (void*) &optVal, optLen);
    
    if (sockfd != -1)
        close(sockfd);

    if (sockfd1 != -1)
        close(sockfd1);
   
   unlink(SERVER_PATH);
    
}