yggdrasil-network/yggdrasil-go
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/yggdrasil-go.git synced 2025-04-28 22:25:07 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Replace repeated subscripts with single TrimPrefix

Browse source 
This stood out to me while reading the code:  [7:] is skipping "unix://",
so why not do that?

Doing so reveals a bug in the last line changed, where chmod(2) failure
would print just the prefix, not everything but it... easy to miss,
but now this kind of bug can no longer happen.
This commit is contained in:
Klemens Nanni 2024-09-30 00:45:53 +03:00
parent 6d5243bd9a
commit 8079a40bc8
No known key found for this signature in database
1 changed files with 12 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

23
src/admin/admin.go
View file

@ -238,27 +238,28 @@ func (a *AdminSocket) listen() {
if err == nil { if err == nil {
switch strings.ToLower(u.Scheme) { switch strings.ToLower(u.Scheme) {
case "unix": case "unix":
if _, err := os.Stat(listenaddr[7:]); err == nil { file := strings.TrimPrefix(listenaddr, "unix://")
a.log.Debugln("Admin socket", listenaddr[7:], "already exists, trying to clean up") if _, err := os.Stat(file); err == nil {
if _, err := net.DialTimeout("unix", listenaddr[7:], time.Second*2); err == nil || err.(net.Error).Timeout() { a.log.Debugln("Admin socket", file, "already exists, trying to clean up")
a.log.Errorln("Admin socket", listenaddr[7:], "already exists and is in use by another process") if _, err := net.DialTimeout("unix", file, time.Second*2); err == nil || err.(net.Error).Timeout() {
a.log.Errorln("Admin socket", file, "already exists and is in use by another process")
os.Exit(1) os.Exit(1)
} else { } else {
if err := os.Remove(listenaddr[7:]); err == nil { if err := os.Remove(file); err == nil {
a.log.Debugln(listenaddr[7:], "was cleaned up") a.log.Debugln(file, "was cleaned up")
} else { } else {
a.log.Errorln(listenaddr[7:], "already exists and was not cleaned up:", err) a.log.Errorln(file, "already exists and was not cleaned up:", err)
os.Exit(1) os.Exit(1)
} }
} }
} }
a.listener, err = net.Listen("unix", listenaddr[7:]) a.listener, err = net.Listen("unix", file)
if err == nil { if err == nil {
switch listenaddr[7:8] { switch file[:1] {
case "@": // maybe abstract namespace case "@": // maybe abstract namespace
default: default:
if err := os.Chmod(listenaddr[7:], 0660); err != nil { if err := os.Chmod(file, 0660); err != nil {
a.log.Warnln("WARNING:", listenaddr[:7], "may have unsafe permissions!") a.log.Warnln("WARNING:", file, "may have unsafe permissions!")
} }
} }
} }