BSD 4 release

[unix-history] / usr / src / cmd / strip.c

static char *sccsid = "@(#)strip.c      4.1 (Berkeley) 10/1/80";
#include <a.out.h>
#include <signal.h>
#include <pagsiz.h>

#define BUFSIZ  BSIZE

char    *tname;
char    *mktemp();
struct exec head;
int     status;
int     tf;

main(argc, argv)
char *argv[];
{
       register i;

       signal(SIGHUP, SIG_IGN);
       signal(SIGINT, SIG_IGN);
       signal(SIGQUIT, SIG_IGN);
       tname = mktemp("/tmp/sXXXXX");
       close(creat(tname, 0600));
       tf = open(tname, 2);
       if(tf < 0) {
               printf("cannot create temp file\n");
               exit(2);
       }
       for(i=1; i<argc; i++) {
               strip(argv[i]);
               if(status > 1)
                       break;
       }
       close(tf);
       unlink(tname);
       exit(status);
}

strip(name)
char *name;
{
       register f;
       long size;
       int i;

       f = open(name, 0);
       if(f < 0) {
               printf("cannot open %s\n", name);
               status = 1;
               goto out;
       }
       read(f, (char *)&head, sizeof(head));
       if (N_BADMAG(head)) {
               printf("%s not in a.out format\n", name);
               status = 1;
               goto out;
       }
       if ((head.a_syms == 0) && (head.a_trsize == 0) && (head.a_drsize ==0)) {
               printf("%s already stripped\n", name);
               goto out;
       }
       size = (long)head.a_text + head.a_data;
       head.a_syms = head.a_trsize = head.a_drsize = 0 ;
       lseek(tf, (long)0, 0);
       write(tf, (char *)&head, sizeof(head));
       if (head.a_magic == ZMAGIC)
               size += PAGSIZ - sizeof (head);
       if (copy(name, f, tf, size)) {
               status = 1;
               goto out;
       }
       size += sizeof(head);
       close(f);
       f = creat(name, 0666);
       if(f < 0) {
               printf("%s cannot recreate\n", name);
               status = 1;
               goto out;
       }
       lseek(tf, (long)0, 0);
       if(copy(name, tf, f, size))
               status = 2;

out:
       close(f);
}

copy(name, fr, to, size)
char *name;
long size;
{
       register s, n;
       char buf[BUFSIZ];

       while(size != 0) {
               s = BUFSIZ;
               if(size < BUFSIZ)
                       s = size;
               n = read(fr, buf, s);
               if(n != s) {
                       printf("%s unexpected eof\n", name);
                       return(1);
               }
               n = write(to, buf, s);
               if(n != s) {
                       printf("%s unexpected write eof\n", name);
                       return(1);
               }
               size -= s;
       }
       return(0);
}