Subversion Repositories pentevo

Rev

Blame | Last modification | View Log | Download | RSS feed

  1. /* SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only                     */
  2.  
  3. #include <stdio.h>
  4. #include <stdlib.h>
  5. #include <string.h>
  6. #include <errno.h>
  7. #include <sys/types.h>
  8. #include <sys/stat.h>
  9.  
  10. #include "strutil.h"
  11. #include "stringlists.h"
  12.  
  13. #define ObjExtension ".o"
  14.  
  15. static StringRecPtr inc_path_list = NULL,
  16.                     def_list = NULL;
  17.  
  18. static char *getobj(const char *pSrc)
  19. {
  20.   static char buffer[255];
  21.   int l = strlen(pSrc), bm5 = sizeof(buffer) - 5;
  22.   char *pSearch;
  23.  
  24.   if (l < bm5)
  25.     l = bm5;
  26.   memcpy(buffer, pSrc, l); buffer[l] = '\0';
  27.  
  28.   pSearch = strrchr(buffer, '.');
  29.   if (pSearch)
  30.     strcpy(pSearch, ObjExtension);
  31.  
  32.   return buffer;
  33. }
  34.  
  35. int MayRecurse(const char *pFileName)
  36. {
  37.   int l = strlen(pFileName);
  38.  
  39.   /* .rsc files are autogenerated and do not contain any
  40.      include statements - no need to scan them */
  41.  
  42.   if ((l > 4) && (!strcmp(pFileName + l - 4, ".rsc")))
  43.     return 0;
  44.  
  45.   return 1;
  46. }
  47.  
  48. static Boolean regard_sym(const char *p_name)
  49. {
  50.   return !strcmp(p_name, "_USE_MSH");
  51. }
  52.  
  53. static Boolean def_present(const char *p_name)
  54. {
  55.   StringRec *p_run = def_list;
  56.  
  57.   for (p_run = def_list; p_run; p_run = p_run->Next)
  58.   {
  59.     if (!strcmp(p_name, p_run->Content))
  60.       return True;
  61.   }
  62.   return False;
  63. }
  64.  
  65. static Boolean curr_if = True;
  66.  
  67. typedef struct if_stack
  68. {
  69.   struct if_stack *p_next;
  70.   Boolean condition, save_curr_if;
  71.   Boolean evaluated, else_occured;
  72. } if_stack_t;
  73. static if_stack_t *p_if_stack_tos = NULL;
  74.  
  75. static void push_if(Boolean condition, Boolean evaluated)
  76. {
  77.   if_stack_t *p_new_if = (if_stack_t*)calloc(1, sizeof(*p_new_if));
  78.  
  79.   p_new_if->save_curr_if = curr_if;
  80.   p_new_if->condition = condition;
  81.   p_new_if->evaluated = evaluated;
  82.   p_new_if->else_occured = False;
  83.   p_new_if->p_next = p_if_stack_tos;
  84.   p_if_stack_tos = p_new_if;
  85.   curr_if = evaluated ? curr_if && condition : curr_if;
  86. }
  87.  
  88. static void toggle_if(const char *p_file, int line)
  89. {
  90.   if (!p_if_stack_tos)
  91.   {
  92.     fprintf(stderr, "%s:%d: #else without #ifdef\n", p_file, line);
  93.     exit(1);
  94.   }
  95.   else if (p_if_stack_tos->else_occured)
  96.   {
  97.     fprintf(stderr, "%s:%d: #ifdef with more than one #else\n", p_file, line);
  98.     exit(1);
  99.   }
  100.   if (p_if_stack_tos->evaluated)
  101.     curr_if = p_if_stack_tos->save_curr_if && !p_if_stack_tos->condition;
  102.   p_if_stack_tos->else_occured = True;
  103. }
  104.  
  105. static void pop_if(const char *p_file, int line)
  106. {
  107.   if_stack_t *p_old_if;
  108.  
  109.   if (!p_if_stack_tos)
  110.   {
  111.     fprintf(stderr, "%s:%d: #endif without #ifdef\n", p_file, line);
  112.     exit(1);
  113.   }
  114.   curr_if = p_if_stack_tos->save_curr_if;
  115.   p_old_if = p_if_stack_tos;
  116.   p_if_stack_tos = p_old_if->p_next;
  117.   free(p_old_if);
  118. }
  119.  
  120. static void ParseFile(const char *pFileName, const char *pParentFileName, StringRecPtr *pFileList)
  121. {
  122.   FILE *pFile;
  123.   int l, file_stat, line_num = 0;
  124.   char line[512], *pCmd, *pName, *pValue;
  125.   String raw_name, eff_name;
  126.   struct stat stat_buf;
  127.  
  128.   pFile = fopen(pFileName, "r");
  129.   if (!pFile)
  130.   {
  131.     if (pParentFileName)
  132.       fprintf(stderr, "%s: ", pParentFileName);
  133.     perror(pFileName);
  134.     return;
  135.   }
  136.  
  137.   while (!feof(pFile))
  138.   {
  139.     if (!fgets(line, sizeof(line), pFile))
  140.       break;
  141.     line_num++;
  142.     l = strlen(line);
  143.     if ((l > 0) && (line[l - 1] == '\n'))
  144.       line[l - 1] = '\0';
  145.  
  146.     if (*line != '#')
  147.       continue;
  148.     pCmd = strtok(line + 1, " \t");
  149.  
  150.     if (!pCmd);
  151.     else if (!strcmp(pCmd, "include") && curr_if)
  152.     {
  153.       pName = strtok(NULL, " \t");
  154.       if (!pName)
  155.         continue;
  156.       l = strlen(pName);
  157.       if ((*pName != '"') || (pName[l - 1] != '"'))
  158.         continue;
  159.  
  160.       if (l - 1 < (int)sizeof(raw_name))
  161.       {
  162.         memcpy(raw_name, pName + 1, l - 2);
  163.         raw_name[l - 2] = '\0';
  164.         strmaxcpy(eff_name, raw_name, sizeof(eff_name));
  165.         file_stat = stat(eff_name, &stat_buf);
  166.         if (file_stat)
  167.         {
  168.           StringRecPtr p_run;
  169.           const char *p_path;
  170.  
  171.           for (p_path = GetStringListFirst(inc_path_list, &p_run);
  172.                p_path; p_path = GetStringListNext(&p_run))
  173.           {
  174.             size_t l = strlen(p_path);
  175.             Boolean term_path = (l > 0) && ((p_path[l -1] == '/') || (p_path[l -1] == '\\'));
  176.             as_snprintf(eff_name, sizeof(eff_name), "%s%s%s", p_path, term_path ? "" : "/", raw_name);
  177.             file_stat = stat(eff_name, &stat_buf);
  178.             if (!file_stat)
  179.               break;
  180.           }
  181.         }
  182.         /* If file is not present, it is created dynamically and expected to appear in the
  183.            object subdirectory, which is passed in as -I path.  Keep last generated eff_name
  184.            for the dependency list: */
  185.         if (!StringListPresent(*pFileList, eff_name))
  186.         {
  187.           AddStringListLast(pFileList, eff_name);
  188.           if (MayRecurse(eff_name))
  189.             ParseFile(eff_name, pFileName, pFileList);
  190.         }
  191.       }
  192.     }
  193.     else if (!strcmp(pCmd, "if"))
  194.     {
  195.       push_if(True, False);
  196.     }
  197.     else if (!strcmp(pCmd, "ifdef"))
  198.     {
  199.       pName = strtok(NULL, " \t");
  200.       if (!curr_if || !regard_sym(pName))
  201.         push_if(True, False);
  202.       else
  203.         push_if(def_present(pName), True);
  204.     }
  205.     else if (!strcmp(pCmd, "ifndef"))
  206.     {
  207.       pName = strtok(NULL, " \t");
  208.       if (!curr_if || !regard_sym(pName))
  209.         push_if(True, False);
  210.       else
  211.         push_if(!def_present(pName), True);
  212.     }
  213.     else if (!strcmp(pCmd, "else"))
  214.     {
  215.       if (curr_if)
  216.         toggle_if(pFileName, line_num);
  217.     }
  218.     else if (!strcmp(pCmd, "endif"))
  219.     {
  220.       pop_if(pFileName, line_num);
  221.     }
  222.     else if (!strcmp(pCmd, "elif"))
  223.     {
  224.       /* TODO */
  225.     }
  226.     else if (!strcmp(pCmd, "define"))
  227.     {
  228.       pName = strtok(NULL, " \t");
  229.       pValue = strtok(NULL, " \t");
  230.  
  231.       if (strchr(pName, '(') || (pValue && *pValue))
  232.         continue;
  233.     }
  234.     else if (!strcmp(pCmd, "undef"))
  235.     {
  236.       pName = strtok(NULL, " \t");
  237.     }
  238.   }
  239.   fclose(pFile);
  240. }
  241.  
  242. int main(int argc, char **argv)
  243. {
  244.   int z, ArgsAreObj = 0;
  245.   FILE *pDepFile;
  246.   const char *pDepFileName = NULL,
  247.              *pOutFileName = NULL,
  248.              *pObjExtension = NULL;
  249.   char used[1024];
  250.   StringRecPtr FileList, SrcList;
  251.  
  252.   if (argc < 2)
  253.   {
  254.     fprintf(stderr, "usage: %s [args] <file(s)>\n", *argv);
  255.     exit(1);
  256.   }
  257.  
  258.   memset(used, 0, sizeof(used));
  259.  
  260.   for (z = 1; z < argc; z++)
  261.     if ((!used[z]) && (*argv[z] == '-'))
  262.     {
  263.       used[z] = 1;
  264.       if (!strcmp(argv[z] + 1, "d"))
  265.       {
  266.         if (z >= argc - 1)
  267.           pDepFileName = NULL;
  268.         else
  269.         {
  270.           pDepFileName = argv[z + 1];
  271.           used[z + 1] = 1;
  272.         }
  273.       }
  274.       else if (!strcmp(argv[z] + 1, "o"))
  275.       {
  276.         if (z >= argc - 1)
  277.           pOutFileName = NULL;
  278.         else
  279.         {
  280.           pOutFileName = argv[z + 1];
  281.           used[z + 1] = 1;
  282.         }
  283.       }
  284.       else if (!strcmp(argv[z] + 1, "c"))
  285.       {
  286.         if (z >= argc - 1)
  287.           pObjExtension = NULL;
  288.         else
  289.         {
  290.           pObjExtension = argv[z + 1];
  291.           used[z + 1] = 1;
  292.         }
  293.       }
  294.       else if (!strcmp(argv[z] + 1, "r"))
  295.       {
  296.         ArgsAreObj = 1;
  297.       }
  298.       else if (argv[z][1] == 'I')
  299.       {
  300.         if (!argv[z][2])
  301.         {
  302.           if (z < argc - 1)
  303.           {
  304.             if (argv[z + 1][0] == '-')
  305.               fprintf(stderr, "%s: taking '%s' as include path, is this correct?\n", argv[0], argv[z + 1]);
  306.             AddStringListLast(&inc_path_list, argv[z + 1]);
  307.             used[z + 1] = 1;
  308.           }
  309.         }
  310.         else
  311.           AddStringListLast(&inc_path_list, argv[z] + 2);
  312.       }
  313.       else if (argv[z][1] == 'D')
  314.       {
  315.         if (!argv[z][2])
  316.         {
  317.           if (z < argc - 1)
  318.           {
  319.             if (argv[z + 1][0] == '-')
  320.               fprintf(stderr, "%s: taking '%s' as macro, is this correct?\n", argv[0], argv[z + 1]);
  321.             AddStringListLast(&def_list, argv[z + 1]);
  322.             used[z + 1] = 1;
  323.           }
  324.         }
  325.         else
  326.           AddStringListLast(&def_list, argv[z] + 2);
  327.       }
  328.       else
  329.       {
  330.         fprintf(stderr, "unknown option: %s\n", argv[z]);
  331.         exit(2);
  332.       }
  333.     }
  334.  
  335.   if (pDepFileName)
  336.   {
  337.     pDepFile = fopen(pDepFileName, "w");
  338.     if (!pDepFile)
  339.     {
  340.       perror(pDepFileName);
  341.       exit(errno);
  342.     }
  343.   }
  344.   else
  345.     pDepFile = stdout;
  346.  
  347.   fprintf(pDepFile, "# auto-generated by %s - do not edit\n\n", *argv);
  348.  
  349.   InitStringList(&FileList);
  350.   InitStringList(&SrcList);
  351.   for (z = 1; z < argc; z++)
  352.   {
  353.     char SrcFileName[512];
  354.     const char *pSrcFileName;
  355.  
  356.     if (used[z])
  357.       continue;
  358.  
  359.     /* object files may be in paths different than root - strip path to deduce associated src file */
  360.  
  361.     if (ArgsAreObj)
  362.     {
  363.       char *pRun;
  364.  
  365.       for (pRun = argv[z] + strlen(argv[z]); pRun > argv[z]; pRun--)
  366.         if ((*(pRun - 1) == '/') || (*(pRun - 1) == '\\'))
  367.           break;
  368.       /* leading h_ in object file name is used for host-side files */
  369.       if (!strncmp(pRun, "h_", 2))
  370.         pRun += 2;
  371.       *SrcFileName = '\0'; strncat(SrcFileName, pRun, sizeof(SrcFileName) - 1);
  372.       if (pObjExtension)
  373.       {
  374.         int l1 = strlen(SrcFileName), l2 = strlen(pObjExtension);
  375.  
  376.         if ((l1 > l2) && !memcmp(SrcFileName + l1 - l2, pObjExtension, l2))
  377.           strcpy(SrcFileName + l1 - l2, ".c");
  378.       }
  379.       pSrcFileName = SrcFileName;
  380.     }
  381.     else
  382.       pSrcFileName = argv[z];
  383.     ClearStringList(&FileList);
  384.     AddStringListLast(&SrcList, pSrcFileName);
  385.     ParseFile(pSrcFileName, NULL, &FileList);
  386.  
  387.     if (!StringListEmpty(FileList))
  388.     {
  389.       fprintf(pDepFile, "%s: %s",
  390.               pOutFileName ? pOutFileName : (ArgsAreObj ? argv[z] : getobj(argv[z])),
  391.               pSrcFileName);
  392.       while (True)
  393.       {
  394.         char *pIncFileName = MoveAndCutStringListFirst(&FileList);
  395.         if (pIncFileName)
  396.         {
  397.           if (*pIncFileName)
  398.             fprintf(pDepFile, " %s", pIncFileName);
  399.           free(pIncFileName); pIncFileName = NULL;
  400.         }
  401.         else
  402.           break;
  403.       }
  404.       fprintf(pDepFile, "\n\n");
  405.     }
  406.   }
  407.  
  408.   if (pDepFileName)
  409.   {
  410.     if (!StringListEmpty(SrcList))
  411.     {
  412.       fprintf(pDepFile, "%s:", pDepFileName);
  413.       while (True)
  414.       {
  415.         char *pIncFileName = MoveAndCutStringListFirst(&SrcList);
  416.         if (pIncFileName)
  417.         {
  418.           if (*pIncFileName)
  419.             fprintf(pDepFile, " %s", pIncFileName);
  420.           free(pIncFileName); pIncFileName = NULL;
  421.         }
  422.         else
  423.           break;
  424.       }
  425.       fprintf(pDepFile, "\n");
  426.     }
  427.     fclose(pDepFile);
  428.   }
  429.  
  430.   ClearStringList(&inc_path_list);
  431.  
  432.   return 0;
  433. }
  434.