xfnw/plan9fox
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

snap: fix readseg() to cope with > 2gb segments

Browse source
This commit is contained in:
cinap_lenrek 2015-03-07 15:21:01 +01:00
parent 7b7c7f4451
commit 0c705580ab
2 changed files with 16 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
sys/src/cmd/snap/snap.h
View file

@ -26,9 +26,9 @@ struct Data {
struct Seg {
char* name;
uvlong offset;
uvlong len;
uvlong len;
Page** pg;
int npg;
ulong npg;
};
struct Page {

30
sys/src/cmd/snap/take.c
View file

@ -96,47 +96,45 @@ readsection(long pid, char *sec)
}
static Seg*
readseg(int fd, vlong off, ulong len, char *name)
readseg(int fd, uvlong off, uvlong len, char *name)
{
char buf[Pagesize];
ulong npg;
Page **pg;
int npg;
Seg *s;
ulong i;
int n;
s = emalloc(sizeof(*s));
s->name = estrdup(name);
if(seek(fd, off, 0) < 0) {
fprint(2, "seek fails\n");
goto Die;
}
s->offset = off;
s->len = 0;
pg = nil;
npg = 0;
for(i=0; i<len; ) {
while(s->len < len){
n = Pagesize;
if(n > len-i)
n = len-i;
if(n > len - s->len)
n = len - s->len;
if((n = readn(fd, buf, n)) <= 0)
break;
pg = erealloc(pg, sizeof(*pg)*(npg+1));
s->len += n;
if((npg & (npg-1)) == 0)
pg = erealloc(pg, sizeof(*pg) * (npg==0 | npg*2));
pg[npg++] = datapage(buf, n);
i += n;
if(n != Pagesize) /* any short read, planned or otherwise */
break;
}
if(i==0 && len!=0)
if(s->len==0 && len!=0){
free(pg);
goto Die;
s->offset = off;
s->len = i;
}
pg = erealloc(pg, sizeof(*pg) * npg);
s->pg = pg;
s->npg = npg;
return s;
Die:
free(s->name);
free(s);