Feedback

google · Aug 9, 2024 · 11db5ea · 11db5ea
1 parent ea2b57f
commit 11db5ea
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/Source/common/SNTFileInfo.mm b/Source/common/SNTFileInfo.mm
@@ -164,7 +164,7 @@ - (instancetype)initWithPath:(NSString *)path {
 - (void)hashSHA1:(NSString **)sha1 SHA256:(NSString **)sha256 {
   const int MAX_CHUNK_SIZE = 256 * 1024;  // 256 KB
   const size_t chunkSize = _fileSize > MAX_CHUNK_SIZE ? MAX_CHUNK_SIZE : _fileSize;
-  char *chunk = (char *)malloc(chunkSize);
+  char *chunk = static_cast<char *>(malloc(chunkSize));
 
   @try {
     CC_SHA1_CTX c1;

diff --git a/Source/common/SNTRule.mm b/Source/common/SNTRule.mm
@@ -229,8 +229,8 @@ - (instancetype)initWithCoder:(NSCoder *)decoder {
   self = [super init];
   if (self) {
     _identifier = DECODE(NSString, @"identifier");
-    _state = (SNTRuleState)[DECODE(NSNumber, @"state") intValue];
-    _type = (SNTRuleType)[DECODE(NSNumber, @"type") intValue];
+    _state = static_cast<SNTRuleState>([DECODE(NSNumber, @"state") intValue]);
+    _type = static_cast<SNTRuleType>([DECODE(NSNumber, @"type") intValue]);
     _customMsg = DECODE(NSString, @"custommsg");
     _customURL = DECODE(NSString, @"customurl");
     _timestamp = [DECODE(NSNumber, @"timestamp") unsignedIntegerValue];
@@ -313,7 +313,9 @@ - (NSString *)digest {
   NSData *ruleData = [ruleDigestFormat dataUsingEncoding:NSUTF8StringEncoding];
 
   unsigned char digest[CC_SHA256_DIGEST_LENGTH];
-  CC_SHA256(ruleData.bytes, (CC_LONG)ruleData.length, (unsigned char *)&digest);
+  CC_SHA256(ruleData.bytes,
+            static_cast<CC_LONG>([ruleData lengthOfBytesUsingEncoding:NSUTF8StringEncoding]),
+            (unsigned char *)&digest);
 
   return santa::SHA256DigestToNSString(digest);
 }

diff --git a/Source/santad/DataLayer/SNTRuleTable.mm b/Source/santad/DataLayer/SNTRuleTable.mm
@@ -601,7 +601,9 @@ - (NSString *)hashOfHashes {
     while ([rs next]) {
       SNTRule *r = [self ruleFromResultSet:rs];
       NSString *digest = r.digest;
-      CC_SHA256_Update(&sha, digest.UTF8String, (CC_LONG)digest.length);
+      CC_SHA256_Update(
+        &sha, digest.UTF8String,
+        static_cast<CC_LONG>([digest lengthOfBytesUsingEncoding:NSUTF8StringEncoding]));
     }
     [rs close];
   }];