Main logic completed
This commit is contained in:
parent
d33281b1ac
commit
76dcf53c8d
113
websh.c
113
websh.c
@ -17,6 +17,7 @@
|
||||
#include <sys/wait.h>
|
||||
|
||||
#define MAX_LEN 256
|
||||
#define MAX_LINES 1024
|
||||
|
||||
static const char *pname;
|
||||
|
||||
@ -26,6 +27,7 @@ static int sFlag = 0;
|
||||
|
||||
static const char *word;
|
||||
static const char *tag;
|
||||
static char *lines[MAX_LINES] = {NULL};
|
||||
|
||||
volatile sig_atomic_t quit = 0;
|
||||
|
||||
@ -99,16 +101,42 @@ void parse(int argc, char *argv[])
|
||||
pname = argv[0];
|
||||
}
|
||||
|
||||
void child2(int fd[], char readbuffer[])
|
||||
void child2(int fd[], char *line)
|
||||
{
|
||||
/* CHILD TWO */
|
||||
char readbuffer[MAX_LEN];
|
||||
close(fd[1]);
|
||||
int nbytes = read(fd[0], readbuffer, MAX_LEN);
|
||||
if (nbytes == -1) {
|
||||
perror(pname);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
printf("%s", readbuffer);
|
||||
readbuffer[nbytes-1] = '\0';
|
||||
if (eFlag && (strcmp(line, lines[0]) == 0)) {
|
||||
printf("<html><head></head><body>\n");
|
||||
}
|
||||
if (hFlag) {
|
||||
char *pch = strstr(line, "\n");
|
||||
if (pch != NULL)
|
||||
strncpy(pch, "\0", 1);
|
||||
printf("<h1>%s</h1>\n", line);
|
||||
}
|
||||
if (sFlag) {
|
||||
if (strstr(readbuffer, word) != NULL) {
|
||||
printf("<%s>%s</%s><br />\n", tag, readbuffer, tag);
|
||||
} else {
|
||||
printf("%s<br />\n", readbuffer);
|
||||
}
|
||||
} else {
|
||||
printf("%s<br />\n", readbuffer);
|
||||
}
|
||||
int i = 0;
|
||||
while (lines[i] != NULL) {
|
||||
i++;
|
||||
}
|
||||
if (eFlag && (strcmp(line, lines[i-1]) == 0)) {
|
||||
printf("</body></html>\n");
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
@ -138,55 +166,62 @@ int main(int argc, char *argv[])
|
||||
char *line = NULL;
|
||||
size_t len = 0;
|
||||
ssize_t nread;
|
||||
char readbuffer[MAX_LEN];
|
||||
int status;
|
||||
int w;
|
||||
|
||||
if (eFlag)
|
||||
printf("<html><head></head><body>\n");
|
||||
|
||||
int fd[2];
|
||||
pid_t childpid1;
|
||||
pid_t childpid2;
|
||||
pipe(fd);
|
||||
|
||||
while ((nread = getline(&line, &len, stdin)) > 0) {
|
||||
childpid1 = fork();
|
||||
if (childpid1 > 0) {
|
||||
/* PARENT ONE */
|
||||
childpid2 = fork();
|
||||
if (childpid2 > 0) {
|
||||
/* PARENT TWO */
|
||||
} else if (childpid2 == 0) {
|
||||
/* CHILD TWO */
|
||||
child2(fd, readbuffer);
|
||||
} else {
|
||||
perror(pname);
|
||||
free(line);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
do {
|
||||
w = waitpid(childpid2, &status, WUNTRACED | WCONTINUED);
|
||||
if (w == -1) {
|
||||
perror(pname);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
int count = 0;
|
||||
|
||||
if (WIFSIGNALED(status)) {
|
||||
printf("Killed by signal %d\n", WTERMSIG(status));
|
||||
} else if (WIFSTOPPED(status)) {
|
||||
printf("Stopped by signal %d\n", WSTOPSIG(status));
|
||||
} else if (WIFCONTINUED(status)) {
|
||||
printf("Continued\n");
|
||||
}
|
||||
} while (!WIFEXITED(status) && !WIFSIGNALED(status));
|
||||
} else if (childpid1 == 0) {
|
||||
/* CHILD ONE */
|
||||
child1(fd, line);
|
||||
} else {
|
||||
while ((nread = getline(&line, &len, stdin)) > 0) {
|
||||
lines[count] = (char *) malloc (len);
|
||||
strcpy(lines[count], line);
|
||||
count++;
|
||||
if (count == 1024)
|
||||
break;
|
||||
}
|
||||
|
||||
for (int i = 0; i < count; i++) {
|
||||
switch (childpid1 = fork()) {
|
||||
case -1:
|
||||
perror(pname);
|
||||
free(line);
|
||||
exit(EXIT_FAILURE);
|
||||
case 0:
|
||||
child1(fd, lines[i]);
|
||||
exit(EXIT_SUCCESS);
|
||||
break;
|
||||
default:
|
||||
switch (childpid2 = fork()) {
|
||||
case -1:
|
||||
perror(pname);
|
||||
free(line);
|
||||
exit(EXIT_FAILURE);
|
||||
case 0:
|
||||
child2(fd, lines[i]);
|
||||
exit(EXIT_SUCCESS);
|
||||
break;
|
||||
default:
|
||||
do {
|
||||
w = waitpid(-1, &status, WUNTRACED | WCONTINUED);
|
||||
if (w == -1) {
|
||||
perror(pname);
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
if (WIFSIGNALED(status)) {
|
||||
printf("Killed by signal %d\n", WTERMSIG(status));
|
||||
} else if (WIFSTOPPED(status)) {
|
||||
printf("Stopped by signal %d\n", WSTOPSIG(status));
|
||||
} else if (WIFCONTINUED(status)) {
|
||||
printf("Continued\n");
|
||||
}
|
||||
} while (!WIFEXITED(status) && !WIFSIGNALED(status));
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user