jeans/ollama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

timecheck

Browse Source
This commit is contained in:
Josh Yan 2024-07-17 14:53:29 -07:00
parent bd8596d32b
commit 35b49739ec
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
cmd/cmd.go
View File

@ -343,6 +343,7 @@ func getLocalPath(ctx context.Context, digest string) (string, error) {
} }
request.Header.Set("Authorization", authz) request.Header.Set("Authorization", authz)
request.Header.Set("Timestamp", time.Now().Format(time.RFC3339))
request.Header.Set("User-Agent", fmt.Sprintf("ollama/%s (%s %s) Go/%s", version.Version, runtime.GOARCH, runtime.GOOS, runtime.Version())) request.Header.Set("User-Agent", fmt.Sprintf("ollama/%s (%s %s) Go/%s", version.Version, runtime.GOARCH, runtime.GOOS, runtime.Version()))
request.Header.Set("X-Redirect-Create", "1") request.Header.Set("X-Redirect-Create", "1")

10
server/routes.go
View File

@ -997,6 +997,16 @@ func (s *Server) IsServerKeyPublicKey(c *gin.Context) bool {
serverPublicKey, err := auth.GetPublicKey() serverPublicKey, err := auth.GetPublicKey()
if err != nil { if err != nil {
slog.Error(fmt.Sprintf("failed to get server public key: %v", err)) slog.Error(fmt.Sprintf("failed to get server public key: %v", err))
return false
}
timestamp, err := time.Parse(time.RFC3339, c.GetHeader("Timestamp"))
if err != nil {
return false
}
if time.Since(timestamp) > time.Minute {
return false
} }
if bytes.Equal(serverPublicKey.Marshal(), clientPublicKey.Marshal()) { if bytes.Equal(serverPublicKey.Marshal(), clientPublicKey.Marshal()) {