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