10322579337

Committed 09 Aug 2024 04:33PM UTC coverage: 63.129% (+0.4%) from 62.69%

Build # 10322579337

Build Type

Pull #1409

github

Committed by

web-flow

Commit Message

Merge c50360495 into 341abf044

Pull Request Pull Request #1409: sync: Send a hash of all database rules at beginning and end of sync

Run Details

5673 of 13542 branches covered (41.89%)

Branch coverage included in aggregate %.

87 of 89 new or added lines in 9 files covered. (97.75%)

13 existing lines in 2 files now uncovered.

18897 of 25378 relevant lines covered (74.46%)

5830.49 hits per line

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

71.43

/Source/santasyncservice/SNTSyncPostflight.mm

/// Copyright 2015 Google Inc. All rights reserved.
///
/// 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
///
///    http://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.

#import "Source/santasyncservice/SNTSyncPostflight.h"

#import <MOLXPCConnection/MOLXPCConnection.h>

#import "Source/common/SNTLogging.h"
#import "Source/common/SNTSyncConstants.h"
#import "Source/common/SNTXPCControlInterface.h"
#import "Source/common/String.h"
#import "Source/santasyncservice/SNTSyncLogging.h"
#import "Source/santasyncservice/SNTSyncState.h"

#include <google/protobuf/arena.h>
#include "Source/santasyncservice/syncv1.pb.h"
namespace pbv1 = ::santa::sync::v1;

using santa::NSStringToUTF8String;

@implementation SNTSyncPostflight

- (NSURL *)stageURL {
  NSString *stageName = [@"postflight" stringByAppendingFormat:@"/%@", self.syncState.machineID];
  return [NSURL URLWithString:stageName relativeToURL:self.syncState.syncBaseURL];
}

- (BOOL)sync {
  google::protobuf::Arena arena;
  auto req = google::protobuf::Arena::Create<::pbv1::PostflightRequest>(&arena);
  req->set_machine_id(NSStringToUTF8String(self.syncState.machineID));
  req->set_rules_received(self.syncState.rulesReceived);
  req->set_rules_processed(self.syncState.rulesProcessed);

  id<SNTDaemonControlXPC> rop = [self.daemonConn synchronousRemoteObjectProxy];
  [rop databaseRulesHash:^(NSString *hash) {
    req->set_rules_hash(NSStringToUTF8String(hash));
  }];

  ::pbv1::PostflightResponse response;
  [self performRequest:[self requestWithMessage:req] intoMessage:&response timeout:30];

  // Set client mode if it changed
  if (self.syncState.clientMode) {
    [rop setClientMode:self.syncState.clientMode
                 reply:^{
                 }];
  }

  // Remove clean sync flag if we did a clean or clean all sync
  if (self.syncState.syncType != SNTSyncTypeNormal) {
    [rop setSyncTypeRequired:SNTSyncTypeNormal
                       reply:^{
                       }];
  }

  // Update allowlist/blocklist regexes
  if (self.syncState.allowlistRegex) {
    [rop setAllowedPathRegex:self.syncState.allowlistRegex
                       reply:^{
                       }];
  }
  if (self.syncState.blocklistRegex) {
    [rop setBlockedPathRegex:self.syncState.blocklistRegex
                       reply:^{
                       }];
  }

  if (self.syncState.blockUSBMount != nil) {
    [rop setBlockUSBMount:[self.syncState.blockUSBMount boolValue]
                    reply:^{
                    }];
  }
  if (self.syncState.remountUSBMode) {
    [rop setRemountUSBMode:self.syncState.remountUSBMode
                     reply:^{
                     }];
  }

  if (self.syncState.enableBundles) {
    [rop setEnableBundles:[self.syncState.enableBundles boolValue]
                    reply:^{
                    }];
  }

  if (self.syncState.enableTransitiveRules) {
    [rop setEnableTransitiveRules:[self.syncState.enableTransitiveRules boolValue]
                            reply:^{
                            }];
  }

  if (self.syncState.enableAllEventUpload) {
    [rop setEnableAllEventUpload:[self.syncState.enableAllEventUpload boolValue]
                           reply:^{
                           }];
  }

  if (self.syncState.disableUnknownEventUpload) {
    [rop setDisableUnknownEventUpload:[self.syncState.disableUnknownEventUpload boolValue]
                                reply:^{
                                }];
  }

  if (self.syncState.overrideFileAccessAction) {
    [rop setOverrideFileAccessAction:self.syncState.overrideFileAccessAction
                               reply:^{
                               }];
  }

  // Update last sync success
  [rop setFullSyncLastSuccess:[NSDate date]
                        reply:^{
                        }];

  return YES;
}

@end

