@@ -8,7 +8,6 @@ package setting
8
8
import (
9
9
"encoding/base64"
10
10
"fmt"
11
- "io"
12
11
"math"
13
12
"net"
14
13
"net/url"
@@ -1080,28 +1079,22 @@ func loadInternalToken(sec *ini.Section) string {
1080
1079
}
1081
1080
switch tempURI .Scheme {
1082
1081
case "file" :
1083
- fp , err := os .OpenFile (tempURI .RequestURI (), os . O_RDWR , 0o600 )
1084
- if err != nil {
1082
+ buf , err := os .ReadFile (tempURI .RequestURI ())
1083
+ if err != nil && ! os . IsNotExist ( err ) {
1085
1084
log .Fatal ("Failed to open InternalTokenURI (%s): %v" , uri , err )
1086
1085
}
1087
- defer fp .Close ()
1088
-
1089
- buf , err := io .ReadAll (fp )
1090
- if err != nil {
1091
- log .Fatal ("Failed to read InternalTokenURI (%s): %v" , uri , err )
1092
- }
1093
1086
// No token in the file, generate one and store it.
1094
1087
if len (buf ) == 0 {
1095
1088
token , err := generate .NewInternalToken ()
1096
1089
if err != nil {
1097
1090
log .Fatal ("Error generate internal token: %v" , err )
1098
1091
}
1099
- if _ , err := io .WriteString (fp , token ); err != nil {
1092
+ err = os .WriteFile (tempURI .RequestURI (), []byte (token ), 0o600 )
1093
+ if err != nil {
1100
1094
log .Fatal ("Error writing to InternalTokenURI (%s): %v" , uri , err )
1101
1095
}
1102
1096
return token
1103
1097
}
1104
-
1105
1098
return strings .TrimSpace (string (buf ))
1106
1099
default :
1107
1100
log .Fatal ("Unsupported URI-Scheme %q (INTERNAL_TOKEN_URI = %q)" , tempURI .Scheme , uri )
0 commit comments