Andrew's git - gitweb.git/blob

   1#include "cache.h"
   2#include "run-command.h"
   3#include "xdiff-interface.h"
   4#include "blob.h"
   5
   6static int fill_mmfile_blob(mmfile_t *f, struct blob *obj)
   7{
   8        void *buf;
   9        unsigned long size;
  10        enum object_type type;
  11
  12        buf = read_sha1_file(obj->object.sha1, &type, &size);
  13        if (!buf)
  14                return -1;
  15        if (type != OBJ_BLOB)
  16                return -1;
  17        f->ptr = buf;
  18        f->size = size;
  19        return 0;
  20}
  21
  22static void free_mmfile(mmfile_t *f)
  23{
  24        free(f->ptr);
  25}
  26
  27static void *three_way_filemerge(mmfile_t *base, mmfile_t *our, mmfile_t *their, unsigned long *size)
  28{
  29        mmbuffer_t res;
  30        xpparam_t xpp;
  31        int merge_status;
  32
  33        memset(&xpp, 0, sizeof(xpp));
  34        merge_status = xdl_merge(base, our, ".our", their, ".their",
  35                &xpp, XDL_MERGE_ZEALOUS, &res);
  36
  37        if (merge_status < 0)
  38                return NULL;
  39
  40        *size = res.size;
  41        return res.ptr;
  42}
  43
  44static int common_outf(void *priv_, mmbuffer_t *mb, int nbuf)
  45{
  46        int i;
  47        mmfile_t *dst = priv_;
  48
  49        for (i = 0; i < nbuf; i++) {
  50                memcpy(dst->ptr + dst->size, mb[i].ptr, mb[i].size);
  51                dst->size += mb[i].size;
  52        }
  53        return 0;
  54}
  55
  56static int generate_common_file(mmfile_t *res, mmfile_t *f1, mmfile_t *f2)
  57{
  58        unsigned long size = f1->size < f2->size ? f1->size : f2->size;
  59        void *ptr = xmalloc(size);
  60        xpparam_t xpp;
  61        xdemitconf_t xecfg;
  62        xdemitcb_t ecb;
  63
  64        xpp.flags = XDF_NEED_MINIMAL;
  65        memset(&xecfg, 0, sizeof(xecfg));
  66        xecfg.ctxlen = 3;
  67        xecfg.flags = XDL_EMIT_COMMON;
  68        ecb.outf = common_outf;
  69
  70        res->ptr = ptr;
  71        res->size = 0;
  72
  73        ecb.priv = res;
  74        return xdi_diff(f1, f2, &xpp, &xecfg, &ecb);
  75}
  76
  77void *merge_file(struct blob *base, struct blob *our, struct blob *their, unsigned long *size)
  78{
  79        void *res = NULL;
  80        mmfile_t f1, f2, common;
  81
  82        /*
  83         * Removed in either branch?
  84         *
  85         * NOTE! This depends on the caller having done the
  86         * proper warning about removing a file that got
  87         * modified in the other branch!
  88         */
  89        if (!our || !their) {
  90                enum object_type type;
  91                if (base)
  92                        return NULL;
  93                if (!our)
  94                        our = their;
  95                return read_sha1_file(our->object.sha1, &type, size);
  96        }
  97
  98        if (fill_mmfile_blob(&f1, our) < 0)
  99                goto out_no_mmfile;
 100        if (fill_mmfile_blob(&f2, their) < 0)
 101                goto out_free_f1;
 102
 103        if (base) {
 104                if (fill_mmfile_blob(&common, base) < 0)
 105                        goto out_free_f2_f1;
 106        } else {
 107                if (generate_common_file(&common, &f1, &f2) < 0)
 108                        goto out_free_f2_f1;
 109        }
 110        res = three_way_filemerge(&common, &f1, &f2, size);
 111        free_mmfile(&common);
 112out_free_f2_f1:
 113        free_mmfile(&f2);
 114out_free_f1:
 115        free_mmfile(&f1);
 116out_no_mmfile:
 117        return res;
 118}