11143619276

Committed 02 Oct 2024 12:18PM UTC coverage: 62.699% (-0.6%) from 63.269%

Build # 11143619276

Build Type

Pull #1428

github

Committed by

web-flow

Commit Message

Merge 8d48ad4af into 737525b74

Pull Request Pull Request #1428: Remove default root label values

Run Details

5606 of 13364 branches covered (41.95%)

Branch coverage included in aggregate %.

0 of 1 new or added line in 1 file covered. (0.0%)

109 existing lines in 4 files now uncovered.

18522 of 25118 relevant lines covered (73.74%)

6637.11 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

52.38

/Source/common/SystemResources.mm

/// Copyright 2022 Google LLC
///
/// Licensed under the Apache License, Version 2.0 (the "License");
/// you may not use this file except in compliance with the License.
/// You may obtain a copy of the License at
///
///     https://www.apache.org/licenses/LICENSE-2.0
///
/// Unless required by applicable law or agreed to in writing, software
/// distributed under the License is distributed on an "AS IS" BASIS,
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
/// See the License for the specific language governing permissions and
/// limitations under the License.

#include "Source/common/SystemResources.h"

#include <dispatch/dispatch.h>
#include <libproc.h>
#include <mach/kern_return.h>
#include <unistd.h>
#include <optional>

#include "Source/common/SNTLogging.h"

static mach_timebase_info_data_t GetTimebase() {
  static dispatch_once_t once_token;
  static mach_timebase_info_data_t timebase;

  dispatch_once(&once_token, ^{
    if (mach_timebase_info(&timebase) != KERN_SUCCESS) {
      // This shouldn't fail. Assume transitory and exit the program.
      // Hopefully fixes itself on restart...
      LOGE(@"Failed to get timebase info. Exiting.");
      exit(EXIT_FAILURE);
    }
  });

  return timebase;
}

uint64_t MachTimeToNanos(uint64_t mach_time) {
  static mach_timebase_info_data_t timebase = GetTimebase();

  return mach_time * timebase.numer / timebase.denom;
}

uint64_t NanosToMachTime(uint64_t nanos) {
  static mach_timebase_info_data_t timebase = GetTimebase();

  return nanos * timebase.denom / timebase.numer;
}

uint64_t AddNanosecondsToMachTime(uint64_t ns, uint64_t machTime) {
  // Convert machtime to nanoseconds
  uint64_t nanoTime = MachTimeToNanos(machTime);

  // Add the nanosecond offset
  nanoTime += ns;

  // Convert back to machTime
  return NanosToMachTime(nanoTime);
}

std::optional<SantaTaskInfo> GetTaskInfo() {
  struct proc_taskinfo pti;

  if (proc_pidinfo(getpid(), PROC_PIDTASKINFO, 0, &pti, PROC_PIDTASKINFO_SIZE) <
      PROC_PIDTASKINFO_SIZE) {
    LOGW(@"Unable to get system resource information");
    return std::nullopt;
  }

  return SantaTaskInfo{
    .virtual_size = pti.pti_virtual_size,
    .resident_size = pti.pti_resident_size,
    .total_user_nanos = MachTimeToNanos(pti.pti_total_user),
    .total_system_nanos = MachTimeToNanos(pti.pti_total_system),
  };
}

1	/// Copyright 2022 Google LLC
2	///
3	/// Licensed under the Apache License, Version 2.0 (the "License");
4	/// you may not use this file except in compliance with the License.
5	/// You may obtain a copy of the License at
6	///
7	/// https://www.apache.org/licenses/LICENSE-2.0
8	///
9	/// Unless required by applicable law or agreed to in writing, software
10	/// distributed under the License is distributed on an "AS IS" BASIS,
11	/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12	/// See the License for the specific language governing permissions and
13	/// limitations under the License.
14
15	#include "Source/common/SystemResources.h"
16
17	#include <dispatch/dispatch.h>
18	#include <libproc.h>
19	#include <mach/kern_return.h>
20	#include <unistd.h>
21	#include <optional>
22
23	#include "Source/common/SNTLogging.h"
24
25	static mach_timebase_info_data_t GetTimebase() {	28✔
26	static dispatch_once_t once_token;	28✔
27	static mach_timebase_info_data_t timebase;	28✔
28
29	dispatch_once(&once_token, ^{	28✔
30	if (mach_timebase_info(&timebase) != KERN_SUCCESS) {	14!
31	// This shouldn't fail. Assume transitory and exit the program.
32	// Hopefully fixes itself on restart...
33	LOGE(@"Failed to get timebase info. Exiting.");	×
34	exit(EXIT_FAILURE);	×
35	}	×
36	});	14✔
37
38	return timebase;	28✔
39	}	28✔
40
41	uint64_t MachTimeToNanos(uint64_t mach_time) {	219✔
42	static mach_timebase_info_data_t timebase = GetTimebase();	219✔
43
44	return mach_time * timebase.numer / timebase.denom;	219✔
45	}	219✔
46
47	uint64_t NanosToMachTime(uint64_t nanos) {	188✔
48	static mach_timebase_info_data_t timebase = GetTimebase();	188✔
49
50	return nanos * timebase.denom / timebase.numer;	188✔
51	}	188✔
52
53	uint64_t AddNanosecondsToMachTime(uint64_t ns, uint64_t machTime) {	188✔
54	// Convert machtime to nanoseconds
55	uint64_t nanoTime = MachTimeToNanos(machTime);	188✔
56
57	// Add the nanosecond offset
58	nanoTime += ns;	188✔
59
60	// Convert back to machTime
61	return NanosToMachTime(nanoTime);	188✔
62	}	188✔
63
UNCOV 64	std::optional<SantaTaskInfo> GetTaskInfo() {	×
UNCOV 65	struct proc_taskinfo pti;	×
66
UNCOV 67	if (proc_pidinfo(getpid(), PROC_PIDTASKINFO, 0, &pti, PROC_PIDTASKINFO_SIZE) <	×
UNCOV 68	PROC_PIDTASKINFO_SIZE) {	×
69	LOGW(@"Unable to get system resource information");	×
70	return std::nullopt;	×
71	}	×
72
UNCOV 73	return SantaTaskInfo{	×
UNCOV 74	.virtual_size = pti.pti_virtual_size,	×
UNCOV 75	.resident_size = pti.pti_resident_size,	×
UNCOV 76	.total_user_nanos = MachTimeToNanos(pti.pti_total_user),	×
UNCOV 77	.total_system_nanos = MachTimeToNanos(pti.pti_total_system),	×
UNCOV 78	};	×
UNCOV 79	}	×

google / santa / 11143619276

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous