Add support for byte type

ExcaliburZero · Nov 5, 2023 · 0dd61f3 · 0dd61f3
1 parent 281c8c8
commit 0dd61f3
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 1 deletion.
diff --git a/src/server/keyword_docs/byte.txt b/src/server/keyword_docs/byte.txt
@@ -0,0 +1,3 @@
+BYTE
+
+8 bit, 0 to 0xff
diff --git a/src/server/state.rs b/src/server/state.rs
@@ -464,6 +464,10 @@ pub fn get_keyword_docs() -> HashMap<String, String> {
             "getvarchr".to_string(),
             include_str!("keyword_docs/getvarchr.txt").to_string(),
         ),
+        (
+            "byte".to_string(),
+            include_str!("keyword_docs/byte.txt").to_string(),
+        ),
     ]
     .iter()
     .cloned()

diff --git a/tree-sitter-quickbms/grammar.js b/tree-sitter-quickbms/grammar.js
@@ -272,6 +272,7 @@ module.exports = grammar({
       $.string,
       $.binary,
       $.unicode,
+      $.byte,
     ),
     _endian_type: $ => choice(
       $.little,
@@ -304,6 +305,7 @@ module.exports = grammar({
     string: $ => /[Ss][Tt][Rr][Ii][Nn][Gg]/,
     binary: $ => /[Bb][Ii][Nn][Aa][Rr][Yy]/,
     unicode: $ => /[Uu][Nn][Ii][Cc][Oo][Dd][Ee]/,
+    byte: $ => /[Bb][Yy][Tt][Ee]/,
     endian: $ => /[Ee][Nn][Dd][Ii][Aa][Nn]/,
     little: $ => /[Ll][Ii][Tt][Tt][Ll][Ee]/,
     intel: $ => /[Ii][Nn][Tt][Ee][Ll]/,

diff --git a/tree-sitter-quickbms/test/corpus/getvarchr.bms b/tree-sitter-quickbms/test/corpus/getvarchr.bms
@@ -11,6 +11,8 @@ Next i
 GetVarChr TMP MEMORY_FILE i # hello
 string NAME - 4
 
+GetVarChr BYTESWAP MEMORY_FILE x byte
+
 --------------------------------------------------------------------------------
 
 (source_file
@@ -47,4 +49,11 @@ string NAME - 4
     (string)
     (identifier)
     (operation)
-    (integer_literal)))
+    (integer_literal))
+  (getvarchr_statement
+    (getvarchr)
+    (identifier)
+    (identifier)
+    (identifier)
+    (type
+      (byte))))