baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

add additional allowed hosts

Browse source
This commit is contained in:
Jeffrey Morgan 2024-03-08 23:23:59 -08:00
parent 6c0af2599e
commit 5c143af726
1 changed files with 33 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
server/routes.go
View file

@ -915,9 +915,9 @@ func allowedHost(host string) bool {
}
var tlds = []string{
".localhost",
".local",
".internal",
"localhost",
"local",
"internal",
}
for _, tld := range tlds {
@ -929,6 +929,27 @@ func allowedHost(host string) bool {
return false
}
func ips() []string {
var ips []string
if interfaces, err := net.Interfaces(); err == nil {
for _, iface := range interfaces {
addrs, err := iface.Addrs()
if err != nil {
continue
}
for _, a := range addrs {
if ip, _, err := net.ParseCIDR(a.String()); err == nil {
ips = append(ips, ip.String())
}
}
}
}
return ips
}
func allowedHostsMiddleware(addr net.Addr) gin.HandlerFunc {
return func(c *gin.Context) {
if addr == nil {
@ -936,17 +957,8 @@ func allowedHostsMiddleware(addr net.Addr) gin.HandlerFunc {
return
}
if !netip.MustParseAddrPort(addr.String()).Addr().IsLoopback() {
c.Next()
return
}
if addrPort, _ := netip.ParseAddrPort(c.Request.Host); addrPort.Addr().IsLoopback() {
c.Next()
return
}
if addr, _ := netip.ParseAddr(c.Request.Host); addr.IsLoopback() {
addr, err := netip.ParseAddrPort(addr.String())
if err == nil && !addr.Addr().IsLoopback() {
c.Next()
return
}
@ -956,6 +968,13 @@ func allowedHostsMiddleware(addr net.Addr) gin.HandlerFunc {
host = c.Request.Host
}
if addr, err := netip.ParseAddr(host); err == nil {
if addr.IsLoopback() || addr.IsPrivate() || slices.Contains(ips(), host) || addr.String() == "0.0.0.0" {
c.Next()
return
}
}
if allowedHost(host) {
c.Next()
return