LBRYCommunity/tracker
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

middleware/client: be more specific with error messages

Browse source 
contributes to #160
This commit is contained in:
Josh de Kock 2016-04-05 01:10:07 +01:00
parent 20cd7c07ce
commit d139335111
2 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
server/store/middleware/client/blacklist.go
View file

@ -15,9 +15,9 @@ func init() {
tracker.RegisterAnnounceMiddleware("client_blacklist", blacklistAnnounceClient) tracker.RegisterAnnounceMiddleware("client_blacklist", blacklistAnnounceClient)
} }
// ErrBlockedClient is returned by an announce middleware if the announcing // ErrBlacklistedClient is returned by an announce middleware if the announcing
// Client is disallowed. // Client is blacklisted.
var ErrBlockedClient = tracker.ClientError("disallowed client") var ErrBlacklistedClient = tracker.ClientError("client blacklisted")
// blacklistAnnounceClient provides a middleware that only allows Clients to // blacklistAnnounceClient provides a middleware that only allows Clients to
// announce that are not stored in the StringStore. // announce that are not stored in the StringStore.
@ -27,7 +27,7 @@ func blacklistAnnounceClient(next tracker.AnnounceHandler) tracker.AnnounceHandl
if err != nil { if err != nil {
return err return err
} else if blacklisted { } else if blacklisted {
return ErrBlockedClient return ErrBlacklistedClient
} }
return next(cfg, req, resp) return next(cfg, req, resp)
} }

2
server/store/middleware/client/whitelist.go
View file

@ -30,7 +30,7 @@ func whitelistAnnounceClient(next tracker.AnnounceHandler) tracker.AnnounceHandl
if err != nil { if err != nil {
return err return err
} else if !whitelisted { } else if !whitelisted {
return ErrBlockedClient return ErrNotWhitelistedClient
} }
return next(cfg, req, resp) return next(cfg, req, resp)
} }