4028431314

Build Type

github

Committed by GitHub

Commit Message

<a href="https://github.com/google/santa/commit/<a class=hub.com/google/santa/commit/<a class="double-link" href="https://git"><a class=hub.com/google/santa/commit/<a class="double-link" href="https://git"><a class=hub.com/google/santa/commit/<a class="double-link" href="https://git"><a class=hub.com/google/santa/commit/<a class="double-link" href="https://git"><a class=hub.com/google/santa/commit/eb93b36f5cd04a0aa09db6bf71470ad356d4aa2a">eb93b36f5">&lt;a href=&quot;https://github.com/google/santa/commit/</a><a class="double-link" href="https://github.com/google/santa/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/google/santa/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/google/santa/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/google/santa/commit/eb93b36f5cd04a0aa09db6bf71470ad356d4aa2a">eb93b36f5</a><a href="https://github.com/google/santa/commit/eb93b36f5cd04a0aa09db6bf71470ad356d4aa2a">&lt;a href=&quot;https://github.com/google/santa/commit/eb93b36f5cd04a0aa09db6bf71470ad356d4aa2a&quot;&gt;&amp;quot;&amp;gt;&amp;amp;lt;a href=&amp;amp;quot;https://github.com/google/santa/commit/&amp;lt;/a&amp;gt;&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://github.com/google/santa/commit/&amp;amp;lt;a class=&amp;amp;quot;double-link&amp;amp;quot; href=&amp;amp;quot;https://git&amp;quot;&amp;gt;&amp;amp;lt;a class=&amp;lt;/a&amp;gt;hub.com/google/santa/commit/eb93b36f5cd04a0aa09db6bf71470ad356d4aa2a&amp;quot;&amp;gt;eb93b36f5&amp;lt;/a&amp;gt;&amp;lt;a href=&amp;quot;https://github.com/google/santa/commit/eb93b36f5cd04a0aa09db6bf71470ad356d4aa2a&amp;quot;&amp;gt;&amp;amp;lt;a href=&amp;amp;quot;https://github.com/google/santa/commit/eb93b36f5cd04a0aa09db6bf71470ad3&amp;lt;/a&amp;gt;</a>56d4aa2a&quot;&gt;&amp;quot;&amp;gt;Merge &amp;lt;/a&amp;gt;&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://github.com/google/santa/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://github.com/google/santa/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://github.com/google/santa/commit/7e10b2784&quot;&gt;7e10b2784&lt;/a&gt;&lt;a href=&quot;https://github.com/google/santa/commit/eb93b36f5cd0... (continued)

Pull Request Pull Request #1019: Fix: Rewrite the SNTMetricHTTPWriter to avoid potential stack corruption

Run Details

3805 of 7740 branches covered (49.16%)

Branch coverage included in aggregate %.

19 of 19 new or added lines in 1 file covered. (100.0%)

13461 of 18946 relevant lines covered (71.05%)

18261.13 hits per line

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

83.13

/Source/santametricservice/Writers/SNTMetricHTTPWriter.m

/// Copyright 2021 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 <MOLAuthenticatingURLSession/MOLAuthenticatingURLSession.h>
#include <dispatch/dispatch.h>

#import "Source/common/SNTConfigurator.h"
#import "Source/common/SNTLogging.h"
#import "Source/santametricservice/Writers/SNTMetricHTTPWriter.h"

@implementation SNTMetricHTTPWriter {
 @private
  MOLAuthenticatingURLSession *_authSession;
}

- (instancetype)init {
  self = [super init];
  if (self) {
    _authSession = [[MOLAuthenticatingURLSession alloc] init];
  }
  return self;
}

/**
 * Post serialzied metrics to the specified URL one object at a time.
 **/
- (BOOL)write:(NSArray<NSData *> *)metrics toURL:(NSURL *)url error:(NSError **)error {
  __block NSError *_blockError = nil;

  NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
  request.HTTPMethod = @"POST";
  [request setValue:@"application/json" forHTTPHeaderField:@"Content-Type"];

  _authSession.serverHostname = url.host;
  NSURLSession *_session = _authSession.session;
  NSURLSessionDataTask *task;

  dispatch_group_t requests = dispatch_group_create();
  for (NSData *metric in metrics) {
    dispatch_group_enter(requests);

    request.HTTPBody = (NSData *)metric;
    task = [_session
      dataTaskWithRequest:request
        completionHandler:^(NSData *_Nullable data, NSURLResponse *_Nullable response,
                            NSError *_Nullable err) {
          if (err != nil) {
            LOGD(@"SNTMetricHTTPWriter: %@", err);
            _blockError = err;
          } else if (response == nil) {
            // Overwrite the last error as any previous error should show up in
            // logs if making multiple requests.
            _blockError = nil;
          } else if ([response isKindOfClass:[NSHTTPURLResponse class]]) {
            NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)response;

            // Check HTTP error codes and create errors for any non-200.
            if (httpResponse && httpResponse.statusCode != 200) {
              _blockError = [[NSError alloc]
                initWithDomain:@"com.google.santa.metricservice.writers.http"
                          code:httpResponse.statusCode
                      userInfo:@{
                        NSLocalizedDescriptionKey :
                          [NSString stringWithFormat:@"received http status code %ld from %@",
                                                     httpResponse.statusCode, url]
                      }];
            LOGD(@"SNTMetricHTTPWriter: %@", err);
            }
          }
          dispatch_group_leave(requests);
        }];

    [task resume];
  }

  SNTConfigurator *config = [SNTConfigurator configurator];
  int64_t timeout = (int64_t)config.metricExportTimeout;

  // Wait up to timeout seconds for the request to complete.
  if (dispatch_group_wait(requests, dispatch_time(DISPATCH_TIME_NOW, timeout * NSEC_PER_SEC)) !=
      0) {
    [task cancel];
    NSString *errMsg = [NSString stringWithFormat:@"HTTP request to %@ timed out after %lu seconds",
                                                  url, (unsigned long)timeout];

    _blockError = [[NSError alloc] initWithDomain:@"com.google.santa.metricservice.writers.http"
                                             code:ETIMEDOUT
                                         userInfo:@{NSLocalizedDescriptionKey : errMsg}];
  }

  if (_blockError != nil) {
    // If the caller hasn't passed us an error then we ignore it.
    if (error != nil) {
      *error = [_blockError copy];
    }

    return NO;
  }

  return YES;
}
@end

