def Run(self, arg): """Returns the client stats.""" if arg is None: arg = rdf_client_action.GetClientStatsRequest() proc = psutil.Process(os.getpid()) meminfo = proc.memory_info() boot_time = rdfvalue.RDFDatetime.FromSecondsSinceEpoch( psutil.boot_time()) create_time = rdfvalue.RDFDatetime.FromSecondsSinceEpoch( proc.create_time()) response = rdf_client_stats.ClientStats( RSS_size=meminfo.rss, VMS_size=meminfo.vms, memory_percent=proc.memory_percent(), bytes_received=communicator.GRR_CLIENT_RECEIVED_BYTES.GetValue(), bytes_sent=communicator.GRR_CLIENT_SENT_BYTES.GetValue(), create_time=create_time, boot_time=boot_time) response.cpu_samples = self.grr_worker.stats_collector.CpuSamplesBetween( start_time=arg.start_time, end_time=arg.end_time) response.io_samples = self.grr_worker.stats_collector.IOSamplesBetween( start_time=arg.start_time, end_time=arg.end_time) self.Send(response)
def testReturnsAllDataByDefault(self): """Checks that stats collection works.""" communicator.GRR_CLIENT_RECEIVED_BYTES.Increment(1566) communicator.GRR_CLIENT_SENT_BYTES.Increment(2000) results = self.RunAction( admin.GetClientStats, grr_worker=MockClientWorker(), arg=rdf_client_action.GetClientStatsRequest()) response = results[0] self.assertEqual(response.bytes_received, 1566) self.assertEqual(response.bytes_sent, 2000) self.assertLen(response.cpu_samples, 3) for i in range(3): self.assertEqual(response.cpu_samples[i].timestamp, rdfvalue.RDFDatetime.FromSecondsSinceEpoch(100 + i * 10)) self.assertAlmostEqual(response.cpu_samples[i].user_cpu_time, 0.1) self.assertAlmostEqual(response.cpu_samples[i].system_cpu_time, 0.1 * (i + 1)) self.assertAlmostEqual(response.cpu_samples[i].cpu_percent, 10.0 + 5 * i) self.assertLen(response.io_samples, 3) for i in range(3): self.assertEqual(response.io_samples[i].timestamp, rdfvalue.RDFDatetime.FromSecondsSinceEpoch(100 + i * 10)) self.assertEqual(response.io_samples[i].read_bytes, 100 * (i + 1)) self.assertEqual(response.io_samples[i].write_bytes, 100 * (i + 1)) self.assertEqual(response.boot_time, 100 * 1e6)
def testFiltersDataPointsByEndTime(self): end_time = rdfvalue.RDFDatetime.FromSecondsSinceEpoch(102) results = self.RunAction( admin.GetClientStats, grr_worker=MockClientWorker(), arg=rdf_client_action.GetClientStatsRequest(end_time=end_time)) response = results[0] self.assertLen(response.cpu_samples, 1) self.assertEqual(response.cpu_samples[0].timestamp, rdfvalue.RDFDatetime.FromSecondsSinceEpoch(100)) self.assertLen(response.io_samples, 1) self.assertEqual(response.io_samples[0].timestamp, rdfvalue.RDFDatetime.FromSecondsSinceEpoch(100))
def _Send(self): if not self._ShouldSend(): return # TODO(hanuszczak): We shouldn't manually create action instances. Instead, # we should refactor action code to some other function and make the action # class use that function. Then here we should use that function as well. # # Also, it looks like there is a very weird dependency triangle: the worker # creates stat collector (which requires a worker), then the stats action # requires a worker and uses stat collector internally. But this action is # spawned by the stat collector. What...? action = admin.GetClientStatsAuto(grr_worker=self._worker) request = rdf_client_action.GetClientStatsRequest( start_time=self._last_send_time) action.Run(request) self._should_send = False self._last_send_time = rdfvalue.RDFDatetime.Now()
def testReturnsAllDataByDefault(self): """Checks that stats collection works.""" stats.STATS.RegisterCounterMetric("grr_client_received_bytes") stats.STATS.IncrementCounter("grr_client_received_bytes", 1566) stats.STATS.RegisterCounterMetric("grr_client_sent_bytes") stats.STATS.IncrementCounter("grr_client_sent_bytes", 2000) results = self.RunAction(admin.GetClientStats, grr_worker=MockClientWorker(), arg=rdf_client_action.GetClientStatsRequest()) response = results[0] self.assertEqual(response.bytes_received, 1566) self.assertEqual(response.bytes_sent, 2000) self.assertEqual(len(response.cpu_samples), 3) for i in range(3): self.assertEqual( response.cpu_samples[i].timestamp, rdfvalue.RDFDatetime.FromSecondsSinceEpoch(100 + i * 10)) self.assertAlmostEqual(response.cpu_samples[i].user_cpu_time, 0.1) self.assertAlmostEqual(response.cpu_samples[i].system_cpu_time, 0.1 * (i + 1)) self.assertAlmostEqual(response.cpu_samples[i].cpu_percent, 10.0 + 5 * i) self.assertEqual(len(response.io_samples), 3) for i in range(3): self.assertEqual( response.io_samples[i].timestamp, rdfvalue.RDFDatetime.FromSecondsSinceEpoch(100 + i * 10)) self.assertEqual(response.io_samples[i].read_bytes, 100 * (i + 1)) self.assertEqual(response.io_samples[i].write_bytes, 100 * (i + 1)) self.assertEqual(response.boot_time, 100 * 1e6)