davidshell.c

#include <pwd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "commands.h"

void  shellprint();
char* readline();
char** splitcommand(char* command);


int main(int argc, char** argv){
  int exit = 0;
  while(!exit){
    shellprint();
    char* command = readline();
    if(strlen(command)){
      char** split = splitcommand(command);
      exit = execute(split);
    }
  }
  //TODO: cleanup
  return 0;
}

void shellprint(){
  struct passwd* pw = getpwuid(geteuid());
  char* wd = getcwd(NULL,0);
  char host[255];
  gethostname(host,255);
  
  if(!strncmp(wd,pw->pw_dir,strlen(pw->pw_dir))){
    wd += strlen(pw->pw_dir)-1;
    *wd = '~';
  }
  printf("%s:%s:%s$ ",host,pw->pw_name,wd);
}

char* readline(){
  size_t length = 80;
  char* line = malloc(length);
  char c;
  int len = 0; 
  if(!line){
    printf("malloc fail");
    exit(1);
  }
 
  while(1){
    c = getchar();
    if(c=='\n'||c==EOF){
      line[len] = '\0';
      return line;
    }
    line[len] = c;
    ++len;
    if(len%length==0){
      line = realloc(line,len+length);
      if(line==NULL){
	printf("realloc fail");
	exit(1);
      }
    }
  }
}

char** splitcommand(char* command){
  int cnt = 1;
  char* ptr;
  char** split;
  int i = strlen(command);
  for(i=0;i<strlen(command);++i){
    if(command[i]==' '){
      ++cnt;
      }
  }
  split = malloc(cnt*sizeof(char*)+1);
  if(split==NULL){
    printf("malloc rip");
    exit(1);
  }
  ptr = strtok(command," ");
  cnt = 0;
  while(ptr!=NULL){
    split[cnt]=ptr;
    ptr = strtok(NULL," ");
    ++cnt;
  }
  split[cnt] = NULL;
  return  split;
}