1	/// Copyright 2015 Google Inc. All rights reserved.
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	/// http://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	#import "Source/santasyncservice/SNTSyncPostflight.h"
16
17	#import <MOLXPCConnection/MOLXPCConnection.h>
18
19	#import "Source/common/SNTLogging.h"
20	#import "Source/common/SNTSyncConstants.h"
21	#import "Source/common/SNTXPCControlInterface.h"
22	#import "Source/common/String.h"
23	#import "Source/santasyncservice/SNTSyncLogging.h"
24	#import "Source/santasyncservice/SNTSyncState.h"
25
26	#include <google/protobuf/arena.h>
27	#include "Source/santasyncservice/syncv1.pb.h"
28	namespace pbv1 = ::santa::sync::v1;
29
30	using santa::NSStringToUTF8String;
31
32	@implementation SNTSyncPostflight
33
34	- (NSURL *)stageURL {	8✔
35	NSString *stageName = [@"postflight" stringByAppendingFormat:@"/%@", self.syncState.machineID];	8✔
36	return [NSURL URLWithString:stageName relativeToURL:self.syncState.syncBaseURL];	8✔
37	}	8✔
38
39	- (BOOL)sync {	8✔
40	google::protobuf::Arena arena;	8✔
41	auto req = google::protobuf::Arena::Create<::pbv1::PostflightRequest>(&arena);	8✔
42	req->set_machine_id(NSStringToUTF8String(self.syncState.machineID));	8✔
43	req->set_rules_received(self.syncState.rulesReceived);	8✔
44	req->set_rules_processed(self.syncState.rulesProcessed);	8✔
45
46	id<SNTDaemonControlXPC> rop = [self.daemonConn synchronousRemoteObjectProxy];	8✔
47	[rop databaseRulesHash:^(NSString *hash) {	8✔
48	req->set_rules_hash(NSStringToUTF8String(hash));	8✔
49	}];	8✔
50
51	::pbv1::PostflightResponse response;	8✔
52	[self performRequest:[self requestWithMessage:req] intoMessage:&response timeout:30];	8✔
53
54	// Set client mode if it changed
55	if (self.syncState.clientMode) {	8✔
56	[rop setClientMode:self.syncState.clientMode	7✔
57	reply:^{	7✔
UNCOV 58	}];	×
59	}	7✔
60
61	// Remove clean sync flag if we did a clean or clean all sync
62	if (self.syncState.syncType != SNTSyncTypeNormal) {	8✔
63	[rop setSyncTypeRequired:SNTSyncTypeNormal	6✔
64	reply:^{	6✔
UNCOV 65	}];	×
66	}	6✔
67
68	// Update allowlist/blocklist regexes
69	if (self.syncState.allowlistRegex) {	8✔
70	[rop setAllowedPathRegex:self.syncState.allowlistRegex	4✔
71	reply:^{	4✔
UNCOV 72	}];	×
73	}	4✔
74	if (self.syncState.blocklistRegex) {	8✔
75	[rop setBlockedPathRegex:self.syncState.blocklistRegex	4✔
76	reply:^{	4✔
UNCOV 77	}];	×
78	}	4✔
79
80	if (self.syncState.blockUSBMount != nil) {	8✔
81	[rop setBlockUSBMount:[self.syncState.blockUSBMount boolValue]	3✔
82	reply:^{	3✔
UNCOV 83	}];	×
84	}	3✔
85	if (self.syncState.remountUSBMode) {	8✔
86	[rop setRemountUSBMode:self.syncState.remountUSBMode	3✔
87	reply:^{	3✔
UNCOV 88	}];	×
89	}	3✔
90
91	if (self.syncState.enableBundles) {	8!
92	[rop setEnableBundles:[self.syncState.enableBundles boolValue]	×
93	reply:^{	×
94	}];	×
UNCOV 95	}	×
96
97	if (self.syncState.enableTransitiveRules) {	8!
98	[rop setEnableTransitiveRules:[self.syncState.enableTransitiveRules boolValue]	×
99	reply:^{	×
100	}];	×
UNCOV 101	}	×
102
103	if (self.syncState.enableAllEventUpload) {	8!
104	[rop setEnableAllEventUpload:[self.syncState.enableAllEventUpload boolValue]	×
105	reply:^{	×
106	}];	×
UNCOV 107	}	×
108
109	if (self.syncState.disableUnknownEventUpload) {	8!
110	[rop setDisableUnknownEventUpload:[self.syncState.disableUnknownEventUpload boolValue]	×
111	reply:^{	×
112	}];	×
UNCOV 113	}	×
114
115	if (self.syncState.overrideFileAccessAction) {	8✔
116	[rop setOverrideFileAccessAction:self.syncState.overrideFileAccessAction	1✔
117	reply:^{	1✔
UNCOV 118	}];	×
119	}	1✔
120
121	// Update last sync success
122	[rop setFullSyncLastSuccess:[NSDate date]	8✔
123	reply:^{	8✔
UNCOV 124	}];	×
125
126	return YES;	8✔
127	}	8✔
128
129	@end

google / santa / 10322579337

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