Cobuild support

XuJiandong · Jan 15, 2024 · 29fb4dd · 29fb4dd
1 parent bcd9b58
commit 29fb4dd
Show file tree

Hide file tree

Showing 11 changed files with 599 additions and 96 deletions.
diff --git a/Makefile b/Makefile
@@ -3,7 +3,7 @@ CC := $(TARGET)-gcc
 LD := $(TARGET)-gcc
 OBJCOPY := $(TARGET)-objcopy
 CFLAGS := -fPIC -O3 -fno-builtin-printf -fno-builtin-memcmp -nostdinc -nostdlib -nostartfiles -fvisibility=hidden -fdata-sections -ffunction-sections -I deps/secp256k1/src -I deps/secp256k1 -I deps/ckb-c-std-lib -I deps/ckb-c-std-lib/libc -I deps/ckb-c-std-lib/molecule -I c -I build -Wall -Werror -Wno-nonnull -Wno-nonnull-compare -Wno-unused-function -g
-LDFLAGS := -Wl,-static -fdata-sections -ffunction-sections -Wl,--gc-sections
+LDFLAGS := -nostdlib -nostartfiles -fno-builtin -Wl,-static -Wl,--gc-sections
 SECP256K1_SRC_20210801 := deps/secp256k1-20210801/src/ecmult_static_pre_context.h
 
 
@@ -54,7 +54,8 @@ ${PROTOCOL_SCHEMA}:
 	curl -L -o $@ ${PROTOCOL_URL}
 
 ALL_C_SOURCE := $(wildcard c/omni_lock.c c/omni_lock_acp.h c/omni_lock_time_lock.h \
-	tests/omni_lock/omni_lock_sim.c tests/omni_lock/ckb_syscall_omni_lock_sim.h tests/omni_lock/omni_lock_supply.h)
+	tests/omni_lock/omni_lock_sim.c tests/omni_lock/ckb_syscall_omni_lock_sim.h tests/omni_lock/omni_lock_supply.h\
+	c/blake2b_decl_only.h c/cobuild.h c/cobuild.c)
 
 fmt:
 	docker run --rm -v `pwd`:/code ${CLANG_FORMAT_DOCKER} bash -c "cd code && clang-format -i -style=Google $(ALL_C_SOURCE)"
@@ -76,17 +77,23 @@ omni_lock_mol:
 	${MOLC} --language - --schema-file c/omni_lock.mol --format json > build/omni_lock_mol2.json
 	moleculec-c2 --input build/omni_lock_mol2.json | clang-format -style=Google > c/omni_lock_mol2.h
 
-build/omni_lock: c/omni_lock.c c/omni_lock_supply.h c/omni_lock_acp.h c/secp256k1_lock.h build/secp256k1_data_info_20210801.h $(SECP256K1_SRC_20210801) c/ckb_identity.h
-	$(CC) $(OMNI_LOCK_CFLAGS) $(LDFLAGS) -o $@ $<
+build/cobuild.o: c/cobuild.c c/cobuild.h
+	$(CC) -c $(OMNI_LOCK_CFLAGS) -o $@ $<
+
+build/omni_lock.o: c/omni_lock.c c/omni_lock_supply.h c/omni_lock_acp.h c/secp256k1_lock.h build/secp256k1_data_info_20210801.h $(SECP256K1_SRC_20210801) c/ckb_identity.h
+	$(CC) -c $(OMNI_LOCK_CFLAGS) -o $@ $<
+
+build/omni_lock: build/omni_lock.o build/cobuild.o 
+	$(CC) $(LDFLAGS) -o $@ $^
 	cp $@ $@.debug
 	$(OBJCOPY) --strip-debug --strip-all $@
 
