def Control(self):
"""Handle POSTS."""
if not master.MASTER_WATCHER.IsMaster():
# We shouldn't be getting requests from the client unless we
# are the active instance.
stats.STATS.IncrementCounter("frontend_inactive_request_count",
fields=["http"])
logging.info("Request sent to inactive frontend from %s",
self.client_address[0])
# Get the api version
try:
api_version = int(cgi.parse_qs(self.path.split("?")[1])["api"][0])
except (ValueError, KeyError, IndexError):
# The oldest api version we support if not specified.
api_version = 3
with GRRHTTPServerHandler.active_counter_lock:
GRRHTTPServerHandler.active_counter += 1
stats.STATS.SetGaugeValue("frontend_active_count",
self.active_counter,
fields=["http"])
try:
length = int(self.headers.getheader("content-length"))
request_comms = rdf_flows.ClientCommunication(self._GetPOSTData(length))
# If the client did not supply the version in the protobuf we use the get
# parameter.
if not request_comms.api_version:
request_comms.api_version = api_version
# Reply using the same version we were requested with.
responses_comms = rdf_flows.ClientCommunication(
api_version=request_comms.api_version)
source_ip = ipaddr.IPAddress(self.client_address[0])
if source_ip.version == 6:
source_ip = source_ip.ipv4_mapped or source_ip
request_comms.orig_request = rdf_flows.HttpRequest(
raw_headers=utils.SmartStr(self.headers),
source_ip=utils.SmartStr(source_ip))
source, nr_messages = self.server.frontend.HandleMessageBundles(
request_comms, responses_comms)
logging.info("HTTP request from %s (%s), %d bytes - %d messages received,"
" %d messages sent.", source, utils.SmartStr(source_ip),
length, nr_messages, responses_comms.num_messages)
self.Send(responses_comms.SerializeToString())
except communicator.UnknownClientCert:
# "406 Not Acceptable: The server can only generate a response that is not
# accepted by the client". This is because we can not encrypt for the
# client appropriately.
self.Send("Enrollment required", status=406)
except Exception as e: # pylint: disable=broad-except
if flags.FLAGS.debug:
pdb.post_mortem()
logging.error("Had to respond with status 500: %s.", e)
self.Send("Error", status=500)
finally:
with GRRHTTPServerHandler.active_counter_lock:
GRRHTTPServerHandler.active_counter -= 1
stats.STATS.SetGaugeValue("frontend_active_count",
self.active_counter,
fields=["http"])
def Control(self):
"""Handle POSTS."""
if not master.MASTER_WATCHER.IsMaster():
# We shouldn't be getting requests from the client unless we
# are the active instance.
stats.STATS.IncrementCounter("frontend_inactive_request_count",
fields=["http"])
logging.info("Request sent to inactive frontend from %s",
self.client_address[0])
# Get the api version
try:
api_version = int(cgi.parse_qs(self.path.split("?")[1])["api"][0])
except (ValueError, KeyError, IndexError):
# The oldest api version we support if not specified.
api_version = 3
with GRRHTTPServerHandler.active_counter_lock:
GRRHTTPServerHandler.active_counter += 1
stats.STATS.SetGaugeValue("frontend_active_count",
self.active_counter,
fields=["http"])
try:
content_length = self.headers.getheader("content-length")
if not content_length:
raise IOError("No content-length header provided.")
length = int(content_length)
request_comms = rdf_flows.ClientCommunication.FromSerializedString(
self._GetPOSTData(length))
# If the client did not supply the version in the protobuf we use the get
# parameter.
if not request_comms.api_version:
request_comms.api_version = api_version
# Reply using the same version we were requested with.
responses_comms = rdf_flows.ClientCommunication(
api_version=request_comms.api_version)
source_ip = ipaddr.IPAddress(self.client_address[0])
if source_ip.version == 6:
source_ip = source_ip.ipv4_mapped or source_ip
request_comms.orig_request = rdf_flows.HttpRequest(
timestamp=rdfvalue.RDFDatetime.Now().AsMicrosecondsSinceEpoch(
),
raw_headers=utils.SmartStr(self.headers),
source_ip=utils.SmartStr(source_ip))
source, nr_messages = self.server.frontend.HandleMessageBundles(
request_comms, responses_comms)
server_logging.LOGGER.LogHttpFrontendAccess(
request_comms.orig_request,
source=source,
message_count=nr_messages)
self.Send(responses_comms.SerializeToString())
except communicator.UnknownClientCert:
# "406 Not Acceptable: The server can only generate a response that is not
# accepted by the client". This is because we can not encrypt for the
# client appropriately.
self.Send("Enrollment required", status=406)
finally:
with GRRHTTPServerHandler.active_counter_lock:
GRRHTTPServerHandler.active_counter -= 1
stats.STATS.SetGaugeValue("frontend_active_count",
self.active_counter,
fields=["http"])
