void output_results()
{
FILE *output_file;
int c, rc;
if ((output_file = fopen(filename, "r")) == NULL)
{
printf("Server Error opening output file\n");
exit(-1);
}
do
{
c = getc(output_file);
if (c == EOF)
break;
Socket_putc(c, connect_socket);
}
while (c != EOF);
Socket_putc('\0', connect_socket);
//Socket_putc(EOF, connect_socket);
fclose(output_file);
remove(filename);
}
void send(char line_data[], void *p){
Socket *s = p;
Socket connect_socket = *s;
int i, rc;
for(i = 0; i < strlen(line_data); i++){
rc = Socket_putc(line_data[i], connect_socket);
check_put(rc, &connect_socket);
}
}
int main(int argc, char* argv[])
{
int i, c, rc, fc;
int count = 0;
char line_data[MAX_LINE];
/* variable to hold socket descriptor */
Socket connect_socket;
if (argc < 3)
{
printf("No host and port\n");
return (-1);
}
/* connect to the server at the specified host and port;
* blocks the process until the connection is accepted
* by the server; creates a new data transfer socket.
*/
connect_socket = Socket_new(argv[1], atoi(argv[2]));
if (connect_socket < 0)
{
printf("Failed to connect to server\n");
return (-1);
}
printf("%% "); // print initial prompt
/* get a string from stdin up to and including
* a newline or to the maximim input line size.
* Continue getting strings from stdin until EOF.
*/
while ((fgets(line_data, sizeof(line_data), stdin) != NULL))
{
count = strlen(line_data) + 1; /* count includes '\0' */
/* send the characters of the input line to the server
* using the data transfer socket.
*/
for (i = 0; i < count; i++)
{
c = line_data[i];
rc = Socket_putc(c, connect_socket);
// if (rc == EOF)
// {
// printf("Socket_putc EOF or error\n");
// Socket_close(connect_socket);
// exit (-1); /* assume socket problem is fatal, end client */
// }
}
/* receive the converted characters for the string from
* the server using the data transfer socket.
*/
printf("boutta receive\n");
while (1) {
fc = Socket_getc(connect_socket);
printf("%c", fc);
if (fc == '\0') {
break;
}
}
printf("%% "); // print prompt for next command
} /* end of while loop; at EOF */
Socket_close(connect_socket);
exit(0);
}
void shell_service(void){
int i = 0, c, character, rc, no_error;
//MAX_LINE is the longest the stdin buffer can be, so we won't overflow it
char line_data[MAX_LINE];
pid_t cpid, term_pid; /* pid_t is typedef for Linux process ID */
int chld_status;
unsigned char new_line[MAX_LINE];
unsigned char tmp_name[MAX_TMP];
unsigned char id_str[MAX_TMP];
char response[MAX_LINE];
char temp[MAX_LINE];
int id;
/* variable names for file "handles" */
FILE *tmpFP;
FILE *fp;
/* get the parent process ID and use it to create a file name for the
* temporary file with the format "tmpxxxxx" where xxxxx is the ID
*/
id = (int) getpid();
sprintf(id_str, "%d", id);
strcpy(tmp_name,"tmp");
strcat(tmp_name, id_str);
/* will not use the server socket */
Socket_close(welcome_socket);
while((c = Socket_getc(connect_socket)) != EOF){
if(c != NEWLINE){
line_data[i] = c;
i ++;
}
else{
line_data[i] = STRING_END;
i++;
//make sure the string is terminated
if(i == MAX_LINE){
line_data[MAX_LINE - 1] = STRING_END;
}
cpid = fork(); /* create child == service process */
if (cpid == ERROR){
perror("fork");
exit (ERROR);
}
if (cpid == NORMAL) {/* code for the service process */
/* dynamically redirect stdout to the named temporary
* file open for writing
*/
fp = freopen(tmp_name, "w", stdout);
parse_input(line_data, &connect_socket);
} /* end service process */
else{
i = 0;
term_pid = waitpid(cpid, &chld_status, 0);
if (term_pid == ERROR)
perror("waitpid");
else{
if (WIFEXITED(chld_status))
sprintf(response, "End of input: PID %d exited, status = %d\n", cpid, WEXITSTATUS(chld_status));
else
sprintf(response, "End of Input: PID %d did not exit normally\n", cpid);
}
if ((tmpFP = fopen (tmp_name, "r")) == NULL) {
fprintf (stderr, "error opening tmp file\n");
exit (ERROR);
}
no_error = 0;
while (!feof (tmpFP)) {
no_error = 1;
/* Get line from file */
if (fgets (new_line, sizeof(new_line), tmpFP) == NULL)
break;
send(new_line, &connect_socket);
}
rc = Socket_putc(NEWLINE, connect_socket);
check_put(rc, &connect_socket);
if(no_error)
send(response, &connect_socket);
rc = Socket_putc(STRING_END, connect_socket);
check_put(rc, &connect_socket);
/* delete the temporary file */
remove(tmp_name);
// }
}
}
}
}
// actually does the execution and reports back to ForkExecuteRedirect
void ForkExecuteRedirect_helper(int argc, char *argv[]){
pid_t cpid, term_pid; /* type for Linux process id */
int chld_status;
int i;
int execCheck;
char *parms_array[NUM_PARMS]; /* arrary of pointers to strings */
unsigned char new_line[MAX_LINE];
unsigned char tmp_name[MAX_TMP];
unsigned char id_str[MAX_TMP];
int id;
/* variable names for file "handles" */
FILE *tmpFP;
FILE *fp;
/* The positional parameters passed to this program as pointers to
* strings can just be passed on to execvp which requires a
* NULL-terminated array of pointers to strings. By convention, the
* first entry in the array is the file name to be executed, and the
* next non-null pointers are for positional parameters to that program.
*/
/* get the parent process ID and use it to create a file name for the
* temporary file with the format "tmpxxxxx" where xxxxx is the ID
*/
id = (int) getpid();
sprintf(id_str, "%d", id);
strcpy(tmp_name,"tmp");
strcat(tmp_name, id_str);
/* create child process, report error if present and exit */
cpid = fork();
if (cpid == -1)
{
perror("fork"); /* perror creates description of OS error codes */
exit(ERR_RETURN);
}
/* conditional execution of statements depending on whether the code is
* if running as the parent or child process.
*/
if (cpid == 0)
{ /* the child process */
/* dynamically redirect stdout to the named temporary
* file open for writing
*/
fp = freopen(tmp_name, "w", stdout);
/* do the OS execvp() call, if there is an error return, report it. Otherwise
* this call will not return and the new image will execute as this process ID.
*/
execCheck = execvp(argv[0], argv);
if (execCheck == -1)
{
perror("execvp, execCheck == -1");
exit(-1);
}
} /* end of code that can be executed by the child process */
else
{/* code executed by the parent process */
/* wait for the child to terminate and determine its exit status */
/* this is necessary to prevent the accumulation of "Zombie" processes */
term_pid = waitpid(cpid, &chld_status, 0);
if (term_pid == -1)
perror("waitpid");
else
{
if (WIFEXITED(chld_status))
printf("PID %d exited, status = %d\n", cpid, WEXITSTATUS(chld_status));
else
printf("PID %d did not exit normally\n", cpid);
}
if ((tmpFP = fopen (tmp_name, "r")) == NULL) {
fprintf (stderr, "error opening tmp file\n");
exit (-1);
}
while (!feof (tmpFP)) {
/* Get line from file */
if (fgets (new_line, sizeof(new_line), tmpFP) == NULL)
break;
int toClientCount = strlen(new_line);
int new_index;
char toClientChar;
int new_rc;
for (new_index = 0; new_index < toClientCount; new_index++){
toClientChar = new_line[new_index];
new_rc = Socket_putc(toClientChar, connect_socket);
if (new_rc == EOF){
printf("Socket_putc EOF or error\n");
Socket_close(connect_socket);
exit (-1); // assume socket problem is fatal, end client
}
}
}
// delete the temporary file
remove(tmp_name);
} // end of code that can be executed by the parent process
exit(0);
}
int main(int argc, char* argv[])
{
bool doneFlag = false;
int i;
int rc;
int c;
unsigned char tmp_name[MAX_TMP];
unsigned char id_str[MAX_TMP];
int id;
int childPID, term_pid;
int chld_status;
char fc;
char file_line[MAX_LINE];
char * args[MAX_ARGS];
/* variable names for file "handles" */
FILE *tmpFP;
FILE *fp;
if (argc < 2)
{
printf("No port specified\n");
return (-1);
}
/* create a "welcoming" socket at the specified port */
welcome_socket = ServerSocket_new(atoi(argv[1]));
if (welcome_socket < 0)
{
printf("Failed new server socket\n");
return (-1);
}
id = (int) getpid();
sprintf(id_str, "%d", id);
strcpy(tmp_name,"tmp");
strcat(tmp_name, id_str);
/* accept an incoming client connection; blocks the
* process until a connection attempt by a client.
* creates a new data transfer socket.
*/
connect_socket = ServerSocket_accept(welcome_socket);
if (connect_socket < 0)
{
printf("Failed accept on server socket\n");
exit (-1);
}
Socket_close(welcome_socket);
char input[MAX_LINE]={'\0'};
memset(input,'\0',sizeof(char)*MAX_LINE);
while(1){
for (i = 0; i < MAX_LINE; i++) {
c = Socket_getc(connect_socket);
if (c == EOF) {
remove(tmp_name);
doneFlag = true;
for (i=0; i<=strlen(ERROR2); i++){
Socket_putc(ERROR2[i], connect_socket);
}
// printf("Socket_getc EOF or error\n");
return; /* assume socket EOF ends service for this client */
}
else {
if (c == '\0') {
input[i] = '\0';
break;
}
if (c == '\n') {
i--;
}
else{
input[i] = c;
}
}
}
/* be sure the string is terminated if max size reached */
if (i == MAX_LINE) input[i-1] = '\0';
fp = freopen(tmp_name, "w", stdout);
childPID = fork();
if (childPID == 0){
//code for child:
char * token = strtok(input, " \t");
args[0] = token; // set first "argument" to the command itself
int count = 1;
// loop to fill args[] with the remaining arguments (if any)
while (token != NULL) {
token = strtok(NULL, " ");
args[count] = token;
count++;
if (count > (MAX_ARGS+2)){
// handle it
}
}
args[count] = NULL; // the array must be null terminated for execvp() to work
int execCode = execvp(args[0], args);
if (execCode == -1) {
for (i=0; i<=strlen(ERROR1); i++){
Socket_putc(ERROR1[i], connect_socket);
}
// exit(0);
//goto end;
// printf("Error: invalid command.\n");
}
}
else if (childPID == -1) {
printf("Problem forking.\n");
}
else {
//parent code:
term_pid = waitpid(childPID, &chld_status, 0);
if (doneFlag == false){
char file_c;
if ((tmpFP = fopen (tmp_name, "r")) == NULL) {
fprintf (stderr, "error opening tmp file\n");
exit (-1);
}
while ((fc = fgetc(tmpFP)) != EOF) {
Socket_putc(fc, connect_socket);
}
}
if (term_pid == -1)
perror("waitpid");
else
{
if (WIFEXITED(chld_status)) {
char str[] = "PID %d exited, status = %d\n";
char str2[256];
sprintf(str2, str, childPID, WEXITSTATUS(chld_status));
for (i=0; i<=strlen(str2); i++){
Socket_putc(str2[i], connect_socket);
}
//printf("PID %d exited, status = %d\n", childPID, WEXITSTATUS(chld_status));
}
else{
printf("PID %d did not exit normally\n", childPID);
}
}
}
end:;
}
Socket_close(connect_socket);
exit (0);
}
