def create_public_redirection(ip_pub, port_pub, ip_priv, port_priv):
'''
This method requests a whole specific public IP to be redirected to a private IP
'''
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
ep_list, info = _ENDPOINT_MANAGER.request_endpoint(ip_pub, port_pub, ip_priv, port_priv)
if len(ep_list) == 0:
return cpyutils.restutils.error(404, "Could not obtain a redirection for %s:%s\n%s" % (ip_priv, port_priv, info))
# return False, "Could not obtain a redirection for %s:%d (%s)" % (ip_priv, port_priv, info)
if len(ep_list) == 1:
ep = ep_list[0]
result, msg = _ENDPOINT_MANAGER.apply_endpoint(ep)
if not result:
return cpyutils.restutils.error(501, "An error ocurred when registering endpoing %s\n%s" % (str(ep), msg))
else:
cpyutils.restutils.set_status(201)
cpyutils.restutils.add_header("Location", "/public/%s/%s" % (ep.public_ip, ep.public_port))
return cpyutils.restutils.response_json(ep.to_json())
else:
eps = {}
for ep in eplist:
eps[ep.public_port] = ep.to_json()
cpyutils.restutils.set_status(201)
return cpyutils.restutils.response_json(eps)
def get_private_redirections(ip):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
eps = _ENDPOINT_MANAGER.get_endpoints_from_private(True)
if ip in eps:
return cpyutils.restutils.response_json(eps[ip])
else:
return cpyutils.restutils.error(404, "IP %s not found" % ip)
def delete_public_redirection_i_p(ip_pub, port_pub):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
ep = _ENDPOINT_MANAGER.get_ep_from_public(ip_pub, port_pub)
if ep is None:
return cpyutils.restutils.error(404, "Could not clean the redirection from %s:%s. Does it exist?" % (ip_pub, port_pub))
return delete_public_redirection_i_p_i_p(ip_pub, port_pub, ep.private_ip, ep.private_port)
def get_public_redirections(ip):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
eps = _ENDPOINT_MANAGER.get_endpoints_from_public(True)
if ip in eps:
return cpyutils.restutils.response_json(eps[ip])
else:
# we'll return an empty list to indicate that there is not any redirection
# this is to enable semantics for the /redirect/ predicate to create redirections
return cpyutils.restutils.response_json({})
def delete_private_redirections(ip_priv):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
eps = _ENDPOINT_MANAGER.get_endpoints_from_private(False)
if ip_priv in eps:
for port_priv, ep in eps[ip_priv].items():
delete_private_redirection(ip_priv, port_priv)
return cpyutils.restutils.response_txt("")
else:
return cpyutils.restutils.error(404, "IP %s not found" % ip)
def get_public_redirections(ip_priv, port_priv):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
try:
port_priv = int(port_priv.strip("/"))
except:
return cpyutils.restutils.error(400, "Malformed request: port is an integer")
eps = _ENDPOINT_MANAGER.get_endpoints_from_private(True)
if (ip_priv in eps) and (port_priv in (eps[ip_priv])):
return cpyutils.restutils.response_json((eps[ip_priv])[port_priv])
else:
return cpyutils.restutils.error(404, "redirection not found")
def get_public_redirections(ip, port):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
try:
port = int(port.strip("/"))
except:
return cpyutils.restutils.error(400, "Malformed request: port must be an integer")
eps = _ENDPOINT_MANAGER.get_endpoints_from_public(True)
if (ip in eps) and (port in (eps[ip])):
return cpyutils.restutils.response_json((eps[ip])[port])
else:
# we'll return an empty list to indicate that there is not any redirection
# this is to enable semantics for the /redirect/ predicate to create redirections
return cpyutils.restutils.response_json({})
def delete_public_redirection_i(ip_pub):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
eps = _ENDPOINT_MANAGER.get_endpoints_from_public(False)
if not ip_pub in eps:
return cpyutils.restutils.error(404, "Could not clean the redirections from %s. Do they exist?" % (ip_pub))
if len(eps[ip_pub]) == 0:
return cpyutils.restutils.error(404, "Could not clean the redirections from %s. Do they exist?" % (ip_pub))
for port_pub, ep in eps[ip_pub].items():
delete_public_redirection_i_p_i_p(ip_pub, port_pub, ep.private_ip, ep.private_port)
return cpyutils.restutils.response_txt("")
def delete_private_redirection(ip_priv, port_priv):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
try:
port_priv = int(port_priv)
except:
return cpyutils.restutils.error(400, "Malformed request: port is an integer")
ep = _ENDPOINT_MANAGER.get_ep_from_private(ip_priv, port_priv)
if ep is None:
return cpyutils.restutils.error(404, "Redirection to %s:%s not found" % (ip_priv, port_priv))
result, msg = _ENDPOINT_MANAGER.terminate_endpoint(ep)
if not result:
return cpyutils.restutils.error(501, "Failed to terminate redirection to %s:%s (%s)" % (ip_priv, port_priv, msg))
return cpyutils.restutils.response_txt("")
def delete_public_redirection_i_p_i_p(ip_pub, port_pub, ip_priv, port_priv):
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
try:
port_pub = int(port_pub)
port_priv = int(port_priv)
except:
return cpyutils.restutils.error(400, "Malformed request: port must be an integer")
ep = _ENDPOINT_MANAGER.get_ep_from_public(ip_pub, port_pub)
if (ep is None) or (ep.private_ip != ip_priv) or (ep.private_port != port_priv):
return cpyutils.restutils.error(404, "Could not delete the redirection %s -> %s. Does it exist?" % (ip_pub, ip_priv))
result, msg = _ENDPOINT_MANAGER.terminate_endpoint(ep)
if not result:
return cpyutils.restutils.error(501, "Failed to terminate redirection %s -> %s (%s)" % (ip_pub, ip_priv, msg))
return cpyutils.restutils.response_txt("")
def get_public_redirections():
_ENDPOINT_MANAGER = ipfloaterd.get_endpoint_manager()
if _ENDPOINT_MANAGER is None:
return cpyutils.restutils.error(500, "Endpoint Manager not found")
return cpyutils.restutils.response_json(_ENDPOINT_MANAGER.get_endpoints_from_public(True))