OSDev.org

Code: Select all

#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;
}

Code: Select all

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