Commit 710d0540 authored by Shenghou Ma's avatar Shenghou Ma Committed by Russ Cox

cmd/dist: make dir check in defaulttarg() more robust

1, strip last path separator from $GOROOT
   The user might define GOROOT=/path/to/go/, but then the dir
   check in defaulttarg() will always complain the current dir
   is not within $GOROOT/src/.
2, resolve symlinks in the default goroot
   Or if getcwd() returns a fully-resolved path, the check in
   defaulttarg() will always fail.

R=rsc
CC=golang-dev
https://golang.org/cl/5649073
parent fc444eba
...@@ -77,8 +77,12 @@ init(void) ...@@ -77,8 +77,12 @@ init(void)
binit(&b); binit(&b);
xgetenv(&b, "GOROOT"); xgetenv(&b, "GOROOT");
if(b.len > 0) if(b.len > 0) {
// if not "/", then strip trailing path separator
if(b.len >= 2 && b.p[b.len - 1] == slash[0])
b.len--;
goroot = btake(&b); goroot = btake(&b);
}
xgetenv(&b, "GOBIN"); xgetenv(&b, "GOBIN");
if(b.len == 0) if(b.len == 0)
...@@ -1373,20 +1377,30 @@ static char* ...@@ -1373,20 +1377,30 @@ static char*
defaulttarg(void) defaulttarg(void)
{ {
char *p; char *p;
Buf pwd, src; Buf pwd, src, real_src;
binit(&pwd); binit(&pwd);
binit(&src); binit(&src);
binit(&real_src);
// xgetwd might return a path with symlinks fully resolved, and if
// there happens to be symlinks in goroot, then the hasprefix test
// will never succeed. Instead, we use xrealwd to get a canonical
// goroot/src before the comparison to avoid this problem.
xgetwd(&pwd); xgetwd(&pwd);
p = btake(&pwd); p = btake(&pwd);
bpathf(&src, "%s/src/", goroot); bpathf(&src, "%s/src/", goroot);
if(!hasprefix(p, bstr(&src))) xrealwd(&real_src, bstr(&src));
fatal("current directory %s is not under %s", p, bstr(&src)); if(!hasprefix(p, bstr(&real_src)))
p += src.len; fatal("current directory %s is not under %s", p, bstr(&real_src));
p += real_src.len;
// guard againt xrealwd return the directory without the trailing /
if(*p == slash[0])
p++;
bfree(&pwd); bfree(&pwd);
bfree(&src); bfree(&src);
bfree(&real_src);
return p; return p;
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment