diff options
author | Luke Shumaker <lukeshu@lukeshu.com> | 2023-07-23 00:46:42 -0600 |
---|---|---|
committer | Luke Shumaker <lukeshu@lukeshu.com> | 2023-07-23 00:46:42 -0600 |
commit | c57cc3d4739ffa0346350c63fb4ec59a63b56e46 (patch) | |
tree | 8d22d276e869e5b816e3113def16b3defcbd538c /lib/btrfs/btrfssum/shortsum.go | |
parent | 77628ce11ce3693d8ac06f1a404a1005ba05f190 (diff) | |
parent | 6e104326f81ec59ece1817988af41b70e4f4cd15 (diff) |
Merge branch 'lukeshu/json'
Diffstat (limited to 'lib/btrfs/btrfssum/shortsum.go')
-rw-r--r-- | lib/btrfs/btrfssum/shortsum.go | 68 |
1 files changed, 5 insertions, 63 deletions
diff --git a/lib/btrfs/btrfssum/shortsum.go b/lib/btrfs/btrfssum/shortsum.go index 754a79d..a3a6d11 100644 --- a/lib/btrfs/btrfssum/shortsum.go +++ b/lib/btrfs/btrfssum/shortsum.go @@ -5,12 +5,13 @@ package btrfssum import ( - "fmt" "io" - "math" "strings" "git.lukeshu.com/go/lowmemjson" + + "git.lukeshu.com/btrfs-progs-ng/lib/jsonutil" + "git.lukeshu.com/btrfs-progs-ng/lib/textui" ) type ShortSum string @@ -27,72 +28,13 @@ func (sum ShortSum) ToFullSum() CSum { } func (sum ShortSum) EncodeJSON(w io.Writer) error { - const hextable = "0123456789abcdef" - var buf [2]byte - buf[0] = '"' - if _, err := w.Write(buf[:1]); err != nil { - return err - } - for i := 0; i < len(sum); i++ { - buf[0] = hextable[sum[i]>>4] - buf[1] = hextable[sum[i]&0x0f] - if _, err := w.Write(buf[:]); err != nil { - return err - } - } - buf[0] = '"' - if _, err := w.Write(buf[:1]); err != nil { - return err - } - return nil -} - -func deHex(r rune) (byte, bool) { - if r > math.MaxUint8 { - return 0, false - } - c := byte(r) - //nolint:gomnd // Hex conversion. - switch { - case '0' <= c && c <= '9': - return c - '0', true - case 'a' <= c && c <= 'f': - return c - 'a' + 10, true - case 'A' <= c && c <= 'F': - return c - 'A' + 10, true - default: - return 0, false - } + return jsonutil.EncodeSplitHexString(w, sum, textui.Tunable(80)) } func (sum *ShortSum) DecodeJSON(r io.RuneScanner) error { var out strings.Builder - if c, _, err := r.ReadRune(); err != nil { + if err := jsonutil.DecodeSplitHexString(r, &out); err != nil { return err - } else if c != '"' { - return fmt.Errorf("expected %q, got %q", '"', c) - } - for { - a, _, err := r.ReadRune() - if err != nil { - return err - } - if a == '"' { - break - } - aN, ok := deHex(a) - if !ok { - return fmt.Errorf("expected a hex digit, got %q", a) - } - b, _, err := r.ReadRune() - if err != nil { - return err - } - bN, ok := deHex(b) - if !ok { - return fmt.Errorf("expected a hex digit, got %q", b) - } - out.WriteByte(aN<<4 | bN) } *sum = ShortSum(out.String()) return nil |