X-Git-Url: https://git.distorted.org.uk/~mdw/catacomb/blobdiff_plain/0f00dc4c8eb47e67bc0f148c2dd109f73a451e0a..8f2287ef5c05d496fcb9b012629af007fe56f897:/math/genprimes.c

diff --git a/math/genprimes.c b/math/genprimes.c
index a4bf64ae..8b47ac94 100644
--- a/math/genprimes.c
+++ b/math/genprimes.c
@@ -52,6 +52,7 @@ int main(int argc, char *argv[])
   char *header = "primetab.h";
   char *source = "primetab.c";
   char *name = "primetab";
+  char *hdrbase;
   char *sym = 0;
   intv p = DA_INIT;
   int i;
@@ -95,6 +96,9 @@ int main(int argc, char *argv[])
   if (!p_max && !p_n)
     die(EXIT_FAILURE, "bad arguments to `-n' or `-m'");
 
+  if ((hdrbase = strrchr(header, '/')) == 0) hdrbase = header;
+  else hdrbase++;
+
   if (p_n || p_max >= 2)
     DA_PUSH(&p, 2);
   for (i = 3; (!p_max && !p_n) ||
@@ -170,7 +174,7 @@ extern const smallprime %s[];\n\
 #include \"%s\"\n\
 \n\
 const %s %s[] = {",
-	    header, type, name);
+	    hdrbase, type, name);
     for (i = 0; i < DA_LEN(&p); i++) {
       if (i % 8 == 0)
 	fputs("\n  ", fp);