2018-11-08 19:45:21 +00:00
|
|
|
import os
|
|
|
|
import json
|
2019-02-05 11:12:28 +00:00
|
|
|
import collections
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
from django.conf import settings
|
|
|
|
|
2019-12-05 16:57:52 +00:00
|
|
|
from peeringdb_server.models import InternetExchange, IXLan, Network
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
import django_namespace_perms.util as nsp
|
|
|
|
|
|
|
|
|
|
|
|
class CacheRedirect(Exception):
|
|
|
|
"""
|
|
|
|
Raise this error to redirect to cache response during viewset.get_queryset
|
|
|
|
or viewset.list()
|
|
|
|
|
|
|
|
Argument should be an APICacheLoader instance
|
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self, loader):
|
|
|
|
super(Exception, self).__init__(self, "Result to be loaded from cache")
|
|
|
|
self.loader = loader
|
|
|
|
|
|
|
|
|
|
|
|
###############################################################################
|
|
|
|
# API CACHE LOADER
|
|
|
|
|
|
|
|
|
2020-07-15 02:07:01 -05:00
|
|
|
class APICacheLoader:
|
2018-11-08 19:45:21 +00:00
|
|
|
"""
|
|
|
|
Checks if an API GET request qualifies for a cache load
|
|
|
|
and if it does allows you to provide the cached result
|
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self, viewset, qset, filters):
|
|
|
|
request = viewset.request
|
|
|
|
self.request = request
|
|
|
|
self.qset = qset
|
|
|
|
self.filters = filters
|
|
|
|
self.model = viewset.model
|
|
|
|
self.viewset = viewset
|
|
|
|
self.depth = min(int(request.query_params.get("depth", 0)), 3)
|
|
|
|
self.limit = int(request.query_params.get("limit", 0))
|
|
|
|
self.skip = int(request.query_params.get("skip", 0))
|
|
|
|
self.since = int(request.query_params.get("since", 0))
|
|
|
|
self.fields = request.query_params.get("fields")
|
|
|
|
if self.fields:
|
|
|
|
self.fields = self.fields.split(",")
|
2019-12-05 16:57:52 +00:00
|
|
|
self.path = os.path.join(
|
2020-07-26 23:36:27 -05:00
|
|
|
settings.API_CACHE_ROOT, f"{viewset.model.handleref.tag}-{self.depth}.json",
|
2019-12-05 16:57:52 +00:00
|
|
|
)
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def qualifies(self):
|
|
|
|
"""
|
|
|
|
Check if request qualifies for a cache load
|
|
|
|
"""
|
|
|
|
|
|
|
|
# api cache use is disabled, no
|
|
|
|
if not getattr(settings, "API_CACHE_ENABLED", False):
|
|
|
|
return False
|
|
|
|
# no depth and a limit lower than 251 seems like a tipping point
|
|
|
|
# were non-cache retrieval is faster still
|
2019-12-05 16:57:52 +00:00
|
|
|
if (
|
|
|
|
not self.depth
|
|
|
|
and self.limit
|
|
|
|
and self.limit <= 250
|
|
|
|
and getattr(settings, "API_CACHE_ALL_LIMITS", False) is False
|
|
|
|
):
|
2018-11-08 19:45:21 +00:00
|
|
|
return False
|
|
|
|
# filters have been specified, no
|
|
|
|
if self.filters or self.since:
|
|
|
|
return False
|
|
|
|
# cache file non-existant, no
|
|
|
|
if not os.path.exists(self.path):
|
|
|
|
return False
|
|
|
|
# request method is anything but GET, no
|
|
|
|
if self.request.method != "GET":
|
|
|
|
return False
|
|
|
|
# primary key set in request, no
|
|
|
|
if self.viewset.kwargs:
|
|
|
|
return False
|
|
|
|
|
|
|
|
return True
|
|
|
|
|
|
|
|
def load(self):
|
|
|
|
"""
|
|
|
|
Load the cached response according to tag and depth
|
|
|
|
"""
|
|
|
|
|
|
|
|
# read cache file
|
2020-07-15 02:07:01 -05:00
|
|
|
with open(self.path) as f:
|
2018-11-08 19:45:21 +00:00
|
|
|
data = json.load(f)
|
|
|
|
|
|
|
|
data = data.get("data")
|
|
|
|
|
|
|
|
# apply permissions to data
|
2019-12-05 16:57:52 +00:00
|
|
|
fnc = getattr(self, "apply_permissions_%s" % self.model.handleref.tag, None)
|
2018-11-08 19:45:21 +00:00
|
|
|
if fnc:
|
|
|
|
data = fnc(data)
|
|
|
|
|
|
|
|
# apply pagination
|
|
|
|
if self.skip and self.limit:
|
2019-12-05 16:57:52 +00:00
|
|
|
data = data[self.skip : self.skip + self.limit]
|
2018-11-08 19:45:21 +00:00
|
|
|
elif self.skip:
|
2019-12-05 16:57:52 +00:00
|
|
|
data = data[self.skip :]
|
2018-11-08 19:45:21 +00:00
|
|
|
elif self.limit:
|
2019-12-05 16:57:52 +00:00
|
|
|
data = data[: self.limit]
|
2018-11-08 19:45:21 +00:00
|
|
|
|
2019-12-05 16:57:52 +00:00
|
|
|
return {"results": data, "__meta": {"generated": os.path.getmtime(self.path)}}
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions(self, ns, data, ruleset={}):
|
|
|
|
"""
|
|
|
|
Wrapper function to apply permissions to a data row and
|
|
|
|
return the sanitized result
|
|
|
|
"""
|
|
|
|
if type(ns) != list:
|
|
|
|
ns = ns.split(".")
|
|
|
|
|
|
|
|
# prepare ruleset
|
|
|
|
if ruleset:
|
|
|
|
_ruleset = {}
|
|
|
|
namespace_str = ".".join(ns)
|
2020-01-08 13:29:58 -06:00
|
|
|
for section, rules in list(ruleset.items()):
|
2018-11-08 19:45:21 +00:00
|
|
|
_ruleset[section] = {}
|
2020-01-08 13:29:58 -06:00
|
|
|
for rule, perms in list(rules.items()):
|
2020-07-15 02:07:01 -05:00
|
|
|
_ruleset[section][f"{namespace_str}.{rule}"] = perms
|
2018-11-08 19:45:21 +00:00
|
|
|
ruleset = _ruleset
|
|
|
|
|
|
|
|
return nsp.dict_get_path(
|
|
|
|
nsp.permissions_apply(
|
2019-12-05 16:57:52 +00:00
|
|
|
nsp.dict_from_namespace(ns, data), self.request.user, ruleset=ruleset
|
|
|
|
),
|
|
|
|
ns,
|
|
|
|
)
|
2018-11-08 19:45:21 +00:00
|
|
|
|
2019-12-05 16:57:52 +00:00
|
|
|
def apply_permissions_generic(self, data, explicit=False, join_ids=[], **kwargs):
|
2018-11-08 19:45:21 +00:00
|
|
|
"""
|
|
|
|
Apply permissions to all rows according to rules
|
|
|
|
specified in parameters
|
|
|
|
|
|
|
|
explicit <function>
|
|
|
|
|
|
|
|
if explicit is passed as a function it will be called and the result will
|
|
|
|
determine whether or not explicit read perms are required for the row
|
|
|
|
|
|
|
|
join_ids [(target_id<str>, proxy_id<str>, model<handleref>), ..]
|
|
|
|
|
|
|
|
Since we are checking permissioning namespaces, and those namespaces may
|
|
|
|
consist of object ids that are not necessarily in the dataset you can
|
|
|
|
join those ids in via the join_ids parameter
|
|
|
|
"""
|
|
|
|
rv = []
|
|
|
|
|
2019-02-05 11:12:28 +00:00
|
|
|
joined_ids = collections.OrderedDict()
|
2018-11-08 19:45:21 +00:00
|
|
|
e = {}
|
|
|
|
inst = self.model()
|
|
|
|
|
|
|
|
# perform id joining
|
|
|
|
if join_ids:
|
|
|
|
for t, p, model in join_ids:
|
|
|
|
joined_ids[t] = {
|
|
|
|
"p": p,
|
2019-12-05 16:57:52 +00:00
|
|
|
"ids": self.join_ids(
|
2020-01-08 13:29:58 -06:00
|
|
|
data,
|
|
|
|
t,
|
|
|
|
p,
|
|
|
|
model,
|
|
|
|
list(joined_ids.get(p, e).get("ids", e).values()),
|
2019-12-05 16:57:52 +00:00
|
|
|
),
|
2018-11-08 19:45:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
for row in data:
|
|
|
|
|
|
|
|
# create dict containing ids needed to build the permissioning
|
|
|
|
# namespace
|
2020-07-15 02:07:01 -05:00
|
|
|
init = {k: row.get(v) for k, v in list(kwargs.items())}
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
# joined ids
|
2020-01-08 13:29:58 -06:00
|
|
|
for t, j in list(joined_ids.items()):
|
2018-11-08 19:45:21 +00:00
|
|
|
if j["p"] in row:
|
|
|
|
init[t] = j["ids"].get(row.get(j["p"]))
|
|
|
|
elif t in joined_ids:
|
|
|
|
init[t] = joined_ids.get(t).get("ids").get(init[j["p"]])
|
|
|
|
|
|
|
|
# build permissioning namespace
|
|
|
|
ns = self.model.nsp_namespace_from_id(**init).lower()
|
|
|
|
|
|
|
|
# apply fields filter
|
|
|
|
if self.fields:
|
2020-01-08 13:29:58 -06:00
|
|
|
for k in list(row.keys()):
|
2018-11-08 19:45:21 +00:00
|
|
|
if k not in self.fields:
|
|
|
|
del row[k]
|
|
|
|
|
|
|
|
# determine whether or not read perms for this object need
|
|
|
|
# to be explicitly set
|
|
|
|
if explicit and callable(explicit):
|
|
|
|
expl = explicit(row)
|
|
|
|
else:
|
|
|
|
expl = False
|
|
|
|
|
|
|
|
# initial read perms check
|
|
|
|
if nsp.has_perms(self.request.user, ns, 0x01, explicit=expl):
|
|
|
|
ruleset = getattr(inst, "nsp_ruleset", {})
|
|
|
|
|
|
|
|
# apply permissions to tree
|
|
|
|
row = self.apply_permissions(ns, row, ruleset=ruleset)
|
|
|
|
|
2020-07-15 02:07:01 -05:00
|
|
|
applicator = getattr(
|
|
|
|
self.model, "api_cache_permissions_applicator", None
|
|
|
|
)
|
|
|
|
|
|
|
|
if applicator:
|
|
|
|
applicator(row, ns, self.request.user)
|
|
|
|
|
2018-11-08 19:45:21 +00:00
|
|
|
# if row still has data aftewards, append to results
|
|
|
|
if row:
|
|
|
|
rv.append(row)
|
|
|
|
|
|
|
|
return rv
|
|
|
|
|
|
|
|
def join_ids(self, data, target_id, proxy_id, model, stash=[]):
|
|
|
|
"""
|
|
|
|
Returns a dict mapping of (proxy_id, target_id)
|
|
|
|
|
|
|
|
target ids are obtained by fetching instances of specified
|
|
|
|
model that match the supplied proxy ids
|
|
|
|
|
|
|
|
proxy ids will be gotten from data or stash
|
|
|
|
|
|
|
|
data [<dict>, ..] list of data rows from cache load, the field
|
|
|
|
name provided in "proxy_id" will be used to obtain the id from
|
|
|
|
each row
|
|
|
|
|
|
|
|
stash [<int>,..] list of ids
|
|
|
|
|
|
|
|
if stash is set, data and proxy_field will be ignored
|
|
|
|
"""
|
|
|
|
|
|
|
|
if stash:
|
|
|
|
ids = stash
|
|
|
|
else:
|
|
|
|
ids = [r[proxy_id] for r in data]
|
|
|
|
|
2020-07-15 02:07:01 -05:00
|
|
|
return {
|
2020-07-26 23:36:27 -05:00
|
|
|
r["id"]: r[target_id]
|
|
|
|
for r in model.objects.filter(id__in=ids).values("id", target_id)
|
2020-07-15 02:07:01 -05:00
|
|
|
}
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
# permissioning functions for each handlref type
|
|
|
|
|
|
|
|
def apply_permissions_org(self, data):
|
|
|
|
return self.apply_permissions_generic(data, id="id")
|
|
|
|
|
|
|
|
def apply_permissions_fac(self, data):
|
2019-12-05 16:57:52 +00:00
|
|
|
return self.apply_permissions_generic(data, fac_id="id", org_id="org_id")
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions_ix(self, data):
|
2019-12-05 16:57:52 +00:00
|
|
|
return self.apply_permissions_generic(data, ix_id="id", org_id="org_id")
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions_net(self, data):
|
2019-12-05 16:57:52 +00:00
|
|
|
return self.apply_permissions_generic(data, net_id="id", org_id="org_id")
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions_ixpfx(self, data):
|
|
|
|
return self.apply_permissions_generic(
|
2019-12-05 16:57:52 +00:00
|
|
|
data,
|
|
|
|
join_ids=[
|
|
|
|
("ix_id", "ixlan_id", IXLan),
|
|
|
|
("org_id", "ix_id", InternetExchange),
|
|
|
|
],
|
|
|
|
ixlan_id="ixlan_id",
|
|
|
|
id="id",
|
|
|
|
)
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions_ixlan(self, data):
|
|
|
|
return self.apply_permissions_generic(
|
2019-12-05 16:57:52 +00:00
|
|
|
data,
|
|
|
|
join_ids=[("org_id", "ix_id", InternetExchange)],
|
|
|
|
ix_id="ix_id",
|
|
|
|
id="id",
|
|
|
|
)
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions_ixfac(self, data):
|
|
|
|
return self.apply_permissions_generic(
|
2019-12-05 16:57:52 +00:00
|
|
|
data,
|
|
|
|
join_ids=[("org_id", "ix_id", InternetExchange)],
|
|
|
|
ix_id="ix_id",
|
|
|
|
id="id",
|
|
|
|
)
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions_netfac(self, data):
|
|
|
|
return self.apply_permissions_generic(
|
2019-12-05 16:57:52 +00:00
|
|
|
data,
|
|
|
|
join_ids=[("org_id", "net_id", Network)],
|
|
|
|
net_id="net_id",
|
|
|
|
fac_id="fac_id",
|
|
|
|
)
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions_netixlan(self, data):
|
|
|
|
return self.apply_permissions_generic(
|
2019-12-05 16:57:52 +00:00
|
|
|
data,
|
|
|
|
join_ids=[("org_id", "net_id", Network)],
|
|
|
|
net_id="net_id",
|
|
|
|
ixlan_id="ixlan_id",
|
|
|
|
)
|
2018-11-08 19:45:21 +00:00
|
|
|
|
|
|
|
def apply_permissions_poc(self, data):
|
|
|
|
return self.apply_permissions_generic(
|
2019-12-05 16:57:52 +00:00
|
|
|
data,
|
|
|
|
explicit=lambda x: (x.get("visible") != "Public"),
|
|
|
|
join_ids=[("org_id", "net_id", Network)],
|
|
|
|
vis="visible",
|
|
|
|
net_id="net_id",
|
|
|
|
)
|