1	/// Copyright 2021 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	#import <MOLAuthenticatingURLSession/MOLAuthenticatingURLSession.h>
15	#include <dispatch/dispatch.h>
16
17	#import "Source/common/SNTConfigurator.h"
18	#import "Source/common/SNTLogging.h"
19	#import "Source/santametricservice/Writers/SNTMetricHTTPWriter.h"
20
21	@implementation SNTMetricHTTPWriter {
22	@private
23	MOLAuthenticatingURLSession *_authSession;
24	}
25
26	- (instancetype)init {	11✔
27	self = [super init];	11✔
28	if (self) {	11!
29	_authSession = [[MOLAuthenticatingURLSession alloc] init];	11✔
30	}	11✔
31	return self;	11✔
32	}	11✔
33
34	/**
35	* Post serialzied metrics to the specified URL one object at a time.
36	**/
37	- (BOOL)write:(NSArray<NSData > )metrics toURL:(NSURL )url error:(NSError *)error {	11✔
38	__block NSError *_blockError = nil;	11✔
39
40	NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];	11✔
41	request.HTTPMethod = @"POST";	11✔
42	[request setValue:@"application/json" forHTTPHeaderField:@"Content-Type"];	11✔
43
44	_authSession.serverHostname = url.host;	11✔
45	NSURLSession *_session = _authSession.session;	11✔
46	NSURLSessionDataTask *task;	11✔
47
48	dispatch_group_t requests = dispatch_group_create();	11✔
49	for (NSData *metric in metrics) {	12✔
50	dispatch_group_enter(requests);	12✔
51
52	request.HTTPBody = (NSData *)metric;	12✔
53	task = [_session	12✔
54	dataTaskWithRequest:request	12✔
55	completionHandler:^(NSData _Nullable data, NSURLResponse _Nullable response,	12✔
56	NSError *_Nullable err) {	12✔
57	if (err != nil) {	12✔
58	LOGD(@"SNTMetricHTTPWriter: %@", err);	3✔
59	_blockError = err;	3✔
60	} else if (response == nil) {	9!
61	// Overwrite the last error as any previous error should show up in
62	// logs if making multiple requests.
63	_blockError = nil;	×
64	} else if ([response isKindOfClass:[NSHTTPURLResponse class]]) {	9!
65	NSHTTPURLResponse httpResponse = (NSHTTPURLResponse )response;	9✔
66
67	// Check HTTP error codes and create errors for any non-200.
68	if (httpResponse && httpResponse.statusCode != 200) {	9!
69	_blockError = [[NSError alloc]	5✔
70	initWithDomain:@"com.google.santa.metricservice.writers.http"	5✔
71	code:httpResponse.statusCode	5✔
72	userInfo:@{	5✔
73	NSLocalizedDescriptionKey :	5✔
74	[NSString stringWithFormat:@"received http status code %ld from %@",	5✔
75	httpResponse.statusCode, url]	5✔
76	}];	5✔
77	LOGD(@"SNTMetricHTTPWriter: %@", err);	5✔
78	}	5✔
79	}	9✔
80	dispatch_group_leave(requests);	12✔
81	}];	12✔
82
83	[task resume];	12✔
84	}	12✔
85
86	SNTConfigurator *config = [SNTConfigurator configurator];	11✔
87	int64_t timeout = (int64_t)config.metricExportTimeout;	11✔
88
89	// Wait up to timeout seconds for the request to complete.
90	if (dispatch_group_wait(requests, dispatch_time(DISPATCH_TIME_NOW, timeout * NSEC_PER_SEC)) !=	11!
91	0) {	×
92	[task cancel];	×
93	NSString *errMsg = [NSString stringWithFormat:@"HTTP request to %@ timed out after %lu seconds",	×
94	url, (unsigned long)timeout];	×
95
96	_blockError = [[NSError alloc] initWithDomain:@"com.google.santa.metricservice.writers.http"	×
97	code:ETIMEDOUT	×
98	userInfo:@{NSLocalizedDescriptionKey : errMsg}];	×
99	}	×
100
101	if (_blockError != nil) {	11✔
102	// If the caller hasn't passed us an error then we ignore it.
103	if (error != nil) {	8✔
104	*error = [_blockError copy];	3✔
105	}	3✔
106
107	return NO;	8✔
108	}	8✔
109
110	return YES;	3✔
111	}	3✔
112	@end

google / santa / 4028431314

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