Use std::vector for charmap output

src/asm/charmap.cpp

@@ -139,17 +139,13 @@ bool charmap_HasChar(char const *input)
 	return charmap.nodes[nodeIdx].isTerminal;
 }
 
-size_t charmap_Convert(char const *input, uint8_t *output)
+void charmap_Convert(char const *input, std::vector<uint8_t> &output)
 {
-	uint8_t *start = output;
-
 	while (charmap_ConvertNext(&input, &output))
 		;
-
-	return output - start;
 }
 
-size_t charmap_ConvertNext(char const **input, uint8_t **output)
+size_t charmap_ConvertNext(char const **input, std::vector<uint8_t> *output)
 {
 	// The goal is to match the longest mapping possible.
 	// For that, advance through the trie with each character read.
@@ -181,22 +177,20 @@ size_t charmap_ConvertNext(char const **input, uint8_t **output)
 
 	if (matchIdx) { // A match was found, use it
 		if (output)
-			*(*output)++ = charmap.nodes[matchIdx].value;
+			output->push_back(charmap.nodes[matchIdx].value);
 
 		return 1;
 
 	} else if (**input) { // No match found, but there is some input left
 		int firstChar = **input;
 		// This will write the codepoint's value to `output`, little-endian
-		size_t codepointLen = readUTF8Char(output ? *output : NULL, *input);
+		size_t codepointLen = readUTF8Char(output, *input);
 
 		if (codepointLen == 0)
 			error("Input string is not valid UTF-8\n");
 
 		// OK because UTF-8 has no NUL in multi-byte chars
 		*input += codepointLen;
-		if (output)
-			*output += codepointLen;
 
 		// Warn if this character is not mapped but any others are
 		if (charmap.nodes.size() > 1)

src/asm/parser.y

@@ -46,8 +46,10 @@ static void lowerstring(char *dest, char const *src)
 	*dest = '\0';
 }
 
-static uint32_t str2int2(uint8_t *s, uint32_t length)
+static uint32_t str2int2(std::vector<uint8_t> const &s)
 {
+	uint32_t length = s.size();
+
 	if (length > 4)
 		warning(WARNING_NUMERIC_STRING_1,
 			"Treating string as a number ignores first %" PRIu32 " character%s\n",
@@ -1343,11 +1345,10 @@ constlist_8bit_entry : reloc_8bit_no_str {
 			sect_RelByte(&$1, 0);
 		}
 		| string {
-			uint8_t *output = (uint8_t *)malloc(strlen($1)); // Cannot be larger than that
-			size_t length = charmap_Convert($1, output);
+			std::vector<uint8_t> output;
 
-			sect_AbsByteGroup(output, length);
-			free(output);
+			charmap_Convert($1, output);
+			sect_AbsByteGroup(output.data(), output.size());
 		}
 ;
 
@@ -1359,11 +1360,10 @@ constlist_16bit_entry : reloc_16bit_no_str {
 			sect_RelWord(&$1, 0);
 		}
 		| string {
-			uint8_t *output = (uint8_t *)malloc(strlen($1)); // Cannot be larger than that
-			size_t length = charmap_Convert($1, output);
+			std::vector<uint8_t> output;
 
-			sect_AbsWordGroup(output, length);
-			free(output);
+			charmap_Convert($1, output);
+			sect_AbsWordGroup(output.data(), output.size());
 		}
 ;
 
@@ -1375,12 +1375,10 @@ constlist_32bit_entry : relocexpr_no_str {
 			sect_RelLong(&$1, 0);
 		}
 		| string {
-			// Charmaps cannot increase the length of a string
-			uint8_t *output = (uint8_t *)malloc(strlen($1));
-			size_t length = charmap_Convert($1, output);
+			std::vector<uint8_t> output;
 
-			sect_AbsLongGroup(output, length);
-			free(output);
+			charmap_Convert($1, output);
+			sect_AbsLongGroup(output.data(), output.size());
 		}
 ;
 
@@ -1421,13 +1419,10 @@ reloc_16bit_no_str : relocexpr_no_str {
 
 relocexpr	: relocexpr_no_str
 		| string {
-			// Charmaps cannot increase the length of a string
-			uint8_t *output = (uint8_t *)malloc(strlen($1));
-			uint32_t length = charmap_Convert($1, output);
-			uint32_t r = str2int2(output, length);
+			std::vector<uint8_t> output;
 
-			free(output);
-			rpn_Number(&$$, r);
+			charmap_Convert($1, output);
+			rpn_Number(&$$, str2int2(output));
 		}
 ;