03/win32/min_shell.c

/* $Id: min_shell.c,v 1.4 2006/06/15 14:21:16 joko Exp $ */

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <windows.h>

#include "min_shell.h"

#define BOOL int
#define TRUE 1
#define FALSE 0


#define MAX_COMMAND_LENGTH 1024

int main(int argc, char * argv[]) {
  if (argc == 1) {
    //printf("No arguments given.");
    //exit(EXIT_FAILURE);
    char filename[255] = "abc\0";
    BOOL background = 0;
    start_process(filename, background);
  }
  
  if (argc == 2) {
    FILE * script = fopen(argv[1], "r");
    read_commands(script);
    fclose(script);
  }
}

void read_commands(FILE * fp) {
  
  char command[MAX_COMMAND_LENGTH];
  int command_length;
  
  fgets(command, MAX_COMMAND_LENGTH, fp);
  command_length = strlen(command);
  
  /* if last char is newline, strip it */
  if (command[command_length-1] == '\n') {
    command[command_length-1] = '\0';
  }
  
  process_command(command);
}

void process_command(char * command) {
  printf("process_command: %s\n", command);
  start_process(command, 0);
}

BOOL start_process(char * filename, BOOL background) {

  STARTUPINFO si;
  PROCESS_INFORMATION pi;

  //printf("start_process: '%s'\n", filename);
  
  ZeroMemory(&si,sizeof(si));
  si.cb = sizeof(STARTUPINFO);
  
  ZeroMemory(&pi,sizeof(pi));
  
  CreateProcess(
    NULL, filename,
    NULL, NULL, FALSE, 0,
    NULL, NULL,
    &si,
    &pi
  );
  check_alert_error("CreateProcess");
  
  if (!background) {
    long status;
    status = WaitForSingleObject(pi.hProcess, INFINITE);
    if (status == WAIT_FAILED) {
      check_alert_error("WaitForSingleObject");
      return FALSE;
    }
  }
  
  return TRUE;

}

void check_alert_error(const char * error_source) {
  DWORD dwError = GetLastError();

  HLOCAL message = NULL;
  BOOL fOk;
  
  if (dwError == 0) 
    return;

  fOk = FormatMessage(
    FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER,
    NULL,
    dwError,
    MAKELANGID(LANG_NEUTRAL, SUBLANG_SYS_DEFAULT),
    (PTSTR)&message,
    0,
    NULL
  );

  if (message != NULL) {
    fprintf(stderr, "Error with '%s': %s (Code %i)\n", error_source, message, dwError);
    LocalFree(message);
  }
  
}
1	/* $Id: min_shell.c,v 1.4 2006/06/15 14:21:16 joko Exp $ */
2
3	#include <stdlib.h>
4	#include <stdio.h>
5	#include <errno.h>
6	#include <windows.h>
7
8	#include "min_shell.h"
9
10	#define BOOL int
11	#define TRUE 1
12	#define FALSE 0
13
14
15	#define MAX_COMMAND_LENGTH 1024
16
17	int main(int argc, char * argv[]) {
18	if (argc == 1) {
19	//printf("No arguments given.");
20	//exit(EXIT_FAILURE);
21	char filename[255] = "abc\0";
22	BOOL background = 0;
23	start_process(filename, background);
24	}
25
26	if (argc == 2) {
27	FILE * script = fopen(argv[1], "r");
28	read_commands(script);
29	fclose(script);
30	}
31	}
32
33	void read_commands(FILE * fp) {
34
35	char command[MAX_COMMAND_LENGTH];
36	int command_length;
37
38	fgets(command, MAX_COMMAND_LENGTH, fp);
39	command_length = strlen(command);
40
41	/* if last char is newline, strip it */
42	if (command[command_length-1] == '\n') {
43	command[command_length-1] = '\0';
44	}
45
46	process_command(command);
47	}
48
49	void process_command(char * command) {
50	printf("process_command: %s\n", command);
51	start_process(command, 0);
52	}
53
54	BOOL start_process(char * filename, BOOL background) {
55
56	STARTUPINFO si;
57	PROCESS_INFORMATION pi;
58
59	//printf("start_process: '%s'\n", filename);
60
61	ZeroMemory(&si,sizeof(si));
62	si.cb = sizeof(STARTUPINFO);
63
64	ZeroMemory(&pi,sizeof(pi));
65
66	CreateProcess(
67	NULL, filename,
68	NULL, NULL, FALSE, 0,
69	NULL, NULL,
70	&si,
71	&pi
72	);
73	check_alert_error("CreateProcess");
74
75	if (!background) {
76	long status;
77	status = WaitForSingleObject(pi.hProcess, INFINITE);
78	if (status == WAIT_FAILED) {
79	check_alert_error("WaitForSingleObject");
80	return FALSE;
81	}
82	}
83
84	return TRUE;
85
86	}
87
88	void check_alert_error(const char * error_source) {
89	DWORD dwError = GetLastError();
90
91	HLOCAL message = NULL;
92	BOOL fOk;
93
94	if (dwError == 0)
95	return;
96
97	fOk = FormatMessage(
98	FORMAT_MESSAGE_FROM_SYSTEM \| FORMAT_MESSAGE_ALLOCATE_BUFFER,
99	NULL,
100	dwError,
101	MAKELANGID(LANG_NEUTRAL, SUBLANG_SYS_DEFAULT),
102	(PTSTR)&message,
103	0,
104	NULL
105	);
106
107	if (message != NULL) {
108	fprintf(stderr, "Error with '%s': %s (Code %i)\n", error_source, message, dwError);
109	LocalFree(message);
110	}
111
112	}
MailToCvsAdmin">MailToCvsAdmin	ViewVC Help
Powered by ViewVC 1.1.26