Andrew's git - gitweb.git/blob

   1#include <sys/types.h>
   2#include <sys/wait.h>
   3
   4#include "cache.h"
   5
   6static const char *pgm = NULL;
   7static const char *arguments[8];
   8static int err;
   9
  10static void run_program(void)
  11{
  12        int pid = fork(), status;
  13
  14        if (pid < 0)
  15                die("unable to fork");
  16        if (!pid) {
  17                execlp(pgm, arguments[0],
  18                            arguments[1],
  19                            arguments[2],
  20                            arguments[3],
  21                            arguments[4],
  22                            arguments[5],
  23                            arguments[6],
  24                            arguments[7],
  25                            NULL);
  26                die("unable to execute '%s'", pgm);
  27        }
  28        if (waitpid(pid, &status, 0) < 0 || !WIFEXITED(status) || WEXITSTATUS(status))
  29                err++;
  30}
  31
  32static int merge_entry(int pos, const char *path)
  33{
  34        int found;
  35        
  36        if (pos >= active_nr)
  37                die("merge-cache: %s not in the cache", path);
  38        arguments[0] = pgm;
  39        arguments[1] = "";
  40        arguments[2] = "";
  41        arguments[3] = "";
  42        arguments[4] = path;
  43        arguments[5] = "";
  44        arguments[6] = "";
  45        arguments[7] = "";
  46        found = 0;
  47        do {
  48                static char hexbuf[4][60];
  49                static char ownbuf[4][60];
  50                struct cache_entry *ce = active_cache[pos];
  51                int stage = ce_stage(ce);
  52
  53                if (strcmp(ce->name, path))
  54                        break;
  55                found++;
  56                strcpy(hexbuf[stage], sha1_to_hex(ce->sha1));
  57                sprintf(ownbuf[stage], "%o", ntohl(ce->ce_mode) & (~S_IFMT));
  58                arguments[stage] = hexbuf[stage];
  59                arguments[stage + 4] = ownbuf[stage];
  60        } while (++pos < active_nr);
  61        if (!found)
  62                die("merge-cache: %s not in the cache", path);
  63        run_program();
  64        return found;
  65}
  66
  67static void merge_file(const char *path)
  68{
  69        int pos = cache_name_pos(path, strlen(path));
  70
  71        /*
  72         * If it already exists in the cache as stage0, it's
  73         * already merged and there is nothing to do.
  74         */
  75        if (pos < 0)
  76                merge_entry(-pos-1, path);
  77}
  78
  79static void merge_all(void)
  80{
  81        int i;
  82        for (i = 0; i < active_nr; i++) {
  83                struct cache_entry *ce = active_cache[i];
  84                if (!ce_stage(ce))
  85                        continue;
  86                i += merge_entry(i, ce->name)-1;
  87        }
  88}
  89
  90int main(int argc, char **argv)
  91{
  92        int i, force_file = 0;
  93
  94        if (argc < 3)
  95                usage("merge-cache <merge-program> (-a | <filename>*)");
  96
  97        read_cache();
  98
  99        pgm = argv[1];
 100        for (i = 2; i < argc; i++) {
 101                char *arg = argv[i];
 102                if (!force_file && *arg == '-') {
 103                        if (!strcmp(arg, "--")) {
 104                                force_file = 1;
 105                                continue;
 106                        }
 107                        if (!strcmp(arg, "-a")) {
 108                                merge_all();
 109                                continue;
 110                        }
 111                        die("merge-cache: unknown option %s", arg);
 112                }
 113                merge_file(arg);
 114        }
 115        if (err)
 116                die("merge program failed");
 117        return 0;
 118}