OSDev.org

#include <fcntl.h>
#include <unistd.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <sys/syscall.h>
#include <sys/sockio.h>
#include <sys/endian.h>
#include <dirent.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <net/if.h>



#define perror(...)   \
   printf("%s", ##__VA_ARGS__); while (1);

#define PRINT_VER()   \
   do {   printf("Welcome to openfunix!\n");   \
      printf("Copyright by Shaun, 2012-2014.\nMail:openfunix at gmail dot com\n"); \
      printf("http://www.openfunix.com\n");   \
      fflush(NULL);   \
   } while (0)   

#define PRINT_PROMT()   \
   do {   printf("\nOPENFUNIX# ");   \
      fflush(NULL);   \
   } while (0)


void help();
void ifconfig();
void ping();
void ls();
void loglevel(char *buf);
void dash();
void makedir(char *buf);
void exec(char *buf);
struct cmd {
   char *c_name;
   void (*c_func)();
} cmds[] = {
      {"help",  help},
      {"ifconfig", ifconfig},
      {"ping", ping},
      {"ls", ls},
      {"loglevel", loglevel},
      {"dash", dash},
      {"mkdir", makedir},
      {"exec", exec},
      {0, 0},
};

struct   sockaddr_in netmask;

void loglevel(char *buf)
{
   int level = atoi((buf));
   printf("set log level to %d\n", level);
   setloglevel(level);
}

void help()
{
   struct cmd *p;
   PRINT_VER();
   printf("command supported:\n");
   for (p=cmds; p->c_name; p++)
      printf("%s ", p->c_name);
}

void dash()
{
   extern char **environ;
   char *const *envp = environ;
   char *argv[] = {NULL,0};
   execve("/root/dash",(char *const *)argv, envp);
   wait(NULL);
}
void makedir(char *buf)
{
   int ret;
   ret = mkdir(buf, S_IRWXU | S_IFDIR);
   if (ret < 0)
      printf("mkdir error:%s\n", strerror(errno));

}

void exec(char *buf)
{

   extern char **environ;
   char *const *envp = environ;
   if (strcmp(buf, "") == 0)
      return;
   char *argv = strchr(++buf, ' ');
   if (argv)
      *argv++ = 0;
   char *argvs[] = {argv, 0};// this should be in a while
   char *name = buf;
   if (execve(name, (char *const *)argvs, envp) < 0)
      return;
   wait(NULL);
}

void ping(char *buf)
{

}
void ls(char *buf)
{
   const char *path =(const char *)(buf);
   DIR *dirp;
   struct dirent *dp;
   struct stat st = {0};
   int ret;

   //should list current directory.
   if (strcmp(buf, "") == 0)
      return;

   ret = stat(buf, &st);
   if (ret < 0) {
      printf("stat error:%s\n", strerror(errno));
      return;
   }

   switch(st.st_mode & S_IFMT) {
   case S_IFDIR:
      goto ls_dir;
   case S_IFBLK:
      printf("b mode:%lo size:%d  %s\n", (unsigned long)st.st_mode, st.st_size, buf);
      break;
   case S_IFREG:
      printf("mode:%lo size:%d  %s\n", (unsigned long)st.st_mode, st.st_size, buf);
      break;
   case S_IFIFO:
      printf("f mode: %lo size:%d  %s\n", (unsigned long)st.st_mode, st.st_size, buf);
      break;
   case S_IFSOCK:
      printf("s mode:%lo size:%d  %s\n", (unsigned long)st.st_mode, st.st_size, buf);
      break;
   default:
      printf("mode:%lo size:%d  %s\n", (unsigned long)st.st_mode, st.st_size, buf);
      break;
   }

   return;
ls_dir:
   dirp = opendir(path);
   if (dirp == NULL){
      printf("open dir %s faild.errno:%d\n", path, errno);
      return;
   }
   dp = readdir(dirp);
   while (dp != NULL)
   {
      dp->d_name[dp->d_namlen] = 0;
      printf("%s ", dp->d_name);
      dp = readdir(dirp);
   }
   printf("\n");
   fflush(NULL);
   closedir(dirp);
   
}
void in_status()
{
   struct sockaddr_in *sin;
   int s, flags;
   struct   ifreq      ifr;

   s = socket(AF_INET, SOCK_DGRAM, 0);
   if (s < 0) {
      perror("ifconfig: socket");
      exit(1);
   }

   memset(&ifr, 0, sizeof(ifr));

   memcpy(ifr.ifr_name, "eth0", 5);
   printf("name:%s", ifr.ifr_name);

   if (ioctl(s, SIOCGIFFLAGS, (caddr_t)&ifr) < 0) {
      perror("ioctl (SIOCGIFFLAGS)");
      exit(1);
   }

   flags = ifr.ifr_flags;


   fflush(NULL);
   if (ioctl(s, SIOCGIFADDR, (caddr_t)&ifr) < 0) {
      if (errno == EADDRNOTAVAIL || errno == EAFNOSUPPORT) {
         bzero((char *)&ifr.ifr_addr, sizeof(ifr.ifr_addr));
      } else
         perror("ioctl (SIOCGIFADDR)");
   }
   sin = (struct sockaddr_in *)&ifr.ifr_addr;
   printf("\tinet %s ", inet_ntoa(sin->sin_addr));
   //strncpy(ifr.ifr_name, name, sizeof (ifr.ifr_name));
   if (ioctl(s, SIOCGIFNETMASK, (caddr_t)&ifr) < 0) {
      if (errno != EADDRNOTAVAIL)
         perror("ioctl (SIOCGIFNETMASK)");
      bzero((char *)&ifr.ifr_addr, sizeof(ifr.ifr_addr));
   } else
      netmask.sin_addr =
          ((struct sockaddr_in *)&ifr.ifr_addr)->sin_addr;

   printf("netmask 0x%x ", ntohl(netmask.sin_addr.s_addr));
   if (flags & IFF_BROADCAST) {
      if (ioctl(s, SIOCGIFBRDADDR, (caddr_t)&ifr) < 0) {
         if (errno == EADDRNOTAVAIL)
             bzero((char *)&ifr.ifr_addr, sizeof(ifr.ifr_addr));
         else
             perror("ioctl (SIOCGIFADDR)");
      }
      //strncpy(ifr.ifr_name, name, sizeof (ifr.ifr_name));
      sin = (struct sockaddr_in *)&ifr.ifr_addr;
      if (sin->sin_addr.s_addr != 0)
         printf("broadcast %s", inet_ntoa(sin->sin_addr));
   }
   putchar('\n');
   fflush(NULL);
   close(s);
}

void if_setaddr(in_addr_t addr)
{
   struct sockaddr_in *sin;
   int fd, ret;
   struct   ifreq   ifr;
   
   fd = socket(AF_INET, SOCK_DGRAM, 0);
   if (fd < 0) {
      printf("socket error:%d\n", errno);
      return;
   }
   memset(&ifr, 0, sizeof(ifr));
   strcpy(ifr.ifr_name, "eth0");
   sin = (struct sockaddr_in *)&ifr.ifr_addr;
   sin->sin_family = AF_INET;
   sin->sin_addr.s_addr = addr;
   //sin->sin_len = sizeof(struct sockaddr_in);
   
   ret = ioctl(fd, SIOCSIFADDR, (caddr_t)&ifr);
   if (ret < 0) {
      printf("ioctl error:%d\n", errno);
   }
   close(fd);
   return;
}


void ifconfig(char *buf)
{
   char *p = (char *)(buf);
   in_addr_t addr;
   if (*p != '\0') {
      addr = inet_addr(p);
      if (addr == INADDR_NONE) {
         printf("invalid addrress\n");
         return;
      }

      if_setaddr(addr);
      return ;
   }


   in_status();
}


int main(int argc, char **argv, char **env)
{
   int ret;
   char buf[32] = {0};
   struct cmd *p;
   int fd = open("/dev/tty", O_RDWR, 0);
   if (fd < 0){
      //printf("open tty error:%s\n",strerror(errno));
      return -1;
   }
   dup2(fd, 1);
   dup2(fd, 2);
   //fcntl(fd, F_SETFD, 1);
   do {
      printf("Password:");
      fflush(NULL);
      ret = read(fd, buf, 31);
      if (ret > 0){
         if (strncmp(buf, "op3nfunix", 9) != 0){
            printf("password error\n");
            fflush(NULL);
            continue;
         }
         //printf("Login OK!\n");
         break;
      }
   }while (1);

   PRINT_VER();

   PRINT_PROMT();
   memset(buf, 0, sizeof(buf));
   while ((ret = read(fd, buf, 31)) > 0) {
      buf[ret-1] = 0;
      for (p=cmds; p->c_name; p++) {
         if (strncmp(p->c_name, buf, strlen(p->c_name)) == 0)
            break;
      }

      if (p->c_func)
         (*p->c_func)(buf + strlen(p->c_name) + 1);

       next:
       memset(buf, 0, sizeof(buf));
       printf("OPENFUNIX# ");   
       fflush(NULL);
   }
   
   return 0;
}

pid_t pid1=fork();
pid_t pid2=fork();
----do some work-------
printf("pid1:%d, pid2:%d", pid1, pid2);