NSData+SA_NSDataExtensions.m

   1 //
   2 //  NSData+SA_NSDataExtensions.m
   3 //
   4 //  Copyright (c) 2015 Said Achmiz.
   5 //
   6 //      Permission is hereby granted, free of charge, to any person obtaining a copy
   7 //      of this software and associated documentation files (the "Software"), to deal
   8 //      in the Software without restriction, including without limitation the rights
   9 //      to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10 //      copies of the Software, and to permit persons to whom the Software is
  11 //      furnished to do so, subject to the following conditions:
  12 //
  13 //      The above copyright notice and this permission notice shall be included in all
  14 //      copies or substantial portions of the Software.
  15 //
  16 //      THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17 //      IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18 //      FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19 //      AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20 //      LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21 //      OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  22 //      SOFTWARE.
  23
  24 #import "NSData+SA_NSDataExtensions.h"
  25
  26 @implementation NSData (SA_NSDataExtensions)
  27
  28 -(BOOL)isNullTerminated
  29 {
  30         return (((char*) self.bytes)[self.length - 1] == '\0');
  31 }
  32
  33 -(const char *)SA_terminatedCString
  34 {
  35         return self.SA_dataWithTerminatedCString.bytes;
  36 }
  37
  38 -(NSData *)SA_dataWithTerminatedCString
  39 {
  40         if(self.length == 0)
  41         {
  42                 return [NSData dataWithBytes:"\0" length:1];
  43         }
  44         else if(self.isNullTerminated)
  45         {
  46                 return self;
  47         }
  48         else
  49         {
  50                 char* terminated_string_buffer = malloc(self.length + 1);
  51                 [self getBytes:terminated_string_buffer length:self.length];
  52                 terminated_string_buffer[self.length] = '\0';
  53
  54                 return [NSData dataWithBytesNoCopy:terminated_string_buffer length:(self.length + 1) freeWhenDone:YES];
  55         }
  56 }
  57
  58 -(const char *)SA_unterminatedByteString
  59 {
  60         return self.SA_dataWithUnterminatedByteString.bytes;
  61 }
  62
  63 -(NSData *)SA_dataWithUnterminatedByteString
  64 {
  65         if(self.length == 0 || self.isNullTerminated == NO)
  66         {
  67                 return self;
  68         }
  69         else
  70         {
  71                 char* unterminated_string_buffer = malloc(self.length - 1);
  72                 [self getBytes:unterminated_string_buffer length:self.length - 1];
  73
  74                 return [NSData dataWithBytesNoCopy:unterminated_string_buffer length:(self.length - 1) freeWhenDone:YES];
  75         }
  76 }
  77
  78 @end