use strip_suffix and xstrfmt to replace suffix

When we want to convert "foo.pack" to "foo.idx", we do it by
duplicating the original string and then munging the bytes
in place. Let's use strip_suffix and xstrfmt instead, which
has several advantages:

  1. It's more clear what the intent is.

  2. It does not implicitly rely on the fact that
     strlen(".idx") <= strlen(".pack") to avoid an overflow.

  3. We communicate the assumption that the input file ends
     with ".pack" (and get a run-time check that this is so).

  4. We drop calls to strcpy, which makes auditing the code
     base easier.

Likewise, we can do this to convert ".pack" to ".bitmap",
avoiding some manual memory computation.

Signed-off-by: Jeff King <peff@peff.net>
Signed-off-by: Junio C Hamano <gitster@pobox.com>
This commit is contained in:
Jeff King 2015-09-24 17:07:09 -04:00 committed by Junio C Hamano
parent 2805bb5970
commit 9ae97018fb
3 changed files with 12 additions and 14 deletions

7
http.c
View File

@ -1511,6 +1511,7 @@ int finish_http_pack_request(struct http_pack_request *preq)
struct packed_git **lst; struct packed_git **lst;
struct packed_git *p = preq->target; struct packed_git *p = preq->target;
char *tmp_idx; char *tmp_idx;
size_t len;
struct child_process ip = CHILD_PROCESS_INIT; struct child_process ip = CHILD_PROCESS_INIT;
const char *ip_argv[8]; const char *ip_argv[8];
@ -1524,9 +1525,9 @@ int finish_http_pack_request(struct http_pack_request *preq)
lst = &((*lst)->next); lst = &((*lst)->next);
*lst = (*lst)->next; *lst = (*lst)->next;
tmp_idx = xstrdup(preq->tmpfile); if (!strip_suffix(preq->tmpfile, ".pack.temp", &len))
strcpy(tmp_idx + strlen(tmp_idx) - strlen(".pack.temp"), die("BUG: pack tmpfile does not end in .pack.temp?");
".idx.temp"); tmp_idx = xstrfmt("%.*s.idx.temp", (int)len, preq->tmpfile);
ip_argv[0] = "index-pack"; ip_argv[0] = "index-pack";
ip_argv[1] = "-o"; ip_argv[1] = "-o";

View File

@ -252,16 +252,11 @@ static int load_bitmap_entries_v1(struct bitmap_index *index)
static char *pack_bitmap_filename(struct packed_git *p) static char *pack_bitmap_filename(struct packed_git *p)
{ {
char *idx_name; size_t len;
int len;
len = strlen(p->pack_name) - strlen(".pack"); if (!strip_suffix(p->pack_name, ".pack", &len))
idx_name = xmalloc(len + strlen(".bitmap") + 1); die("BUG: pack_name does not end in .pack");
return xstrfmt("%.*s.bitmap", (int)len, p->pack_name);
memcpy(idx_name, p->pack_name, len);
memcpy(idx_name + len, ".bitmap", strlen(".bitmap") + 1);
return idx_name;
} }
static int open_pack_bitmap_1(struct packed_git *packfile) static int open_pack_bitmap_1(struct packed_git *packfile)

View File

@ -671,13 +671,15 @@ static int check_packed_git_idx(const char *path, struct packed_git *p)
int open_pack_index(struct packed_git *p) int open_pack_index(struct packed_git *p)
{ {
char *idx_name; char *idx_name;
size_t len;
int ret; int ret;
if (p->index_data) if (p->index_data)
return 0; return 0;
idx_name = xstrdup(p->pack_name); if (!strip_suffix(p->pack_name, ".pack", &len))
strcpy(idx_name + strlen(idx_name) - strlen(".pack"), ".idx"); die("BUG: pack_name does not end in .pack");
idx_name = xstrfmt("%.*s.idx", (int)len, p->pack_name);
ret = check_packed_git_idx(idx_name, p); ret = check_packed_git_idx(idx_name, p);
free(idx_name); free(idx_name);
return ret; return ret;