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