/*
* Copyright (c) 2008-2024 Jonathan Schleifer <js@nil.im>
*
* All rights reserved.
*
* This file is part of ObjFW. It may be distributed under the terms of the
* Q Public License 1.0, which can be found in the file LICENSE.QPL included in
* the packaging of this file.
*
* Alternatively, it may be distributed under the terms of the GNU General
* Public License, either version 2 or 3, which can be found in the file
* LICENSE.GPLv2 or LICENSE.GPLv3 respectively included in the packaging of this
* file.
*/
#include "config.h"
#include <stdlib.h>
#include <string.h>
#import "OFString.h"
#import "OFOutOfMemoryException.h"
int _OFString_XMLEscaping_reference;
@implementation OFString (XMLEscaping)
- (OFString *)stringByXMLEscaping
{
void *pool = objc_autoreleasePoolPush();
char *retCString;
const char *string, *append;
size_t length, retLength, appendLen;
size_t j;
OFString *ret;
string = self.UTF8String;
length = self.UTF8StringLength;
j = 0;
retLength = length;
retCString = OFAllocMemory(retLength, 1);
for (size_t i = 0; i < length; i++) {
switch (string[i]) {
case '<':
append = "<";
appendLen = 4;
break;
case '>':
append = ">";
appendLen = 4;
break;
case '"':
append = """;
appendLen = 6;
break;
case '\'':
append = "'";
appendLen = 6;
break;
case '&':
append = "&";
appendLen = 5;
break;
case '\r':
append = "
";
appendLen = 5;
break;
default:
append = NULL;
appendLen = 0;
}
if (append != NULL) {
@try {
retCString = OFResizeMemory(retCString, 1,
retLength + appendLen);
} @catch (id e) {
OFFreeMemory(retCString);
@throw e;
}
retLength += appendLen - 1;
memcpy(retCString + j, append, appendLen);
j += appendLen;
} else
retCString[j++] = string[i];
}
OFAssert(j == retLength);
objc_autoreleasePoolPop(pool);
@try {
ret = [OFString stringWithUTF8String: retCString
length: retLength];
} @finally {
OFFreeMemory(retCString);
}
return ret;
}
@end