home *** CD-ROM | disk | FTP | other *** search
/ Frozen Fish 1: Amiga / FrozenFish-Apr94.iso / bbs / gnu / binutils-1.8.x.tar.gz / binutils-1.8.x.tar / binutils / ar.c < prev    next >
C/C++ Source or Header  |  1990-06-13  |  47KB  |  1,991 lines
  1. /* ar.c - Archive modify and extract.
  2.    Copyright (C) 1988 Free Software Foundation, Inc.
  3.  
  4.    This program is free software; you can redistribute it and/or modify
  5.    it under the terms of the GNU General Public License as published by
  6.    the Free Software Foundation; either version 1, or (at your option)
  7.    any later version.
  8.  
  9.    This program is distributed in the hope that it will be useful,
  10.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.    GNU General Public License for more details.
  13.  
  14.    You should have received a copy of the GNU General Public License
  15.    along with this program; if not, write to the Free Software
  16.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17.  
  18. #include <stdio.h>
  19. #include <ar.h>
  20. #include <errno.h>
  21. #include <sys/types.h>
  22. #include <sys/stat.h>
  23.  
  24. #if !defined(A_OUT) && !defined(MACH_O)
  25. #define A_OUT
  26. #endif
  27.  
  28. #ifdef A_OUT
  29. #ifdef COFF_ENCAPSULATE
  30. #include "a.out.encap.h"
  31. #else
  32. #include <a.out.h>
  33. #endif
  34. #endif
  35.  
  36. #ifdef MACH_O
  37. #ifndef A_OUT
  38. #include <nlist.h>
  39. #endif
  40. #include <sys/loader.h>
  41. #endif
  42.  
  43. #ifdef USG
  44. #include <time.h>
  45. #include <fcntl.h>
  46. #else
  47. #include <sys/file.h>
  48. #include <sys/time.h>
  49. #endif
  50.  
  51. #ifdef    __GNUC__
  52. #define    alloca    __builtin_alloca
  53. #else
  54. # ifdef sparc
  55. #  include <alloca.h>
  56. # else
  57. char *alloca ();
  58. # endif
  59. #endif
  60.  
  61. #ifdef    USG
  62. #define    bcopy(source, dest, size)    memcpy((dest), (source), (size))
  63. #define    bcmp(a, b, size)        memcmp((a), (b), (size))
  64. #define    bzero(s, size)            memset((s), 0, (size))
  65. #endif
  66.  
  67. /* Locking is normally disabled because fcntl hangs on the Sun
  68.    and it isn't supported properly across NFS anyway.  */
  69. #ifdef LOCKS
  70. /* You might need to compile with -I/usr/include/sys if your fcntl.h
  71.    isn't in /usr/include (which is where it should be according to POSIX).  */
  72. #include <fcntl.h>
  73. #endif
  74.  
  75. /* This structure is used internally to represent the info
  76.    on a member of an archive.  This is to make it easier to change format.  */
  77.  
  78. struct member_desc
  79.   {
  80.     /* Name of member.  */
  81.     char *name;
  82.  
  83.     /* The following fields are stored in the member header as decimal or octal
  84.        numerals, but in this structure they are stored as machine numbers.  */
  85.     int mode;        /* Protection mode from member header.  */
  86.     long int date;    /* Last modify date as stored in member header.  */
  87.     unsigned int size;    /* Bytes of member's data, from member header.  */
  88.     int uid, gid;    /* UID and GID fields copied from member header.  */
  89.     unsigned int offset;/* Offset in archive of the header of this member.  */
  90.     unsigned int data_offset;/* Offset of first data byte of the member.  */
  91.  
  92.     /* The next field does not describe where the member was in the
  93.        old archive, but rather where it will be in the modified archive.
  94.        It is set up by write_archive.  */
  95.     unsigned int new_offset;    /* Offset of this member in new archive */
  96.  
  97.     /* Symdef data for member.  Used only for files being inserted.  */
  98.     struct symdef *symdefs;
  99.     unsigned int nsymdefs;    /* Number of entries of symdef data.  */
  100.     unsigned int string_size;    /* Size of strings needed by symdef data.  */
  101.   };
  102.  
  103. /* Each symbol is recorded by something like this.  */
  104.  
  105. struct symdef
  106.   {
  107.     union
  108.       {
  109.     unsigned long int stringoffset;
  110.     char *name;
  111.       } s;
  112.     unsigned long int offset;
  113.   };
  114.  
  115. /* Nonzero means it's the name of an existing member;
  116.    position new or moved files with respect to this one.  */
  117.  
  118. char *posname;
  119.  
  120.  
  121. /* How to use `posname':
  122.    POS_BEFORE means position before that member.
  123.    POS_AFTER means position after that member.
  124.    POS_DEFAULT if position by default; then `posname' should also be zero. */
  125.  
  126. enum { POS_DEFAULT, POS_BEFORE, POS_AFTER } postype;
  127.  
  128. /* Nonzero means describe each action performed.  */
  129.  
  130. int verbose;
  131.  
  132. /* Nonzero means don't warn about creating the archive file if necessary.  */
  133.  
  134. int silent_create;
  135.  
  136. /* Nonzero means don't replace existing members whose
  137.    dates are more recent than the corresponding files.  */
  138.  
  139. int newer_only;
  140.  
  141. /* Nonzero means preserve dates of members when extracting them.  */
  142.  
  143. int preserve_dates;
  144.  
  145. /* Operation to be performed.  */
  146.  
  147. #define DELETE 1
  148. #define REPLACE 2
  149. #define PRINT_TABLE 3
  150. #define PRINT_FILES 4
  151. #define EXTRACT 5
  152. #define MOVE 6
  153. #define QUICK_APPEND 7
  154.  
  155. int operation;
  156.  
  157. /* Name of archive file.  */
  158.  
  159. char *archive;
  160.  
  161. /* Descriptor on which we have locked the original archive file,
  162.    or -1 if this has not been done.  */
  163.  
  164. int lock_indesc;
  165.  
  166. /* Pointer to tail of `argv', at first subfile name argument,
  167.  or zero if no such were specified.  */
  168.  
  169. char **files;
  170.  
  171. /* Nonzero means write a __.SYMDEF member into the modified archive.  */
  172.  
  173. int symdef_flag;
  174.  
  175. /* Nonzero means __.SYMDEF member exists in old archive.  */
  176.  
  177. int symdef_exists;
  178.  
  179. /* Nonzero means don't update __.SYMDEF unless the flag was given.  */
  180.  
  181. int ignore_symdef;
  182.  
  183. /* Total number of symdef entries we will have. */
  184.  
  185. unsigned long int nsymdefs;
  186.  
  187. /* Symdef data from old archive (set up only if we need it) */
  188.  
  189. struct symdef *old_symdefs;
  190.  
  191. /* Number of symdefs in remaining in old_symdefs.  */
  192.  
  193. unsigned int num_old_symdefs;
  194.  
  195. /* Number of symdefs old_symdefs had when it was read in.  */
  196.  
  197. unsigned long int original_num_symdefs;
  198.  
  199. /* String table from old __.SYMDEF member.  */
  200.  
  201. char *old_strings;
  202.  
  203. /* Size of old_strings */
  204.  
  205. unsigned long int old_strings_size;
  206.  
  207. /* String table to be written into __.SYMDEF member.  */
  208.  
  209. char *new_strings;
  210.  
  211. /* Size of new_strings */
  212.  
  213. unsigned long int new_strings_size;
  214.  
  215. /* An archive map is a chain of these structures.
  216.   Each structure describes one member of the archive.
  217.   The chain is in the same order as the members are.  */
  218.  
  219. struct mapelt
  220.   {
  221.     struct member_desc info;
  222.     struct mapelt *next;
  223.   };
  224.  
  225. struct mapelt *maplast;
  226.  
  227. /* If nonzero, this is the map-element for the __.SYMDEF member
  228.    and we should update the time of that member just before finishing.  */
  229.  
  230. struct mapelt *symdef_mapelt;
  231.  
  232. /* Header that we wrote for the __.SYMDEF member.  */
  233.  
  234. struct ar_hdr symdef_header;
  235.  
  236. char *xmalloc (), *xrealloc ();
  237. void free ();
  238.  
  239. void add_to_map (), delete_from_map ();
  240. int insert_in_map ();
  241. void print_descr ();
  242. char *concat ();
  243. void scan ();
  244. void extract_members ();
  245. void extract_member ();
  246. void print_contents ();
  247. void write_symdef_member ();
  248. void read_old_symdefs ();
  249. void two_operations ();
  250. void usage (), fatal (), error (), error_with_file ();
  251. void perror_with_name (), pfatal_with_name ();
  252. void write_archive ();
  253. void touch_symdef_member ();
  254. void update_symdefs ();
  255. void delete_members (), move_members (), replace_members ();
  256. void quick_append ();
  257.  
  258. /* Output BYTES of data at BUF to the descriptor DESC.
  259.    FILE is the name of the file (for error messages).  */
  260.  
  261. void
  262. mywrite (desc, buf, bytes, file)
  263.      int desc;
  264.      char *buf;
  265.      int bytes;
  266.      char *file;
  267. {
  268.   register int val;
  269.  
  270.   while (bytes > 0)
  271.     {
  272.       val = write (desc, buf, bytes);
  273.       if (val <= 0)
  274.     perror_with_name (file);
  275.       buf += val;
  276.       bytes -= val;
  277.     }
  278. }
  279.  
  280. int
  281. main (argc, argv)
  282.      int argc;
  283.      char **argv;
  284. {
  285.   int i;
  286.  
  287.   operation = 0;
  288.   verbose = 0;
  289.   newer_only = 0;
  290.   silent_create = 0;
  291.   posname = 0;
  292.   postype = POS_DEFAULT;
  293.   preserve_dates = 0;
  294.   symdef_flag = 0;
  295.   symdef_exists = 0;
  296.   ignore_symdef = 0;
  297.   symdef_mapelt = 0;
  298.   files = 0;
  299.   lock_indesc = -1;
  300.  
  301.   if (argc < 2)
  302.     usage ("too few command arguments", 0);
  303.  
  304.   {
  305.     char *key = argv[1];
  306.     char *p = key;
  307.     char c;
  308.  
  309.     while (c = *p++)
  310.       {
  311.     switch (c)
  312.       {
  313.       case 'a':
  314.         postype = POS_AFTER;
  315.         break;
  316.  
  317.       case 'b':
  318.         postype = POS_BEFORE;
  319.         break;
  320.  
  321.       case 'c':
  322.         silent_create = 1;
  323.         break;
  324.  
  325.       case 'd':
  326.         if (operation)
  327.           two_operations ();
  328.  
  329.         operation = DELETE;
  330.         break;
  331.  
  332.       case 'i':
  333.         postype = POS_BEFORE;
  334.         break;
  335.  
  336.       case 'l':
  337.         break;
  338.  
  339.       case 'm':
  340.         if (operation)
  341.           two_operations ();
  342.         operation = MOVE;
  343.         break;
  344.  
  345.       case 'o':
  346.         preserve_dates = 1;
  347.         break;
  348.  
  349.       case 'p':
  350.         if (operation)
  351.           two_operations ();
  352.         operation = PRINT_FILES;
  353.         break;
  354.  
  355.       case 'q':
  356.         if (operation)
  357.           two_operations ();
  358.         operation = QUICK_APPEND;
  359.         break;
  360.  
  361.       case 'r':
  362.         if (operation)
  363.           two_operations ();
  364.         operation = REPLACE;
  365.         break;
  366.  
  367.       case 's':
  368.         symdef_flag = 1;
  369.         break;
  370.  
  371.       case 't':
  372.         if (operation)
  373.           two_operations ();
  374.         operation = PRINT_TABLE;
  375.         break;
  376.  
  377.       case 'u':
  378.         operation = REPLACE;
  379.         newer_only = 1;
  380.         break;
  381.  
  382.       case 'v':
  383.         verbose = 1;
  384.         break;
  385.  
  386.       case 'x':
  387.         if (operation)
  388.           two_operations ();
  389.         operation = EXTRACT;
  390.         break;
  391.       }
  392.       }
  393.   
  394.   }
  395.  
  396.   if (operation == 0 && symdef_flag)
  397.     operation = REPLACE;
  398.  
  399.   if (operation == 0)
  400.     usage ("no operation specified", 0);
  401.  
  402.   i = 2;
  403.  
  404.   if (postype != POS_DEFAULT)
  405.     posname = argv[i++];
  406.  
  407.   archive = argv[i++];
  408.  
  409.   if (i < argc)
  410.     {
  411.       files = &argv[i];
  412.       while (i < argc)
  413.     if (!strcmp (argv[i++], "__.SYMDEF"))
  414.       {
  415.         ignore_symdef = 1;
  416.         break;
  417.       }
  418.     }
  419.  
  420.   switch (operation)
  421.     {
  422.     case EXTRACT:
  423.     extract_members (extract_member);
  424.     break;
  425.  
  426.     case PRINT_TABLE:
  427.     extract_members (print_descr);
  428.     break;
  429.  
  430.     case PRINT_FILES:
  431.     extract_members (print_contents);
  432.     break;
  433.  
  434.     case DELETE:
  435.     if (files != 0)
  436.       delete_members ();
  437.     break;
  438.  
  439.     case MOVE:
  440.     if (files != 0)
  441.       move_members ();
  442.     break;
  443.  
  444.     case REPLACE:
  445.     if (files != 0 || symdef_flag)
  446.       replace_members ();
  447.     break;
  448.  
  449.     case QUICK_APPEND:
  450.     if (files != 0)
  451.       quick_append ();
  452.     break;
  453.  
  454.     default:
  455.     usage ("invalid operation %d", operation);
  456.     }
  457.  
  458.   exit (0);
  459.   return 0;
  460. }
  461.  
  462. void
  463. two_operations ()
  464. {
  465.   usage ("two different operation switches specified", 0);
  466. }
  467.  
  468. void
  469. scan (function, crflag)
  470.      void (*function) ();
  471.      int crflag;
  472. {
  473.   FILE *arcstream = fopen (archive, "r");
  474.  
  475.   if (arcstream == 0 && crflag)
  476.     /* Creation-warning, if desired, will happen later.  */
  477.     return;
  478.  
  479.   if (arcstream == 0)
  480.     {
  481.       perror_with_name (archive);
  482.       exit (1);
  483.     }
  484.   {
  485.     char buf[SARMAG];
  486.     int nread = fread (buf, 1, SARMAG, arcstream);
  487.     if (nread != SARMAG || bcmp (buf, ARMAG, SARMAG))
  488.       fatal ("file %s not a valid archive", archive);
  489.   }
  490.  
  491.   /* Now find the members one by one.  */
  492.   {
  493.     int member_offset = SARMAG;
  494.     while (1)
  495.       {
  496.     int nread;
  497.     struct ar_hdr member_header;
  498.     struct member_desc member_desc;
  499.     char name [1 + sizeof member_header.ar_name];
  500.  
  501.     if (fseek (arcstream, member_offset, 0) < 0)
  502.       perror_with_name (archive);
  503.  
  504.     nread = fread (&member_header, 1, sizeof (struct ar_hdr), arcstream);
  505.     if (nread == 0)
  506.       /* No data left means end of file; that is OK.  */
  507.       break;
  508.  
  509.     if (nread != sizeof (member_header)
  510.         || bcmp (member_header.ar_fmag, ARFMAG, 2))
  511.       fatal ("file %s not a valid archive", archive);
  512.     bcopy (member_header.ar_name, name, sizeof member_header.ar_name);
  513.     {
  514.       char *p = name + sizeof member_header.ar_name;
  515.       *p = '\0';
  516.       while (p > name && *--p == ' ')
  517.         *p = '\0';
  518.     }
  519.     member_desc.name = name;
  520.     sscanf (member_header.ar_mode, "%o", &member_desc.mode);
  521.     member_desc.date = atoi (member_header.ar_date);
  522.     member_desc.size = atoi (member_header.ar_size);
  523.     member_desc.uid = atoi (member_header.ar_uid);
  524.     member_desc.gid = atoi (member_header.ar_gid);
  525.     member_desc.offset = member_offset;
  526.     member_desc.data_offset = member_offset + sizeof (member_header);
  527.  
  528.     member_desc.new_offset = 0;
  529.     member_desc.symdefs = 0;
  530.     member_desc.nsymdefs = 0;
  531.     member_desc.string_size = 0;
  532.  
  533.     if (!ignore_symdef && !strcmp (name, "__.SYMDEF"))
  534.       symdef_exists = 1;
  535.  
  536.     function (member_desc, arcstream);
  537.  
  538.     member_offset += sizeof (member_header) + member_desc.size;
  539.     if (member_offset & 1)
  540.       ++member_offset;
  541.       }
  542.   }
  543.  
  544.   
  545.   fclose (arcstream);
  546. }
  547.  
  548. void print_modes ();
  549.  
  550. void
  551. print_descr (member)
  552.      struct member_desc member;
  553. {
  554.   char *timestring;
  555.   if (!verbose)
  556.     {
  557.       puts (member.name);
  558.       return;
  559.     }
  560.   print_modes (member.mode);
  561.   timestring = ctime (&member.date);
  562.   printf (" %2d/%2d %6d %12.12s %4.4s %s\n",
  563.       member.uid, member.gid,
  564.       member.size, timestring + 4, timestring + 20,
  565.       member.name);
  566. }
  567.  
  568. void
  569. print_modes (modes)
  570.      int modes;
  571. {
  572.   putchar (modes & 0400 ? 'r' : '-');
  573.   putchar (modes & 0200 ? 'w' : '-');
  574.   putchar (modes & 0100 ? 'x' : '-');
  575.   putchar (modes & 040 ? 'r' : '-');
  576.   putchar (modes & 020 ? 'w' : '-');
  577.   putchar (modes & 010 ? 'x' : '-');
  578.   putchar (modes & 04 ? 'r' : '-');
  579.   putchar (modes & 02 ? 'w' : '-');
  580.   putchar (modes & 01 ? 'x' : '-');
  581. }
  582.  
  583. #define BUFSIZE 1024
  584.  
  585. void
  586. extract_member (member, istream)
  587.      struct member_desc member;
  588.      FILE *istream;
  589. {
  590.   int ncopied = 0;
  591.   FILE *ostream;
  592.  
  593.   fseek (istream, member.data_offset, 0);
  594.   ostream = fopen (member.name, "w");
  595.   if (!ostream)
  596.     {
  597.       perror_with_name (member.name);
  598.       return;
  599.     }
  600.  
  601.   if (verbose)
  602.     printf ("x - %s\n", member.name);
  603.  
  604.   while (ncopied < member.size)
  605.     {
  606.       char buf [BUFSIZE];
  607.       int tocopy = member.size - ncopied;
  608.       int nread;
  609.       if (tocopy > BUFSIZE) tocopy = BUFSIZE;
  610.       nread = fread (buf, 1, tocopy, istream);
  611.       if (nread != tocopy)
  612.     fatal ("file %s not a valid archive", archive);
  613.       fwrite (buf, 1, nread, ostream);
  614.       ncopied += tocopy;
  615.     }
  616.  
  617. #ifdef USG
  618.   chmod (member.name, member.mode);
  619. #else
  620.   fchmod (fileno (ostream), member.mode);
  621. #endif
  622.   if (ferror (ostream) || fclose (ostream) != 0)
  623.     error ("%s: I/O error", member.name);
  624.  
  625.   if (preserve_dates)
  626.     {
  627. #ifdef USG
  628.       long tv[2];
  629.       tv[0] = member.date;
  630.       tv[1] = member.date;
  631.       utime (member.name, tv);
  632. #else
  633.       struct timeval tv[2];
  634.       tv[0].tv_sec = member.date;
  635.       tv[0].tv_usec = 0;
  636.       tv[1].tv_sec = member.date;
  637.       tv[1].tv_usec = 0;
  638.       utimes (member.name, tv);
  639. #endif
  640.     }
  641. }
  642.  
  643. void
  644. print_contents (member, istream)
  645.      struct member_desc member;
  646.      FILE *istream;
  647. {
  648.   int ncopied = 0;
  649.  
  650.   fseek (istream, member.data_offset, 0);
  651.  
  652.   if (verbose)
  653.   printf ("\n<member %s>\n\n", member.name);
  654.  
  655.   while (ncopied < member.size)
  656.     {
  657.       char buf [BUFSIZE];
  658.       int tocopy = member.size - ncopied;
  659.       int nread;
  660.       if (tocopy > BUFSIZE) tocopy = BUFSIZE;
  661.       nread = fread (buf, 1, tocopy, istream);
  662.       if (nread != tocopy)
  663.     fatal ("file %s not a valid archive", archive);
  664.       fwrite (buf, 1, nread, stdout);
  665.       ncopied += tocopy;
  666.     }
  667. }
  668.  
  669. /* Make a map of the existing members of the archive: their names,
  670.  positions and sizes.  */
  671.  
  672. /* If `nonexistent_ok' is nonzero,
  673.  just return 0 for an archive that does not exist.
  674.  This will cause the ordinary supersede procedure to
  675.  create a new archive.  */
  676.  
  677. struct mapelt *
  678. make_map (nonexistent_ok)
  679.      int nonexistent_ok;
  680. {
  681.   struct mapelt mapstart;
  682.   mapstart.next = 0;
  683.   maplast = &mapstart;
  684.   scan (add_to_map, nonexistent_ok);
  685.   return mapstart.next;
  686. }
  687.  
  688. void
  689. add_to_map (member)
  690.      struct member_desc member;
  691. {
  692.   struct mapelt *mapelt = (struct mapelt *) xmalloc (sizeof (struct mapelt));
  693.   mapelt->info = member;
  694.   mapelt->info.name = concat (mapelt->info.name, "", "");
  695.   maplast->next = mapelt;
  696.   mapelt->next = 0;
  697.   maplast = mapelt;
  698. }
  699.  
  700. /* Return the last element of the specified map.  */
  701.  
  702. struct mapelt *
  703. last_mapelt (map)
  704.      struct mapelt *map;
  705. {
  706.   struct mapelt *tail = map;
  707.   while (tail->next) tail = tail->next;
  708.   return tail;
  709. }
  710.  
  711. /* Return the element of the specified map which precedes elt.  */
  712.  
  713. struct mapelt *
  714. prev_mapelt (map, elt)
  715.      struct mapelt *map, *elt;
  716. {
  717.   struct mapelt *tail = map;
  718.   while (tail->next && tail->next != elt)
  719.     tail = tail->next;
  720.   if (tail->next) return tail;
  721.   return 0;
  722. }
  723.  
  724. /* Return the element of the specified map which has the specified name.  */
  725.  
  726. struct mapelt *
  727. find_mapelt_noerror (map, name)
  728.      struct mapelt *map;
  729.      register char *name;
  730. {
  731.   register struct mapelt *tail;
  732.   unsigned int len = strlen (name);
  733.   int dot_o = name[len - 2] == '.' && name[len - 1] == 'o';
  734.  
  735.   for (tail = map; tail != 0; tail = tail->next)
  736.     {
  737.       if (tail->info.name == 0)
  738.     continue;
  739.       if (!strncmp (tail->info.name, name, 13))
  740.     {
  741.       unsigned int eltlen = strlen (tail->info.name);
  742.       if (len <= 13 || eltlen <= 13)
  743.         return tail;
  744.       else
  745.         {
  746.           char *p = tail->info.name + 13;
  747.           if (dot_o && p[0] == '.' && p[1] == 'o' && p[2] == '\0')
  748.         return tail;
  749.           else if (!strncmp (p, name + 13,
  750.                  (len > eltlen ? len : eltlen) - 13))
  751.         return tail;
  752.         }
  753.     }
  754.     }
  755.  
  756.   return 0;
  757. }
  758.  
  759. struct mapelt *
  760. find_mapelt (map, name)
  761.      struct mapelt *map;
  762.      char *name;
  763. {
  764.   register struct mapelt *found = find_mapelt_noerror (map, name);
  765.   if (found == 0)
  766.     error ("no member named `%s'", name);
  767.   return found;
  768. }
  769.  
  770. /* Before looking at the archive, if we are going to update it
  771.    based on looking at its current contents, make an exclusive lock on it.
  772.    The lock is released when `write_archive' is called.  */
  773.  
  774. void
  775. lock_for_update ()
  776. {
  777.   /* Open the existing archive file; if that fails, create an empty one.  */
  778.  
  779.   lock_indesc = open (archive, O_RDWR, 0);
  780.  
  781.   if (lock_indesc < 0)
  782.     {
  783.       int outdesc;
  784.  
  785.       if (!silent_create)
  786.     printf ("Creating archive file `%s'\n", archive);
  787.       outdesc = open (archive, O_WRONLY | O_APPEND | O_CREAT, 0666);
  788.       if (outdesc < 0)
  789.     pfatal_with_name (archive);
  790.       mywrite (outdesc, ARMAG, SARMAG, archive);
  791.       close (outdesc);
  792.  
  793.       /* Now we had better be able to open for update!  */
  794.  
  795.       lock_indesc = open (archive, O_RDWR, 0);
  796.       if (lock_indesc < 0)
  797.     pfatal_with_name (archive);
  798.     }
  799.  
  800. #ifdef LOCKS
  801.   /* Lock the old file so that it won't be updated by two programs at once.
  802.      This uses the fcntl locking facility found on Sun systems
  803.      which is also in POSIX.  (Perhaps it comes from sysV.)
  804.  
  805.      Note that merely reading an archive does not require a lock,
  806.      because we use `rename' to update the whole file atomically.  */
  807.  
  808.   {
  809.     struct flock lock;
  810.  
  811.     lock.l_type = F_WRLCK;
  812.     lock.l_whence = 0;
  813.     lock.l_start = 0;
  814.     lock.l_len = 0;
  815.  
  816.     while (1)
  817.       {
  818.     int value = fcntl (lock_indesc, F_SETLKW, &lock);
  819.     if (value >= 0)
  820.       break;
  821.     else if (errno == EINTR)
  822.       continue;
  823.     else
  824.       pfatal_with_name ("locking archive");
  825.       }
  826.   }
  827. #endif
  828. }
  829.  
  830. /* Unlock archive and close the file descriptor.  */
  831.  
  832. void
  833. close_archive ()
  834. {
  835. #ifdef LOCKS
  836.   {
  837.     struct flock lock;
  838.  
  839.     /* Unlock the old archive.  */
  840.  
  841.     lock.l_type = F_UNLCK;
  842.     lock.l_whence = 0;
  843.     lock.l_start = 0;
  844.     lock.l_len = 0;
  845.  
  846.     fcntl (lock_indesc, F_SETLK, &lock);
  847.   }
  848. #endif
  849.  
  850.   /* Close the archive.  If we renamed a new one, the old one disappears.  */
  851.   close (lock_indesc);
  852. }
  853.  
  854. /* Write a new archive file from a given map.  */
  855. /* When a map is used as the pattern for a new archive,
  856.  each element represents one member to put in it, and
  857.  the order of elements controls the order of writing.
  858.  
  859.  Ordinarily, the element describes a member of the old
  860.  archive, to be copied into the new one.
  861.  
  862.  If the `offset' field of the element's info is 0,
  863.  then the element describes a file to be copied into the
  864.  new archive.  The `name' field is the file's name.
  865.  
  866.  If the `name' field of an element is 0, the element is ignored.
  867.  This makes it easy to specify deletion of archive members.
  868.  
  869.  Every operation that will eventually call `write_archive'
  870.  should call `lock_for_update' before beginning
  871.  to do any I/O on the archive file.
  872. */
  873.  
  874. char *make_tempname ();
  875. void copy_out_member ();
  876.  
  877. void
  878. write_archive (map, appendflag)
  879.      struct mapelt *map;
  880.      int appendflag;
  881. {
  882.   char *tempname = make_tempname (archive);
  883.   int indesc = lock_indesc;
  884.   int outdesc;
  885.   char *outname;
  886.   struct mapelt *tail;
  887.  
  888.   /* Now open the output.  */
  889.  
  890.   if (!appendflag)
  891.     {
  892.       /* Updating an existing archive normally.
  893.      Write output as TEMPNAME and rename at the end.
  894.      There can never be two invocations trying to do this at once,
  895.      because of the lock made on the old archive file.  */
  896.  
  897.       outdesc = open (tempname, O_WRONLY | O_CREAT, 0666);
  898.       if (outdesc < 0)
  899.     pfatal_with_name (tempname);
  900.       outname = tempname;
  901.       mywrite (outdesc, ARMAG, SARMAG, outname);
  902.     }
  903.   else
  904.     {
  905.       /* Fast-append to existing archive.  */
  906.  
  907.       outdesc = open (archive, O_WRONLY | O_APPEND, 0);
  908.       if (outdesc < 0)
  909.     pfatal_with_name (archive);
  910.       outname = archive;
  911.     }
  912.  
  913.   /* If archive has or should have a __.SYMDEF member,
  914.      compute the contents for it.  */
  915.  
  916.   if (symdef_flag || symdef_exists)
  917.     {
  918.       if (symdef_exists)
  919.     read_old_symdefs (map, indesc);
  920.       else
  921.     {
  922.       struct mapelt *this = (struct mapelt *)
  923.         xmalloc (sizeof (struct mapelt));
  924.       this->info.name = "__.SYMDEF";
  925.       this->info.offset = SARMAG;
  926.       this->info.data_offset = SARMAG + sizeof (struct ar_hdr);
  927.       this->info.new_offset = 0;
  928.       this->info.date = 0;
  929.       this->info.size = 0;
  930.       this->info.uid = 0;
  931.       this->info.gid = 0;
  932.       this->info.mode = 0666;
  933.       this->info.symdefs = 0;
  934.       this->info.nsymdefs = 0;
  935.       this->info.string_size = 0;
  936.       this->next = map;
  937.       map = this;
  938.       original_num_symdefs = 0;
  939.       old_strings_size = 0;
  940.     }
  941.  
  942.       update_symdefs (map, indesc);
  943.     }
  944.  
  945.   /* Copy the members into the output, either from the old archive
  946.      or from specified files.  */
  947.  
  948.   for (tail = map; tail != 0; tail = tail->next)
  949.     {
  950.       if ((symdef_flag || symdef_exists) && tail->info.name
  951.       && !strcmp (tail->info.name, "__.SYMDEF")
  952. #if 0
  953.       && tail->info.date==0
  954. #endif
  955.       )
  956.     write_symdef_member (tail, map, outdesc, outname);
  957.       else
  958.     copy_out_member (tail, indesc, outdesc, outname);
  959.     }
  960.  
  961.   if (symdef_mapelt != 0)
  962.     {
  963.       /* Check for members whose data offsets weren't
  964.      known when the symdef member was first written.  */
  965.       int doneany = 0;
  966.       for (tail = map; tail != 0; tail = tail->next)
  967.     if (tail->info.offset == 0)
  968.       {
  969.         /* Fix up the symdefs.  */
  970.         register unsigned int i;
  971.         for (i = 0; i < tail->info.nsymdefs; ++i)
  972.           tail->info.symdefs[i].offset = tail->info.new_offset;
  973.         doneany = 1;
  974.       }
  975.       if (doneany)
  976.     {
  977.       /* Some files had bad symdefs; rewrite the symdef member.  */
  978.       lseek (outdesc, symdef_mapelt->info.offset, 0);
  979.       write_symdef_member (symdef_mapelt, map, outdesc, outname);
  980.     }
  981.     }
  982.  
  983.   /* Mark the __.SYMDEF member as up to date.  */
  984.  
  985.   if (symdef_mapelt != 0)
  986.     touch_symdef_member (outdesc, outname);
  987.  
  988.   /* Install the new output under the intended name.  */
  989.  
  990. #ifndef USG
  991.   fsync (outdesc);
  992. #endif
  993.   close (outdesc);
  994.  
  995.   if (!appendflag)
  996.     if (rename (tempname, archive))
  997.       pfatal_with_name (tempname);
  998.  
  999.   close_archive ();
  1000. }
  1001.  
  1002. void
  1003. header_from_map (header, mapelt)
  1004.      struct ar_hdr *header;
  1005.      struct mapelt *mapelt;
  1006. {
  1007.   unsigned int namelen;
  1008.  
  1009.   /* Zero the header, then store in the data as text.  */
  1010.   bzero ((char *) header, sizeof (*header));
  1011.  
  1012.   strncpy (header->ar_name, mapelt->info.name, sizeof (header->ar_name));
  1013.   namelen = strlen (mapelt->info.name);
  1014.   if (namelen >= sizeof (header->ar_name))
  1015.     {
  1016.       if (mapelt->info.name[namelen - 2] == '.' &&
  1017.       mapelt->info.name[namelen - 1] == 'o')
  1018.     {
  1019.       header->ar_name[sizeof (header->ar_name) - 3] = '.';
  1020.       header->ar_name[sizeof (header->ar_name) - 2] = 'o';
  1021.     }
  1022.       header->ar_name[sizeof (header->ar_name) - 1] = '\0';
  1023.       error ("member name `%s' truncated to `%s'",
  1024.          mapelt->info.name, header->ar_name);
  1025.     }
  1026.  
  1027.   sprintf (header->ar_date, "%ld", mapelt->info.date);
  1028.   sprintf (header->ar_size, "%d", mapelt->info.size);
  1029.   sprintf (header->ar_uid, "%d", mapelt->info.uid);
  1030.   sprintf (header->ar_gid, "%d", mapelt->info.gid);
  1031.   sprintf (header->ar_mode, "%o", mapelt->info.mode);
  1032.   strncpy (header->ar_fmag, ARFMAG, sizeof (header->ar_fmag));
  1033.  
  1034.   /* Change all remaining nulls in the header into spaces.  */
  1035.   {
  1036.     char *end = (char *) &header[1];
  1037.     register char *p;
  1038.     for (p = (char *) header; p < end; ++p)
  1039.       if (*p == '\0')
  1040.     *p = ' ';
  1041.   }
  1042. }
  1043.  
  1044. /* writes to file open on OUTDESC with name OUTNAME.  */
  1045. void
  1046. copy_out_member (mapelt, archive_indesc, outdesc, outname)
  1047.      struct mapelt *mapelt;
  1048.      int archive_indesc;
  1049.      int outdesc;
  1050. {
  1051.   struct ar_hdr header;
  1052.   int indesc;
  1053.  
  1054.   if (mapelt->info.name == 0)
  1055.     /* This element was cancelled.  */
  1056.     return;
  1057.  
  1058.   header_from_map (&header, mapelt);
  1059.  
  1060.   if (mapelt->info.offset != 0)
  1061.     {
  1062.       indesc = archive_indesc;
  1063.       lseek (indesc, mapelt->info.data_offset, 0);
  1064.     }
  1065.   else
  1066.     {
  1067.       indesc = open (mapelt->info.name, 0, 0);
  1068.       if (indesc < 0)
  1069.     {
  1070.       perror_with_name (mapelt->info.name);
  1071.       return;
  1072.     }
  1073.     }
  1074.  
  1075.   mywrite (outdesc, &header, sizeof (header), outname);
  1076.  
  1077.   if (mapelt->info.data_offset == 0)
  1078.     mapelt->info.data_offset = lseek (outdesc, 0L, 1);
  1079.  
  1080.   {
  1081.     char buf[BUFSIZE];
  1082.     int tocopy = mapelt->info.size;
  1083.     while (tocopy > 0)
  1084.       {
  1085.     int thistime = tocopy;
  1086.     if (thistime > BUFSIZE) thistime = BUFSIZE;
  1087.         read (indesc, buf, thistime);
  1088.     mywrite (outdesc, buf, thistime, outname);
  1089.     tocopy -= thistime;
  1090.       }
  1091.   }
  1092.  
  1093.   if (indesc != archive_indesc)
  1094.     close (indesc);
  1095.  
  1096.   if (mapelt->info.size & 1)
  1097.     mywrite (outdesc, "\n", 1, outname);
  1098. }
  1099.  
  1100. /* Update the time of the __.SYMDEF member; done when we updated
  1101.    that member, just before we close the new archive file.
  1102.    It is open on OUTDESC and its name is OUTNAME.  */
  1103.  
  1104. void
  1105. touch_symdef_member (outdesc, outname)
  1106.      int outdesc;
  1107.      char *outname;
  1108. {
  1109.   struct stat statbuf;
  1110.   int i;
  1111.  
  1112.   /* See what mtime the archive file has as a result of our writing it.  */
  1113.   fstat (outdesc, &statbuf);
  1114.  
  1115.   /* Advance member's time to that time.  */
  1116.   bzero (symdef_header.ar_date, sizeof symdef_header.ar_date);
  1117.   sprintf (symdef_header.ar_date, "%ld", statbuf.st_mtime);
  1118.   for (i = 0; i < sizeof symdef_header.ar_date; i++)
  1119.     if (symdef_header.ar_date[i] == 0)
  1120.       symdef_header.ar_date[i] = ' ';
  1121.  
  1122.   /* Write back this member's header with the new time.  */
  1123.   if (lseek (outdesc, symdef_mapelt->info.new_offset, 0) >= 0)
  1124.     mywrite (outdesc, &symdef_header, sizeof symdef_header, outname);
  1125. }
  1126.  
  1127. char *
  1128. make_tempname (name)
  1129.      char *name;
  1130. {
  1131.   return concat (name, "", "_supersede");
  1132. }
  1133.  
  1134. void
  1135. delete_members ()
  1136. {
  1137.   struct mapelt *map = make_map (0);
  1138.   struct mapelt *tail;
  1139.   struct mapelt mapstart;
  1140.   char **p;
  1141.  
  1142.   mapstart.info.name = 0;
  1143.   mapstart.next = map;
  1144.   map = &mapstart;
  1145.  
  1146.   lock_for_update ();
  1147.  
  1148.   if (files)
  1149.     for (p = files; *p; p++)
  1150.       {
  1151.     /* If user says to delete the __.SYMDEF member,
  1152.        don't make a new one to replace it.  */
  1153.     if (!strcmp (*p, "__.SYMDEF"))
  1154.       symdef_exists = 0;
  1155.     delete_from_map (*p, map);
  1156.       }
  1157.  
  1158.   write_archive (map->next, 0);
  1159. }
  1160.  
  1161. void
  1162. delete_from_map (name, map)
  1163.      char *name;
  1164.      struct mapelt *map;
  1165. {
  1166.   struct mapelt *this = find_mapelt (map, name);
  1167.  
  1168.   if (!this) return;
  1169.   this->info.name = 0;
  1170.   if (verbose)
  1171.     printf ("d - %s\n", name);
  1172. }
  1173.  
  1174. void
  1175. move_members ()
  1176. {
  1177.   struct mapelt *map = make_map (0);
  1178.   char **p;
  1179.   struct mapelt *after_mapelt;
  1180.   struct mapelt mapstart;
  1181.   struct mapelt *change_map;
  1182.  
  1183.   mapstart.info.name = 0;
  1184.   mapstart.next = map;
  1185.   change_map = &mapstart;
  1186.  
  1187.   lock_for_update ();
  1188.  
  1189.   switch (postype)
  1190.     {
  1191.     case POS_DEFAULT:
  1192.       after_mapelt = last_mapelt (change_map);
  1193.       break;
  1194.  
  1195.     case POS_AFTER:
  1196.       after_mapelt = find_mapelt (map, posname);
  1197.       break;
  1198.  
  1199.     case POS_BEFORE:
  1200.       after_mapelt = prev_mapelt (change_map, find_mapelt (map, posname));
  1201.     }
  1202.  
  1203.   /* Failure to find specified "before" or "after" member
  1204.      is a fatal error; message has already been printed.  */
  1205.  
  1206.   if (!after_mapelt) exit (1);
  1207.  
  1208.   if (files)
  1209.     for (p = files; *p; p++)
  1210.       {
  1211.     if (move_in_map (*p, change_map, after_mapelt))
  1212.       after_mapelt = after_mapelt->next;
  1213.       }
  1214.  
  1215.   write_archive (map, 0);
  1216. }
  1217.  
  1218. int
  1219. move_in_map (name, map, after)
  1220.      char *name;
  1221.      struct mapelt *map, *after;
  1222. {
  1223.   struct mapelt *this = find_mapelt (map, name);
  1224.   struct mapelt *prev;
  1225.  
  1226.   if (!this) return 0;
  1227.   prev = prev_mapelt (map, this);
  1228.   prev->next = this->next;
  1229.   this->next = after->next;
  1230.   after->next = this;
  1231.   return 1;
  1232. }
  1233.  
  1234. /* Insert files into the archive.  */
  1235.  
  1236. void
  1237. replace_members ()
  1238. {
  1239.   struct mapelt *map = make_map (1);
  1240.   struct mapelt mapstart;
  1241.   struct mapelt *after_mapelt;
  1242.   struct mapelt *change_map;
  1243.   char **p;
  1244.   int changed;
  1245.  
  1246.   mapstart.info.name = 0;
  1247.   mapstart.next = map;
  1248.   change_map = &mapstart;
  1249.  
  1250.   lock_for_update ();
  1251.  
  1252.   switch (postype)
  1253.     {
  1254.     case POS_DEFAULT:
  1255.       after_mapelt = last_mapelt (change_map);
  1256.       break;
  1257.  
  1258.     case POS_AFTER:
  1259.       after_mapelt = find_mapelt (map, posname);
  1260.       break;
  1261.  
  1262.     case POS_BEFORE:
  1263.       after_mapelt = prev_mapelt (change_map, find_mapelt (map, posname));
  1264.     }
  1265.  
  1266.   /* Failure to find specified "before" or "after" member
  1267.      is a fatal error; the message has already been printed.  */
  1268.   if (after_mapelt == 0)
  1269.     exit (1);
  1270.  
  1271.   changed = 0;
  1272.   if (files != 0)
  1273.     for (p = files; *p != 0; ++p)
  1274.       if (insert_in_map (*p, change_map, after_mapelt))
  1275.     {
  1276.       after_mapelt = after_mapelt->next;
  1277.       changed = 1;
  1278.     }
  1279.  
  1280.   change_map = change_map->next;
  1281.   if (!changed && (!symdef_flag || symdef_exists))
  1282.     /* Nothing changed.  */
  1283.     close_archive (change_map);
  1284.   else
  1285.     write_archive (change_map, 0);
  1286. }
  1287.  
  1288. /* Handle the "quick insert" operation.  */
  1289.  
  1290. void
  1291. quick_append ()
  1292. {
  1293.   struct mapelt *map;
  1294.   struct mapelt *after;
  1295.   struct mapelt mapstart;
  1296.   char **p;
  1297.  
  1298.   mapstart.info.name = 0;
  1299.   mapstart.next = 0;
  1300.   map = &mapstart;
  1301.   after = map;
  1302.  
  1303.   lock_for_update ();
  1304.  
  1305.   /* Insert the specified files into the "map",
  1306.      but is a map of the inserted files only,
  1307.      and starts out empty.  */
  1308.   if (files)
  1309.     for (p = files; *p; p++)
  1310.       {
  1311.     if (insert_in_map (*p, map, after))
  1312.       after = after->next;
  1313.       }
  1314.  
  1315.   /* Append these files to the end of the existing archive file.  */
  1316.  
  1317.   write_archive (map->next, 1);
  1318. }
  1319.  
  1320. /* Insert an entry for name NAME into the map MAP after the map entry AFTER.
  1321.    Delete an old entry for NAME.
  1322.    MAP is assumed to start with a dummy entry, which facilitates
  1323.    insertion at the beginning of the list.
  1324.    Return 1 if successful, 0 if did nothing because file NAME doesn't
  1325.    exist or (optionally) is older.  */
  1326.  
  1327. int
  1328. insert_in_map (name, map, after)
  1329.      char *name;
  1330.      struct mapelt *map, *after;
  1331. {
  1332.   struct mapelt *old = find_mapelt_noerror (map, name);
  1333.   struct mapelt *this;
  1334.   struct stat status;
  1335.  
  1336.   if (stat (name, &status))
  1337.     {
  1338.       perror_with_name (name);
  1339.       return 0;
  1340.     }
  1341.   if (old && newer_only && status.st_mtime <= old->info.date)
  1342.     return 0;
  1343.   if (old)
  1344.     /* Delete the old one.  */
  1345.     old->info.name = 0;
  1346.   this = (struct mapelt *) xmalloc (sizeof (struct mapelt));
  1347.   this->info.name = name;
  1348.   this->info.offset = 0;
  1349.   this->info.data_offset = 0;
  1350.   this->info.date = status.st_mtime;
  1351.   this->info.size = status.st_size;
  1352.   this->info.uid = status.st_uid;
  1353.   this->info.gid = status.st_gid;
  1354.   this->info.mode = status.st_mode;
  1355.   this->next = after->next;
  1356.   after->next = this;
  1357.  
  1358.   if (verbose)
  1359.     printf ("%c - %s\n", old == 0 ? 'a' : 'r', this->info.name);
  1360.  
  1361.   return 1;
  1362. }
  1363.  
  1364. /* Apply a function to each of the specified members.
  1365. */
  1366.  
  1367. void
  1368. extract_members (function)
  1369.      void (*function) ();
  1370. {
  1371.   struct mapelt *map;
  1372.   FILE *arcstream;
  1373.   char **p;
  1374.  
  1375.   if (!files)
  1376.     {
  1377.       /* Handle case where we want to operate on every member.
  1378.      No need to make a map and search it for this.  */
  1379.       scan (function, 0);
  1380.       return;
  1381.     }
  1382.  
  1383.   arcstream = fopen (archive, "r");
  1384.   if (!arcstream)
  1385.     fatal ("failure opening archive %s for the second time", archive);
  1386.   map = make_map (0);
  1387.  
  1388.   for (p = files; *p; p++)
  1389.     {
  1390.       struct mapelt *this = find_mapelt (map, *p);
  1391.       if (!this) continue;
  1392.       function (this->info, arcstream);
  1393.     }
  1394.  
  1395.   fclose (arcstream);
  1396. }
  1397.  
  1398. /* Write the __.SYMDEF member from data in core.  OUTDESC and OUTNAME
  1399.    are descriptor and name of file to write to.  */
  1400.  
  1401. void
  1402. write_symdef_member (mapelt, map, outdesc, outname)
  1403.      struct mapelt *mapelt;
  1404.      struct mapelt *map;
  1405.      int outdesc;
  1406.      char *outname;
  1407. {
  1408.   struct ar_hdr header;
  1409.   int indesc;
  1410.   struct mapelt *mapptr;
  1411.   unsigned long int symdefs_size;
  1412.  
  1413.   if (mapelt->info.name == 0)
  1414.     /* This element was cancelled.  */
  1415.     return;
  1416.  
  1417.   header_from_map (&header, mapelt);
  1418.  
  1419.   bcopy (&header, &symdef_header, sizeof header);
  1420.  
  1421.   mywrite (outdesc, &header, sizeof (header), outname);
  1422.  
  1423.   /* Write the number of symdefs.  */
  1424.   symdefs_size = nsymdefs * sizeof (struct symdef);
  1425.   mywrite (outdesc, &symdefs_size, sizeof symdefs_size, outname);
  1426.  
  1427.   /* Write symdefs surviving from old archive.  */
  1428.   mywrite (outdesc, old_symdefs, num_old_symdefs * sizeof (struct symdef),
  1429.        outname);
  1430.  
  1431.   /* Write symdefs for new members.  */
  1432.   for (mapptr = map; mapptr != 0; mapptr = mapptr->next)
  1433.     if (mapptr->info.nsymdefs != 0)
  1434.       write (outdesc, mapptr->info.symdefs,
  1435.          mapptr->info.nsymdefs * sizeof (struct symdef));
  1436.  
  1437.   /* Write the string table size.  */
  1438.   mywrite (outdesc, &new_strings_size, sizeof new_strings_size, outname);
  1439.  
  1440.   /* Write the string table.  */
  1441.   mywrite (outdesc, new_strings, new_strings_size, outname);
  1442.  
  1443.   if (mapelt->info.size & 1)
  1444.     mywrite (outdesc, "", 1, outname);
  1445. }
  1446.  
  1447. void
  1448. read_old_symdefs (map, archive_indesc)
  1449.      struct mapelt *map;
  1450.      int archive_indesc;
  1451. {
  1452.   struct mapelt *mapelt;
  1453.   char *data;
  1454.   int val;
  1455.   int symdefs_size;
  1456.  
  1457.   mapelt = find_mapelt_noerror (map, "__.SYMDEF");
  1458.   if (!mapelt)
  1459.     abort ();            /* Only call here if an old one exists */
  1460.  
  1461.   data  = (char *) xmalloc (mapelt->info.size);
  1462.   lseek (archive_indesc, mapelt->info.data_offset, 0);
  1463.   val = read (archive_indesc, data, mapelt->info.size);
  1464.  
  1465.   symdefs_size = *(unsigned long int *) data;
  1466.   original_num_symdefs = symdefs_size / sizeof (struct symdef);
  1467.   old_symdefs = (struct symdef *) (data + sizeof (symdefs_size));
  1468.   old_strings = ((char *) (old_symdefs + original_num_symdefs)
  1469.          + sizeof (symdefs_size));
  1470.   old_strings_size
  1471.     = *(unsigned long int *) (old_symdefs + original_num_symdefs);
  1472. }
  1473.  
  1474. /* Read various information from the header of an object file.
  1475.    Return 0 for failure or 1 for success.  */
  1476.  
  1477. int
  1478. read_header_info (mapelt, desc, offset, syms_offset, syms_size, strs_offset, strs_size)
  1479.      struct mapelt *mapelt;
  1480.      int desc;
  1481.      long int offset;
  1482.      long int *syms_offset;
  1483.      unsigned int *syms_size;
  1484.      long int *strs_offset;
  1485.      unsigned int *strs_size;
  1486. {
  1487. #ifdef A_OUT
  1488.   {
  1489.     struct exec hdr;
  1490.  
  1491.     lseek (desc, offset, 0);
  1492. #ifdef HEADER_SEEK_FD
  1493.     HEADER_SEEK_FD (desc);
  1494. #endif
  1495.  
  1496.     if (read (desc, (char *) &hdr, sizeof hdr) == sizeof hdr && !N_BADMAG(hdr))
  1497.       {
  1498.     *syms_offset = N_SYMOFF (hdr);
  1499.     *syms_size = hdr.a_syms;
  1500.     *strs_offset = N_STROFF (hdr);
  1501.     lseek (desc, N_STROFF (hdr) + offset, 0);
  1502.     if (read (desc, (char *) strs_size, sizeof *strs_size) != sizeof *strs_size)
  1503.       {
  1504.         error_with_file ("failure reading string table size in ", mapelt);
  1505.         return 0;
  1506.       }
  1507.     return 1;
  1508.       }
  1509.   }
  1510. #endif
  1511.  
  1512. #ifdef MACH_O
  1513.   {
  1514.     struct mach_header mach_header;
  1515.     struct load_command *load_command;
  1516.     struct symtab_command *symtab_command;
  1517.     char *hdrbuf;
  1518.     int cmd, symtab_seen;
  1519.  
  1520.     lseek (desc, offset, 0);
  1521.     if (read (desc, (char *) &mach_header, sizeof mach_header) == sizeof mach_header
  1522.     && mach_header.magic == MH_MAGIC)
  1523.       {
  1524.     hdrbuf = xmalloc (mach_header.sizeofcmds);
  1525.     if (read (desc, hdrbuf, mach_header.sizeofcmds) != mach_header.sizeofcmds)
  1526.       {
  1527.         error_with_file ("failure reading load commands of ", mapelt);
  1528.         return 0;
  1529.       }
  1530.     load_command = (struct load_command *) hdrbuf;
  1531.     symtab_seen = 0;
  1532.     for (cmd = 0; cmd < mach_header.ncmds; ++cmd)
  1533.       {
  1534.         if (load_command->cmd == LC_SYMTAB)
  1535.           {
  1536.         symtab_seen = 1;
  1537.         symtab_command = (struct symtab_command *) load_command;
  1538.         *syms_offset = symtab_command->symoff;
  1539.         *syms_size = symtab_command->nsyms * sizeof (struct nlist);
  1540.         *strs_offset = symtab_command->stroff;
  1541.         *strs_size = symtab_command->strsize;
  1542.           }
  1543.         load_command = (struct load_command *) ((char *) load_command + load_command->cmdsize);
  1544.       }
  1545.     free (hdrbuf);
  1546.     if (!symtab_seen)
  1547.       *syms_offset = *syms_size = *strs_offset = *strs_size = 0;
  1548.     return 1;
  1549.       }
  1550.   }
  1551. #endif
  1552.  
  1553.   error_with_file ("bad format (not an object file) in ", mapelt);
  1554.   return 0;
  1555. }
  1556.  
  1557. /* Create the info.symdefs for a new member
  1558.    by reading the file it is coming from.  */
  1559.  
  1560. void
  1561. make_new_symdefs (mapelt, archive_indesc)
  1562.      struct mapelt *mapelt;
  1563.      int archive_indesc;
  1564. {
  1565.   int indesc;
  1566.   char *name = mapelt->info.name;
  1567.   long int syms_offset, strs_offset;
  1568.   unsigned int syms_size, strs_size;
  1569.   struct nlist *symbols;
  1570.   int symcount;
  1571.   char *strings;
  1572.   register unsigned int i;
  1573.   unsigned long int offset;
  1574.  
  1575.   if (name == 0)
  1576.     /* Deleted member.  */
  1577.     abort ();
  1578.  
  1579.   if (mapelt->info.offset != 0)
  1580.     {
  1581.       indesc = archive_indesc;
  1582.       lseek (indesc, mapelt->info.data_offset, 0);
  1583.       offset = mapelt->info.data_offset;
  1584.     }
  1585.   else
  1586.     {
  1587.       indesc = open (mapelt->info.name, 0, 0);
  1588.       if (indesc < 0)
  1589.     {
  1590.       perror_with_name (mapelt->info.name);
  1591.       return;
  1592.     }
  1593.       offset = 0;
  1594.     }
  1595.  
  1596.   if (!read_header_info (mapelt, indesc, offset, &syms_offset, &syms_size, &strs_offset, &strs_size))
  1597.     {
  1598.       if (mapelt->info.offset == 0)
  1599.     close (indesc);
  1600.       return;
  1601.     }
  1602.  
  1603.   /* Number of symbol entries in the file.  */
  1604.   symcount = syms_size / sizeof (struct nlist);
  1605.   /* Allocate temporary space for the symbol entries.  */
  1606.   symbols = (struct nlist *) alloca (syms_size);
  1607.   /* Read in the symbols.  */
  1608.   lseek (indesc, syms_offset + offset, 0);
  1609.   if (read (indesc, (char *) symbols, syms_size) != syms_size)
  1610.     {
  1611.       error_with_file ("premature end of file in symbols of ", mapelt);
  1612.       if (mapelt->info.offset == 0)
  1613.     (void) close (indesc);
  1614.       return;
  1615.     }
  1616.  
  1617.   /* The string table size includes the size word.  */
  1618.   if (strs_size < sizeof (strs_size))
  1619.     {
  1620.       error_with_file ("bad string table size in ", mapelt);
  1621.       if (mapelt->info.offset == 0)
  1622.     (void) close (indesc);
  1623.       return;
  1624.     }
  1625.   strs_size -= sizeof (strs_size);
  1626.  
  1627.   /* Allocate permanent space for the string table.  */
  1628.   strings = (char *) xmalloc (strs_size);
  1629.  
  1630.   /* Read in the strings.  */
  1631.   lseek (indesc, offset + strs_offset + sizeof strs_size, 0);
  1632.   if (read (indesc, strings, strs_size) != strs_size)
  1633.     {
  1634.       error_with_file ("premature end of file in strings of ", mapelt);
  1635.       if (mapelt->info.offset == 0)
  1636.     (void) close (indesc);
  1637.       return;
  1638.     }
  1639.  
  1640.   if (indesc != archive_indesc)
  1641.     (void) close (indesc);
  1642.  
  1643.   /* Discard the symbols we don't want to mention; compact the rest down.  */
  1644.   symcount = filter_symbols (symbols, symcount);
  1645.  
  1646.   mapelt->info.symdefs = (struct symdef *)
  1647.     xmalloc (symcount * sizeof (struct symdef));
  1648.   mapelt->info.nsymdefs = symcount;
  1649.   mapelt->info.string_size = 0;
  1650.  
  1651.   for (i = 0; i < symcount; ++i)
  1652.     {
  1653.       unsigned long int stroff = symbols[i].n_un.n_strx - sizeof (strs_size);
  1654.       char *symname = strings + stroff;
  1655.       if (stroff > strs_size)
  1656.     {
  1657.       char buf[100];
  1658.       sprintf (buf, "ridiculous string offset %lu in symbol %u of ",
  1659.            stroff + sizeof (strs_size), i);
  1660.       error_with_file (buf, mapelt);
  1661.       return;
  1662.     }
  1663.       mapelt->info.symdefs[i].s.name = symname;
  1664.       mapelt->info.string_size += strlen (symname) + 1;
  1665.     }
  1666. }
  1667.  
  1668. /* Choose which symbol entries to mention in __.SYMDEF;
  1669.    compact them downward to get rid of the rest.
  1670.    Return the number of symbols left.  */
  1671.  
  1672. int
  1673. filter_symbols (syms, symcount)
  1674.      struct nlist *syms;
  1675.      unsigned int symcount;
  1676. {
  1677.   struct nlist *from, *to;
  1678.   struct nlist *end = syms + symcount;
  1679.  
  1680.   for (to = from = syms; from < end; ++from)
  1681.     if ((from->n_type & N_EXT)
  1682.     && (from->n_type != N_EXT || from->n_value != 0))
  1683.       *to++ = *from;
  1684.  
  1685.   return to - syms;
  1686. }
  1687.  
  1688.  
  1689. /* Update the __.SYMDEF data before writing a new archive.  */
  1690.  
  1691. void
  1692. update_symdefs (map, archive_indesc)
  1693.      struct mapelt *map;
  1694.      int archive_indesc;
  1695. {
  1696.   struct mapelt *tail;
  1697.   int pos;
  1698.   register unsigned int i;
  1699.   unsigned int len;
  1700.   struct symdef *s;
  1701.   unsigned long int deleted_strings_size = 0;
  1702.  
  1703.   nsymdefs = original_num_symdefs;
  1704.   num_old_symdefs = original_num_symdefs;
  1705.   new_strings_size = old_strings_size;
  1706.  
  1707.   if (nsymdefs != 0)
  1708.     {
  1709.       /* We already had a __.SYMDEF member, so just update it.  */
  1710.  
  1711.       /* Mark as canceled any old symdefs for members being deleted.  */
  1712.  
  1713.       for (tail = map; tail != 0; tail = tail->next)
  1714.     {
  1715.       if (tail->info.name == 0)
  1716.         {
  1717.           /* Old member being deleted.  Delete its symdef entries too.  */
  1718.           for (i = 0; i < nsymdefs; i++)
  1719.         if (old_symdefs[i].offset == tail->info.offset)
  1720.           {
  1721.             old_symdefs[i].offset = 0;
  1722.             --nsymdefs;
  1723.             deleted_strings_size
  1724.               += strlen (old_strings
  1725.                  + old_symdefs[i].s.stringoffset) + 1;
  1726.           }
  1727.         }
  1728.     }
  1729.  
  1730.       /* Compactify old symdefs.  */
  1731.       {
  1732.     register unsigned int j = 0;
  1733.     for (i = 0; i < num_old_symdefs; ++i)
  1734.       {
  1735.         if (j != i)
  1736.           old_symdefs[j] = old_symdefs[i];
  1737.         if (old_symdefs[i].offset != 0)
  1738.           ++j;
  1739.       }
  1740.     num_old_symdefs -= i - j;
  1741.       }
  1742.  
  1743.       /* Create symdef data for any new members.  */
  1744.       for (tail = map; tail != 0; tail = tail->next)
  1745.     {
  1746.       if (tail->info.offset != 0
  1747.           || tail->info.name == 0
  1748.           || !strcmp (tail->info.name, "__.SYMDEF"))
  1749.         continue;
  1750.       make_new_symdefs (tail, archive_indesc);
  1751.       nsymdefs += tail->info.nsymdefs;
  1752.       new_strings_size += tail->info.string_size;
  1753.     }
  1754.     }
  1755.   else
  1756.     {
  1757.       /* Create symdef data for all existing members.  */
  1758.  
  1759.       for (tail = map; tail != 0; tail = tail->next)
  1760.     {
  1761.       if (tail->info.name == 0
  1762.           || !strcmp (tail->info.name, "__.SYMDEF"))
  1763.         continue;
  1764.       make_new_symdefs (tail, archive_indesc);
  1765.       nsymdefs += tail->info.nsymdefs;
  1766.       new_strings_size += tail->info.string_size;
  1767.     }
  1768.     }
  1769.  
  1770.   new_strings_size -= deleted_strings_size;
  1771.   old_strings_size -= deleted_strings_size;
  1772.  
  1773.   /* Now we know the size of __.SYMDEF,
  1774.      so assign the positions of all the members.  */
  1775.  
  1776.   tail = find_mapelt_noerror (map, "__.SYMDEF");
  1777.   tail->info.size = (sizeof (nsymdefs) + (nsymdefs * sizeof (struct symdef))
  1778.              + sizeof (new_strings_size) + new_strings_size);
  1779.   symdef_mapelt = tail;
  1780.  
  1781.   pos = SARMAG;
  1782.   for (tail = map; tail != 0; tail = tail->next)
  1783.     {
  1784.       if (tail->info.name == 0)
  1785.     /* Ignore deleted members.  */
  1786.     continue;
  1787.       tail->info.new_offset = pos;
  1788.       pos += sizeof (struct ar_hdr) + tail->info.size;
  1789.       if (tail->info.size & 1)
  1790.     ++pos;
  1791.     }
  1792.  
  1793.   /* Now update the offsets in the symdef data
  1794.      to be the new offsets rather than the old ones.  */
  1795.  
  1796.   for (tail = map; tail != 0; tail = tail->next)
  1797.     {
  1798.       if (tail->info.name == 0)
  1799.     continue;
  1800.       if (tail->info.symdefs == 0)
  1801.     /* Member without new symdef data.
  1802.        Check the old symdef data; it may be included there. */
  1803.     for (i = 0; i < num_old_symdefs; i++)
  1804.       {
  1805.         if (old_symdefs[i].offset == tail->info.offset)
  1806.           old_symdefs[i].offset = tail->info.new_offset;
  1807.       }
  1808.       else
  1809.     for (i = 0; i < tail->info.nsymdefs; i++)
  1810.       tail->info.symdefs[i].offset = tail->info.new_offset;
  1811.     }
  1812.  
  1813.   /* Generate new, combined string table and put each string's offset into the
  1814.      symdef that refers to it.  Note that old symdefs ref their strings by
  1815.      offsets into old_strings but new symdefs contain addresses of strings.  */
  1816.  
  1817.   new_strings = (char *) xmalloc (new_strings_size);
  1818.   pos = 0;
  1819.  
  1820.   /* Write the strings of the old symdefs and update the structures
  1821.      to contain indices into the string table instead of strings.  */
  1822.   for (i = 0; i < num_old_symdefs; i++)
  1823.     {
  1824.       strcpy (new_strings + pos, old_strings + old_symdefs[i].s.stringoffset);
  1825.       old_symdefs[i].s.stringoffset = pos;
  1826.       pos += strlen (new_strings + pos) + 1;
  1827.     }
  1828.   if (pos < old_strings_size)
  1829.     {
  1830.       unsigned int d = old_strings_size - pos;
  1831.       /* Correct the string table size.  */
  1832.       new_strings_size -= d;
  1833.       /* Correct the size of the `__.SYMDEF' member,
  1834.      since it contains the string table.  */
  1835.       symdef_mapelt->info.size -= d;
  1836.     }
  1837.   else if (pos > old_strings_size)
  1838.     fatal ("Old archive's string size was %u too small.",
  1839.        pos - old_strings_size);
  1840.  
  1841.   for (tail = map; tail != 0; tail = tail->next)
  1842.     if (tail->info.symdefs)
  1843.       {
  1844.     len = tail->info.nsymdefs;
  1845.     s = tail->info.symdefs;
  1846.  
  1847.     for (i = 0; i < len; i++)
  1848.       {
  1849.         strcpy (new_strings + pos, s[i].s.name);
  1850.         s[i].s.stringoffset = pos;
  1851.         pos += strlen (new_strings + pos) + 1;
  1852.       }
  1853.       }
  1854.   if (pos != new_strings_size)
  1855.     fatal ("internal error: inconsistency in new_strings_size", 0);
  1856. }
  1857.  
  1858. /* Print error message and usage message, and exit.  */
  1859.  
  1860. void
  1861. usage (s1, s2)
  1862.      char *s1, *s2;
  1863. {
  1864.   error (s1, s2);
  1865.   fprintf (stderr, "\
  1866. Usage: ar [d|m|p|q|r|t|x [[abi [position-name] [cilouv]] archive file...\n");
  1867.   exit (1);
  1868. }
  1869.  
  1870. /* Print error message and exit.  */
  1871.  
  1872. void
  1873. fatal (s1, s2)
  1874.      char *s1, *s2;
  1875. {
  1876.   error (s1, s2);
  1877.   exit (1);
  1878. }
  1879.  
  1880. /* Print error message.  `s1' is printf control string, the rest are args.  */
  1881.  
  1882. void
  1883. error (s1, s2, s3, s4, s5)
  1884.      char *s1, *s2, *s3, *s4, *s5;
  1885. {
  1886.   fprintf (stderr, "ar: ");
  1887.   fprintf (stderr, s1, s2, s3, s4, s5);
  1888.   fprintf (stderr, "\n");
  1889. }
  1890.  
  1891. void
  1892. error_with_file (string, mapelt)
  1893.      char *string;
  1894.      struct mapelt *mapelt;
  1895. {
  1896.   fprintf (stderr, "ar: ");
  1897.   fprintf (stderr, string);
  1898.   if (mapelt->info.offset != 0)
  1899.     fprintf (stderr, "%s(%s)", archive, mapelt->info.name);
  1900.   else
  1901.     fprintf (stderr, "%s", mapelt->info.name);
  1902.   fprintf (stderr, "\n");
  1903. }
  1904.  
  1905. void
  1906. perror_with_name (name)
  1907.      char *name;
  1908. {
  1909.   extern int errno, sys_nerr;
  1910.   extern char *sys_errlist[];
  1911.   char *s;
  1912.  
  1913.   if (errno < sys_nerr)
  1914.     s = concat ("", sys_errlist[errno], " for %s");
  1915.   else
  1916.     s = "unknown error for %s";
  1917.   error (s, name);
  1918. }
  1919.  
  1920. void
  1921. pfatal_with_name (name)
  1922.      char *name;
  1923. {
  1924.   extern int errno, sys_nerr;
  1925.   extern char *sys_errlist[];
  1926.   char *s;
  1927.  
  1928.   if (errno < sys_nerr)
  1929.     s = concat ("", sys_errlist[errno], " for %s");
  1930.   else
  1931.     s = "cannot open %s";
  1932.   fatal (s, name);
  1933. }
  1934.  
  1935. /* Return a newly-allocated string whose contents
  1936.    concatenate those of S1, S2, and S3.  */
  1937.  
  1938. char *
  1939. concat (s1, s2, s3)
  1940.      char *s1, *s2, *s3;
  1941. {
  1942.   int len1 = strlen (s1), len2 = strlen (s2), len3 = strlen (s3);
  1943.   char *result = (char *) xmalloc (len1 + len2 + len3 + 1);
  1944.  
  1945.   strcpy (result, s1);
  1946.   strcpy (result + len1, s2);
  1947.   strcpy (result + len1 + len2, s3);
  1948.   *(result + len1 + len2 + len3) = 0;
  1949.  
  1950.   return result;
  1951. }
  1952.  
  1953. /* Like malloc but get fatal error if memory is exhausted.  */
  1954.  
  1955. char *
  1956. xmalloc (size)
  1957.      unsigned int size;
  1958. {
  1959.   extern char *malloc ();
  1960.   char *result = malloc (size);
  1961.   if (result == 0)
  1962.     fatal ("virtual memory exhausted", 0);
  1963.   return result;
  1964. }
  1965.  
  1966. char *
  1967. xrealloc (ptr, size)
  1968.      char *ptr;
  1969.      unsigned int size;
  1970. {
  1971.   extern char *realloc ();
  1972.   char *result = realloc (ptr, size);
  1973.   if (result == 0)
  1974.     fatal ("virtual memory exhausted");
  1975.   return result;
  1976. }
  1977.  
  1978. #ifdef    USG
  1979. int
  1980. rename (from, to)
  1981.      char *from, *to;
  1982. {
  1983.   (void)unlink (to);
  1984.   if (link (from, to) < 0
  1985.       || unlink (from) < 0)
  1986.     return -1;
  1987.   else
  1988.     return 0;
  1989. }
  1990. #endif
  1991.