Merge pull request 'codepoint: prevent panic when last cp too short' (#6) from rockorager/zg:master into master

Reviewed-on: https://codeberg.org/dude_the_builder/zg/pulls/6

1 files changed, 11 insertions, 0 deletions

diff --git a/src/code_point.zig b/src/code_point.zig
index ac37562..2f2e80f 100644
--- a/src/code_point.zig
+++ b/src/code_point.zig
@@ -46,6 +46,17 @@ pub const Iterator = struct {
             .offset = self.i,
         };
 
+        // Return replacement if we don' have a complete codepoint remaining. Consumes only one byte
+        if (self.i + cp.len > self.bytes.len) {
+            defer self.i += 1;
+            // Unicode replacement code point.
+            return .{
+                .code = 0xfffd,
+                .len = 1,
+                .offset = self.i,
+            };
+        }
+
         const cp_bytes = self.bytes[self.i..][0..cp.len];
         self.i += cp.len;