예제 #1
0
파일: hunt.py 프로젝트: tkuennen/grr
  def __init__(self, data=None, context=None):
    super(HuntResult, self).__init__()
    self.data = data

    self.client = client.ClientRef(
        client_id=utils.UrnStringToClientId(data.client_id), context=context)
    self.timestamp = data.timestamp
예제 #2
0
파일: hunt.py 프로젝트: tkuennen/grr
  def __init__(self, data=None, context=None):
    super(HuntError, self).__init__()

    self.data = data
    self.log_message = self.data.log_message
    self.backtrace = self.data.backtrace

    self.client = client.ClientRef(
        client_id=utils.UrnStringToClientId(data.client_id), context=context)
예제 #3
0
    def __init__(
        self,
        data: hunt_pb2.ApiHuntResult,
        context: context_lib.GrrApiContext,
    ):
        super().__init__()
        self.data = data  # type: hunt_pb2.ApiHuntResult

        self.client = client.ClientRef(
            client_id=utils.UrnStringToClientId(data.client_id),
            context=context)  # type: client.ClientRef
        self.timestamp = data.timestamp  # type: int
예제 #4
0
    def __init__(
        self,
        data: hunt_pb2.ApiHuntError,
        context: context_lib.GrrApiContext,
    ):
        super().__init__()

        self.data = data  # type: hunt_pb2.ApiHuntError
        self.log_message = self.data.log_message  # type: str
        self.backtrace = self.data.backtrace  # type: str

        self.client = client.ClientRef(
            client_id=utils.UrnStringToClientId(data.client_id),
            context=context)  # type: client.ClientRef
예제 #5
0
    def __init__(
        self,
        data: hunt_pb2.ApiHuntLog,
        context: context_lib.GrrApiContext,
    ):
        super().__init__()

        self.data = data  # type: hunt_pb2.ApiHuntLog
        self.log_message = self.data.log_message  # str

        self.client = None  # type: Optional[client.ClientRef]
        if data.client_id:
            self.client = client.ClientRef(client_id=utils.UrnStringToClientId(
                data.client_id),
                                           context=context)
예제 #6
0
  last_clock: 1511174989272124
  age: 1510710503319681
  client_id: "C.0000000000000001"
""".format(
    int((datetime.datetime.utcnow() - datetime.timedelta(25)).timestamp() *
        1000000))

MOCK_CLIENT = client.Client(data=text_format.Parse(client_proto1,
                                                   client_pb2.ApiClient()),
                            context=True)
MOCK_CLIENT_RECENT = client.Client(data=text_format.Parse(
    client_proto2, client_pb2.ApiClient()),
                                   context=True)
MOCK_CLIENT_LIST = [MOCK_CLIENT, MOCK_CLIENT_RECENT]

MOCK_CLIENT_REF = client.ClientRef(MOCK_CLIENT.client_id, context=True)

flow_pb_terminated = flow_pb2.ApiFlow(urn="C.0000000000000001",
                                      flow_id="F:12345",
                                      state=flows_pb2.FlowContext.TERMINATED)
flow_pb_error = flow_pb2.ApiFlow(urn="C.0000000000000001",
                                 flow_id="F:12345",
                                 state=flows_pb2.FlowContext.ERROR)
flow_pb_running = flow_pb2.ApiFlow(urn="C.0000000000000001",
                                   flow_id="F:12345",
                                   state=flows_pb2.FlowContext.RUNNING)
MOCK_FLOW = flow.Flow(data=flow_pb_terminated, context=True)
MOCK_FLOW_ERROR = flow.Flow(data=flow_pb_error, context=True)
MOCK_FLOW_RUNNING = flow.Flow(data=flow_pb_running, context=True)

hunt_pb = hunt_pb2.ApiHunt(urn="hunts/12345", hunt_id="F:12345", state=1)
예제 #7
0
파일: api.py 프로젝트: skirankumar/grr
 def Client(self, client_id):
     return client.ClientRef(client_id=client_id, context=self._context)
예제 #8
0
 def Client(self, client_id: str) -> client.ClientRef:
     return client.ClientRef(client_id=client_id, context=self._context)