forksort/forksort.c

   1 /**************\r
   2 * Name:    forksort.c\r
   3 * Author:       Jan Vales (e0726236@student.tuwien.ac.at)\r
   4 * Date:         13.1.2010\r
   5 * Purpose:  Forksorting\r
   6 * Usage:  forksort\r
   7 * Input:  <lines to sort>\n<line>\n<line>\n...\r
   8 * Output:  sorted input\r
   9 **************/\r
  10 #include "globals.h"\r
  11 \r
  12 char *szCommand = "";\r
  13 \r
  14 void usage(){\r
  15   (void)fprintf(stderr, "Usage: %s\n", szCommand);\r
  16   exit(EXIT_FAILURE); \r
  17 }\r
  18 \r
  19 void bailout(const char *msgstr){\r
  20   if(errno == 0) {\r
  21     (void)fprintf(stderr, "%s(%d(%d)): [ERROR] %s\n", szCommand, getpid(), getppid(), msgstr);\r
  22   }else{\r
  23     (void)fprintf(stderr, "%s(%d(%d)): [ERROR(%d)] %s: %s\n", szCommand, getpid(), getppid(), errno, msgstr, strerror(errno));\r
  24   }\r
  25   exit(EXIT_FAILURE);\r
  26 }\r
  27 \r
  28 void myTrim(char *str){\r
  29   unsigned int c;\r
  30   if(str == 0 || strlen(str) == 0) return;\r
  31   while(isspace(str[0]))for(c=0; c<strlen(str); c++)str[c] = str[c+1];\r
  32   for(c = strlen(str)-1; isspace(str[c]); c--)str[c] = 0;\r
  33 }\r
  34 \r
  35 int readLine(char *lineRet, FILE *fp){\r
  36   char tmpLine[MAX_CHARS];\r
  37   if(fgets(tmpLine, MAX_CHARS, fp)==NULL) {\r
  38 /*    (void)fprintf(stderr, "%s (pid %i): [DBG] readLine(): error at fgets, returning EOF\n", szCommand, getpid()); */\r
  39     return EOF;\r
  40   }\r
  41   if((strlen(tmpLine) == (MAX_CHARS-1)) && (tmpLine[strlen(tmpLine)-1] != '\n'))while(fgetc(fp) != '\n');\r
  42   myTrim(tmpLine);\r
  43   strcpy(lineRet, tmpLine);\r
  44   return EXIT_SUCCESS;\r
  45 }\r
  46 \r
  47 \r
  48 int main(int argc, char *argv[]){\r
  49 fprintf(stderr, "%d \n", strcmp("aa","bb"));\r
  50 fprintf(stderr, "%d \n", strcmp("bb","bb"));\r
  51 \r
  52 fprintf(stderr, "%d \n", strcmp("cc","bb"));\r
  53 \r
  54 \r
  55 return 0;\r
  56 \r
  57   int linecnt;\r
  58   char tmpLine[MAX_CHARS];\r
  59   int i;\r
  60 \r
  61 /*\r
  62 p2c = parent 2 child\r
  63 c2p = child 2 parent\r
  64 */\r
  65 \r
  66   int lPp2c[2];\r
  67   int lPc2p[2];\r
  68   FILE *lfp2c;\r
  69   FILE *lfc2p;\r
  70   pid_t lPid;\r
  71   int lStat;\r
  72 \r
  73   int rPp2c[2];\r
  74   int rPc2p[2];\r
  75   FILE *rfp2c;\r
  76   FILE *rfc2p;\r
  77   pid_t rPid;\r
  78   char tmpLineR[MAX_CHARS];\r
  79   int rStat;\r
  80 \r
  81   szCommand = argv[0];\r
  82   if(argc != 1)usage();\r
  83 \r
  84   if(readLine(tmpLine, stdin) == EOF)bailout("Cannot read Line");\r
  85   if(sscanf(tmpLine, "%i", &linecnt) != 1)bailout("Cannot parse Line-Count");\r
  86 \r
  87   if(linecnt <= 0)bailout("Linecount must be positive!");\r
  88   if(linecnt == 1){\r
  89     readLine(tmpLine, stdin);\r
  90     (void)fprintf(stdout, "%s\n", tmpLine);\r
  91     return 0;\r
  92   }\r
  93 \r
  94   /* only if linecnt >1: do stuff below this line */\r
  95 \r
  96   if(pipe(lPc2p) == EOF || pipe(lPp2c) == EOF)bailout("LC: Cannot create pipes");\r
  97   if(pipe(rPc2p) == EOF || pipe(rPp2c) == EOF)bailout("Cannot create pipes for R child");\r
  98 \r
  99 \r
 100   /*** MERGE-SORT::SPLIT DATA ***/\r
 101   lPid = fork();\r
 102   switch(lPid) {\r
 103     case -1: bailout("error forking left child!");\r
 104     case 0:\r
 105       if((close(lPc2p[0]) < 0) || (close(lPp2c[1]) < 0))bailout("LC: Cannot close unneeded pipe ends");\r
 106       if((dup2(lPc2p[1], fileno(stdout)) == EOF) || (dup2(lPp2c[0], fileno(stdin)) == EOF))bailout("LC: dup2 failed");\r
 107       if(execlp(argv[0], argv[0], (const char*)0) < 0)bailout("LC: execlp() failed");\r
 108       break;\r
 109     default:\r
 110       close(lPc2p[1]);\r
 111       close(lPp2c[0]);\r
 112       if((lfp2c = fdopen(lPp2c[1], "w")) == NULL)bailout("LC: fdopen(lPp2c[1]) failed");\r
 113 \r
 114       /* write stuff to left child's stdin */\r
 115       if(fprintf(lfp2c, "%i\n", (linecnt/2)) < 0)bailout("LC: Cannot write linecnt.");\r
 116       for(i = 0; i < (linecnt/2); i++){\r
 117         if(readLine(tmpLine, stdin) == EOF)bailout("Cannot read stdin");\r
 118         if(fprintf(lfp2c, "%s\n", tmpLine) < 0)bailout("LC: Cannot write to lfp2c");\r
 119       }\r
 120       if(fclose(lfp2c) == EOF)bailout("LC: fclose(lfp2c) failed");\r
 121   }\r
 122 \r
 123   rPid = fork();\r
 124   switch(rPid) {\r
 125     case -1: bailout("error forking right child!");\r
 126     case 0:\r
 127       if((close(rPc2p[0]) < 0) || (close(rPp2c[1]) < 0))bailout("RC: Cannot close unneeded pipe ends");\r
 128       if((dup2(rPc2p[1], fileno(stdout)) == EOF) || (dup2(rPp2c[0], fileno(stdin)) == EOF))bailout("RC: dup2 failed");\r
 129       if(execlp(argv[0], argv[0], (const char*)0) < 0)bailout("RC: execlp() failed");\r
 130       break;\r
 131     default:\r
 132       close(rPc2p[1]);\r
 133       close(rPp2c[0]);\r
 134       if((rfp2c = fdopen(rPp2c[1], "w")) == NULL)bailout("RC: fdopen(rPp2c[1]) failed");\r
 135 \r
 136       /* write stuff to left child's stdin */\r
 137       if(fprintf(rfp2c, "%i\n", (linecnt-(linecnt/2))) < 0)bailout("RC: Cannot write linecnt.");\r
 138       for(i = (linecnt/2); i < linecnt; i++) {\r
 139         if(readLine(tmpLine, stdin) == EOF)bailout("Cannot read stdin");\r
 140         if(fprintf(rfp2c, "%s\n", tmpLine) < 0)bailout("RC: Cannot write to rfp2c");\r
 141       }\r
 142 \r
 143       if(fclose(rfp2c) == EOF)bailout("RC: fclose(rfp2c) failed");\r
 144   }\r
 145 \r
 146   /*** MERGE-SORT::MERGE DATA ***/\r
 147   if((lfc2p = fdopen(lPc2p[0], "r")) == NULL) bailout("fdopen(lPc2p[0], r)");\r
 148   if((rfc2p = fdopen(rPc2p[0], "r")) == NULL) bailout("fdopen(rPc2p[0], r)");\r
 149 \r
 150   if((lStat = readLine(tmpLine, lfc2p)) == EXIT_FAILURE)bailout("LC: Cannot read results");\r
 151   if((rStat = readLine(tmpLineR, rfc2p)) == EXIT_FAILURE)bailout("RC: Cannot read results");\r
 152 \r
 153   while(lStat != EOF || rStat != EOF) {\r
 154 fprintf(stderr, "%d hier %d - %d\n", getpid(), lStat, rStat);\r
 155 \r
 156     if(lStat == EOF){\r
 157       while(rStat != EOF) {\r
 158         fprintf(stdout, "%s\n", tmpLineR);\r
 159         if((rStat = readLine(tmpLineR, rfc2p)) == EXIT_FAILURE)bailout("RC: Cannot read results");\r
 160       }\r
 161 fprintf(stderr, "%d hier1 %d - %d\n", getpid(), lStat, rStat);\r
 162 \r
 163     }\r
 164 \r
 165     if(rStat == EOF) {\r
 166       while(lStat != EOF) {\r
 167         fprintf(stdout, "%s\n", tmpLine);\r
 168         if((lStat = readLine(tmpLine, lfc2p)) == EXIT_FAILURE)bailout("LC: Cannot read results");\r
 169       }\r
 170 fprintf(stderr, "%d hier2 %d - %d\n", getpid(), lStat, rStat);\r
 171 \r
 172     }\r
 173 \r
 174     while((lStat != EOF && rStat != EOF) && (strcmp(tmpLine, tmpLineR) > 0)) {\r
 175       fprintf(stdout, "%s\n", tmpLineR);\r
 176       if((rStat = readLine(tmpLineR, rfc2p)) == EXIT_FAILURE)bailout("RC: Cannot read results");\r
 177 fprintf(stderr, "%d hier3 %d - %d\n", getpid(), lStat, rStat);\r
 178     }\r
 179 \r
 180     while((lStat != EOF && rStat != EOF) && (strcmp(tmpLine, tmpLineR) <= 0)) {\r
 181       fprintf(stdout, "%s\n", tmpLine);\r
 182       if((lStat = readLine(tmpLine, lfc2p)) == EXIT_FAILURE)bailout("LC: Cannot read results");\r
 183 fprintf(stderr, "%d hier3 %d - %d\n", getpid(), lStat, rStat);\r
 184     }\r
 185 fprintf(stderr, "%d hier4 %d - %d\n", getpid(), lStat, rStat);\r
 186   }\r
 187 \r
 188 fprintf(stderr, "%d hier5 %d - %d\n", getpid(), lStat, rStat);\r
 189 \r
 190   if(fclose(lfc2p) == EOF) return EXIT_FAILURE;\r
 191   if(fclose(rfc2p) == EOF) return EXIT_FAILURE;\r
 192 \r
 193   waitpid(lPid, 0, 0);\r
 194   waitpid(rPid, 0, 0);\r
 195   return 0;\r
 196 }\r
 197 \r