-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move NodePtr => Asn1Ptr into Asn1Decode.sol
- Loading branch information
Showing
5 changed files
with
282 additions
and
215 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,116 @@ | ||
| // SPDX-License-Identifier: MIT | ||
| pragma solidity ^0.8.15; | ||
|
|
||
| import {LibBytes} from "./LibBytes.sol"; | ||
|
|
||
| type CborElement is uint256; | ||
|
|
||
| library LibCborElement { | ||
| // Cbor element type | ||
| function cborType(CborElement self) internal pure returns (uint8) { | ||
| return uint8(CborElement.unwrap(self)); | ||
| } | ||
|
|
||
| // First byte index of the content | ||
| function start(CborElement self) internal pure returns (uint256) { | ||
| return uint80(CborElement.unwrap(self) >> 80); | ||
| } | ||
|
|
||
| // First byte index of the next element (exclusive end of content) | ||
| function end(CborElement self) internal pure returns (uint256) { | ||
| return start(self) + length(self); | ||
| } | ||
|
|
||
| // Content length (0 for non-string types) | ||
| function length(CborElement self) internal pure returns (uint256) { | ||
| uint8 _type = cborType(self); | ||
| if (_type == 0x40 || _type == 0x60) { | ||
| // length is non-zero only for byte strings and text strings | ||
| return value(self); | ||
| } | ||
| return 0; | ||
| } | ||
|
|
||
| // Value of the element (length for string/map/array types, value for others) | ||
| function value(CborElement self) internal pure returns (uint64) { | ||
| return uint64(CborElement.unwrap(self) >> 160); | ||
| } | ||
|
|
||
| // Returns true if the element is null | ||
| function isNull(CborElement self) internal pure returns (bool) { | ||
| return cborType(self) == 0xF6; | ||
| } | ||
|
|
||
| // Pack 3 uint80s into a uint256 | ||
| function toCborElement(uint256 _type, uint256 _start, uint256 _length) internal pure returns (CborElement) { | ||
| return CborElement.wrap(_type | _start << 80 | _length << 160); | ||
| } | ||
| } | ||
|
|
||
| library CborDecode { | ||
| using LibBytes for bytes; | ||
| using LibCborElement for CborElement; | ||
|
|
||
| // Calculate the keccak256 hash of the given cbor element | ||
| function keccak(bytes memory cbor, CborElement ptr) internal pure returns (bytes32) { | ||
| return cbor.keccak(ptr.start(), ptr.length()); | ||
| } | ||
|
|
||
| // Take a slice of the given cbor element | ||
| function slice(bytes memory cbor, CborElement ptr) internal pure returns (bytes memory) { | ||
| return cbor.slice(ptr.start(), ptr.length()); | ||
| } | ||
|
|
||
| function byteStringAt(bytes memory cbor, uint256 ix) internal pure returns (CborElement) { | ||
| return elementAt(cbor, ix, 0x40, true); | ||
| } | ||
|
|
||
| function nextByteString(bytes memory cbor, CborElement ptr) internal pure returns (CborElement) { | ||
| return elementAt(cbor, ptr.end(), 0x40, true); | ||
| } | ||
|
|
||
| function nextByteStringOrNull(bytes memory cbor, CborElement ptr) internal pure returns (CborElement) { | ||
| return elementAt(cbor, ptr.end(), 0x40, false); | ||
| } | ||
|
|
||
| function nextTextString(bytes memory cbor, CborElement ptr) internal pure returns (CborElement) { | ||
| return elementAt(cbor, ptr.end(), 0x60, true); | ||
| } | ||
|
|
||
| function nextPositiveInt(bytes memory cbor, CborElement ptr) internal pure returns (CborElement) { | ||
| return elementAt(cbor, ptr.end(), 0x00, true); | ||
| } | ||
|
|
||
| function mapAt(bytes memory cbor, uint256 ix) internal pure returns (CborElement) { | ||
| return elementAt(cbor, ix, 0xa0, true); | ||
| } | ||
|
|
||
| function nextMap(bytes memory cbor, CborElement ptr) internal pure returns (CborElement) { | ||
| return mapAt(cbor, ptr.end()); | ||
| } | ||
|
|
||
| function nextArray(bytes memory cbor, CborElement ptr) internal pure returns (CborElement) { | ||
| return elementAt(cbor, ptr.end(), 0x80, true); | ||
| } | ||
|
|
||
| function elementAt(bytes memory cbor, uint256 ix, uint8 expectedType, bool required) internal pure returns (CborElement) { | ||
| uint8 _type = uint8(cbor[ix] & 0xe0); | ||
| uint8 ai = uint8(cbor[ix] & 0x1f); | ||
| if (_type == 0xe0) { | ||
| require(!required || ai != 22, "null value for required element"); | ||
| // primitive type, retain the additional information | ||
| return LibCborElement.toCborElement(_type | ai, ix + 1, 0); | ||
| } | ||
| require(_type == expectedType, "unexpected type"); | ||
| if (ai == 24) { | ||
| return LibCborElement.toCborElement(_type, ix + 2, uint8(cbor[ix + 1])); | ||
| } else if (ai == 25) { | ||
| return LibCborElement.toCborElement(_type, ix + 3, cbor.readUint16(ix + 1)); | ||
| } else if (ai == 26) { | ||
| return LibCborElement.toCborElement(_type, ix + 5, cbor.readUint32(ix + 1)); | ||
| } else if (ai == 27) { | ||
| return LibCborElement.toCborElement(_type, ix + 9, cbor.readUint64(ix + 1)); | ||
| } | ||
| return LibCborElement.toCborElement(_type, ix + 1, ai); | ||
| } | ||
| } |
Oops, something went wrong.