-
 clean:
 	rm -rf build/secp256k1_data_info_20210801.h build/dump_secp256k1_data_20210801
 	rm -rf build/secp256k1_data_20210801
 	rm -rf build/*.debug
 	rm -f build/omni_lock
+	rm -f build/*.o
 	cd deps/secp256k1-20210801 && [ -f "Makefile" ] && make clean
 
 install-tools:

diff --git a/c/blake2b_decl_only.h b/c/blake2b_decl_only.h
@@ -0,0 +1,61 @@
+
+#ifndef __BLAKE2B_DECL_ONLY_H__
+#define __BLAKE2B_DECL_ONLY_H__
+
+#include <stddef.h>
+#include <stdint.h>
+
+#define BLAKE2_PACKED(x) x __attribute__((packed))
+
+enum blake2b_constant {
+  BLAKE2B_BLOCKBYTES = 128,
+  BLAKE2B_OUTBYTES = 64,
+  BLAKE2B_KEYBYTES = 64,
+  BLAKE2B_SALTBYTES = 16,
+  BLAKE2B_PERSONALBYTES = 16
+};
+BLAKE2_PACKED(struct blake2b_param__ {
+  uint8_t digest_length;                   /* 1 */
+  uint8_t key_length;                      /* 2 */
+  uint8_t fanout;                          /* 3 */
+  uint8_t depth;                           /* 4 */
+  uint32_t leaf_length;                    /* 8 */
+  uint32_t node_offset;                    /* 12 */
+  uint32_t xof_length;                     /* 16 */
+  uint8_t node_depth;                      /* 17 */
+  uint8_t inner_length;                    /* 18 */
+  uint8_t reserved[14];                    /* 32 */
+  uint8_t salt[BLAKE2B_SALTBYTES];         /* 48 */
+  uint8_t personal[BLAKE2B_PERSONALBYTES]; /* 64 */
+});
+
+typedef struct blake2b_param__ blake2b_param;
+
+typedef struct blake2b_state__ {
+  uint64_t h[8];
+  uint64_t t[2];
+  uint64_t f[2];
+  uint8_t buf[BLAKE2B_BLOCKBYTES];
+  size_t buflen;
+  size_t outlen;
+  uint8_t last_node;
+} blake2b_state;
+
+/* Streaming API */
+int ckb_blake2b_init(blake2b_state *S, size_t outlen);
+int blake2b_init(blake2b_state *S, size_t outlen);
+int blake2b_init_key(blake2b_state *S, size_t outlen, const void *key,
+                     size_t keylen);
+int blake2b_update(blake2b_state *S, const void *in, size_t inlen);
+int blake2b_final(blake2b_state *S, void *out, size_t outlen);
+/* Simple API */
+int blake2b(void *out, size_t outlen, const void *in, size_t inlen,
+            const void *key, size_t keylen);
+
+/* This is simply an alias for blake2b */
+int blake2(void *out, size_t outlen, const void *in, size_t inlen,
+           const void *key, size_t keylen);
+
+int blake2b_init_param(blake2b_state *S, const blake2b_param *P);
+
+#endif
diff --git a/c/ckb_identity.h b/c/ckb_identity.h
@@ -83,6 +83,9 @@ typedef int (*validate_signature_t)(void *prefilled_data, const uint8_t *sig,
 typedef int (*convert_msg_t)(const uint8_t *msg, size_t msg_len,
                              uint8_t *new_msg, size_t new_msg_len);
 
+bool g_cobuild_enabled = false;
+uint8_t g_cobuild_signing_message_hash[32];
+
 static void bin_to_hex(const uint8_t *source, uint8_t *dest, size_t len) {
   const static uint8_t HEX_TABLE[] = {'0', '1', '2', '3', '4', '5', '6', '7',
                                       '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
@@ -375,6 +378,11 @@ int validate_signature_eos(void *prefilled_data, const uint8_t *sig,
 }
 
 int generate_sighash_all(uint8_t *msg, size_t msg_len) {
+  if (g_cobuild_enabled) {
+    memcpy(msg, g_cobuild_signing_message_hash, BLAKE2B_BLOCK_SIZE);
+    return 0;
+  }
+
   int ret;
   uint64_t len = 0;
   unsigned char temp[MAX_WITNESS_SIZE];