Add atomic store instructions to the parser

Author: MaxDesiatov

Sources/WAT/BinaryInstructionEncoder.swift

@@ -149,6 +149,13 @@ extension BinaryInstructionEncoder {
         case .i64Store8: opcode = [0x3C]
         case .i64Store16: opcode = [0x3D]
         case .i64Store32: opcode = [0x3E]
+        case .i32AtomicStore: opcode = [0xFE, 0x17]
+        case .i64AtomicStore: opcode = [0xFE, 0x18]
+        case .i32AtomicStore8: opcode = [0xFE, 0x19]
+        case .i32AtomicStore16: opcode = [0xFE, 0x1A]
+        case .i64AtomicStore8: opcode = [0xFE, 0x1B]
+        case .i64AtomicStore16: opcode = [0xFE, 0x1C]
+        case .i64AtomicStore32: opcode = [0xFE, 0x1D]
         }
 
         try encodeInstruction(opcode)

Sources/WAT/ParseTextInstruction.swift

@@ -354,6 +354,27 @@ func parseTextInstruction<V: InstructionVisitor>(keyword: String, expressionPars
     case "i64.atomic.load32_u":
         let (memarg) = try expressionParser.visitLoad(.i64AtomicLoad32U, wat: &wat)
         return { return try $0.visitLoad(.i64AtomicLoad32U, memarg: memarg) }
+    case "i32.atomic.store":
+        let (memarg) = try expressionParser.visitStore(.i32AtomicStore, wat: &wat)
+        return { return try $0.visitStore(.i32AtomicStore, memarg: memarg) }
+    case "i64.atomic.store":
+        let (memarg) = try expressionParser.visitStore(.i64AtomicStore, wat: &wat)
+        return { return try $0.visitStore(.i64AtomicStore, memarg: memarg) }
+    case "i32.atomic.store8":
+        let (memarg) = try expressionParser.visitStore(.i32AtomicStore8, wat: &wat)
+        return { return try $0.visitStore(.i32AtomicStore8, memarg: memarg) }
+    case "i32.atomic.store16":
+        let (memarg) = try expressionParser.visitStore(.i32AtomicStore16, wat: &wat)
+        return { return try $0.visitStore(.i32AtomicStore16, memarg: memarg) }
+    case "i64.atomic.store8":
+        let (memarg) = try expressionParser.visitStore(.i64AtomicStore8, wat: &wat)
+        return { return try $0.visitStore(.i64AtomicStore8, memarg: memarg) }
+    case "i64.atomic.store16":
+        let (memarg) = try expressionParser.visitStore(.i64AtomicStore16, wat: &wat)
+        return { return try $0.visitStore(.i64AtomicStore16, memarg: memarg) }
+    case "i64.atomic.store32":
+        let (memarg) = try expressionParser.visitStore(.i64AtomicStore32, wat: &wat)
+        return { return try $0.visitStore(.i64AtomicStore32, memarg: memarg) }
     default: return nil
     }
 }

Sources/WasmParser/BinaryInstructionDecoder.swift

@@ -600,6 +600,27 @@ func parseBinaryInstruction(visitor: inout some InstructionVisitor, decoder: ino
         case 0x16:
             let (memarg) = try decoder.visitLoad(.i64AtomicLoad32U)
             try visitor.visitLoad(.i64AtomicLoad32U, memarg: memarg)
+        case 0x17:
+            let (memarg) = try decoder.visitStore(.i32AtomicStore)
+            try visitor.visitStore(.i32AtomicStore, memarg: memarg)
+        case 0x18:
+            let (memarg) = try decoder.visitStore(.i64AtomicStore)
+            try visitor.visitStore(.i64AtomicStore, memarg: memarg)
+        case 0x19:
+            let (memarg) = try decoder.visitStore(.i32AtomicStore8)
+            try visitor.visitStore(.i32AtomicStore8, memarg: memarg)
+        case 0x1A:
+            let (memarg) = try decoder.visitStore(.i32AtomicStore16)
+            try visitor.visitStore(.i32AtomicStore16, memarg: memarg)
+        case 0x1B:
+            let (memarg) = try decoder.visitStore(.i64AtomicStore8)
+            try visitor.visitStore(.i64AtomicStore8, memarg: memarg)
+        case 0x1C:
+            let (memarg) = try decoder.visitStore(.i64AtomicStore16)
+            try visitor.visitStore(.i64AtomicStore16, memarg: memarg)
+        case 0x1D:
+            let (memarg) = try decoder.visitStore(.i64AtomicStore32)
+            try visitor.visitStore(.i64AtomicStore32, memarg: memarg)
         default:
             if try !visitor.visitUnknown([opcode0, opcode1]) { try decoder.throwUnknown([opcode0, opcode1]) }
         }

Sources/WasmParser/InstructionVisitor.swift

@@ -38,6 +38,13 @@ public enum Instruction: Equatable {
         case i64Store8
         case i64Store16
         case i64Store32
+        case i32AtomicStore
+        case i64AtomicStore
+        case i32AtomicStore8
+        case i32AtomicStore16
+        case i64AtomicStore8
+        case i64AtomicStore16
+        case i64AtomicStore32
     }
     public enum Cmp: Equatable {
         case i32Eq