Buriburizaemon/nezha_domains
1
0
Fork 0
mirror of https://github.com/Buriburizaem0n/nezha_domains.git synced 2026-02-04 12:40:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

update profile api (#16)

Browse Source 
* update profile api

* rename

* fix realip assertion

* add waf api
This commit is contained in:
UUBulb
2024-11-23 16:22:22 +08:00
committed by GitHub
parent 22738b6244
commit 885330e948
7 changed files with 85 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
cmd/dashboard/rpc/rpc.go
View File

@@ -27,7 +27,8 @@ func ServeRPC() *grpc.Server {
}
func waf(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) {
if err := model.CheckIP(singleton.DB, ctx.Value(model.CtxKeyRealIP{}).(string)); err != nil {
realip, _ := ctx.Value(model.CtxKeyRealIP{}).(string)
if err := model.CheckIP(singleton.DB, realip); err != nil {
return nil, err
}
return handler(ctx, req)