github.com/theQRL/go-zond@v0.2.1/common/bytes.go (about)

     1  // Copyright 2014 The go-ethereum Authors
     2  // This file is part of the go-ethereum library.
     3  //
     4  // The go-ethereum library is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Lesser General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // The go-ethereum library is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU Lesser General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Lesser General Public License
    15  // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  // Package common contains various helper functions.
    18  package common
    19  
    20  import (
    21  	"encoding/hex"
    22  	"errors"
    23  
    24  	"github.com/theQRL/go-zond/common/hexutil"
    25  )
    26  
    27  // FromHex returns the bytes represented by the hexadecimal string s.
    28  // s may be prefixed with "0x".
    29  func FromHex(s string) []byte {
    30  	if has0xPrefix(s) {
    31  		s = s[2:]
    32  	}
    33  	if len(s)%2 == 1 {
    34  		s = "0" + s
    35  	}
    36  	return Hex2Bytes(s)
    37  }
    38  
    39  // CopyBytes returns an exact copy of the provided bytes.
    40  func CopyBytes(b []byte) (copiedBytes []byte) {
    41  	if b == nil {
    42  		return nil
    43  	}
    44  	copiedBytes = make([]byte, len(b))
    45  	copy(copiedBytes, b)
    46  
    47  	return
    48  }
    49  
    50  // has0xPrefix validates str begins with '0x' or '0X'.
    51  func has0xPrefix(str string) bool {
    52  	return len(str) >= 2 && str[0] == '0' && (str[1] == 'x' || str[1] == 'X')
    53  }
    54  
    55  // hasZPrefix validates str begins with 'Z'.
    56  func hasZPrefix(str string) bool {
    57  	return len(str) >= 1 && str[0] == 'Z'
    58  }
    59  
    60  // isHexCharacter returns bool of c being a valid hexadecimal.
    61  func isHexCharacter(c byte) bool {
    62  	return ('0' <= c && c <= '9') || ('a' <= c && c <= 'f') || ('A' <= c && c <= 'F')
    63  }
    64  
    65  // isHex validates whether each byte is valid hexadecimal string.
    66  func isHex(str string) bool {
    67  	if len(str)%2 != 0 {
    68  		return false
    69  	}
    70  	for _, c := range []byte(str) {
    71  		if !isHexCharacter(c) {
    72  			return false
    73  		}
    74  	}
    75  	return true
    76  }
    77  
    78  // Bytes2Hex returns the hexadecimal encoding of d.
    79  func Bytes2Hex(d []byte) string {
    80  	return hex.EncodeToString(d)
    81  }
    82  
    83  // Hex2Bytes returns the bytes represented by the hexadecimal string str.
    84  func Hex2Bytes(str string) []byte {
    85  	h, _ := hex.DecodeString(str)
    86  	return h
    87  }
    88  
    89  // Hex2BytesFixed returns bytes of a specified fixed length flen.
    90  func Hex2BytesFixed(str string, flen int) []byte {
    91  	h, _ := hex.DecodeString(str)
    92  	if len(h) == flen {
    93  		return h
    94  	}
    95  	if len(h) > flen {
    96  		return h[len(h)-flen:]
    97  	}
    98  	hh := make([]byte, flen)
    99  	copy(hh[flen-len(h):flen], h)
   100  	return hh
   101  }
   102  
   103  // ParseHexOrString tries to hexdecode b, but if the prefix is missing, it instead just returns the raw bytes
   104  func ParseHexOrString(str string) ([]byte, error) {
   105  	b, err := hexutil.Decode(str)
   106  	if errors.Is(err, hexutil.ErrMissingPrefix) {
   107  		return []byte(str), nil
   108  	}
   109  	return b, err
   110  }
   111  
   112  // RightPadBytes zero-pads slice to the right up to length l.
   113  func RightPadBytes(slice []byte, l int) []byte {
   114  	if l <= len(slice) {
   115  		return slice
   116  	}
   117  
   118  	padded := make([]byte, l)
   119  	copy(padded, slice)
   120  
   121  	return padded
   122  }
   123  
   124  // LeftPadBytes zero-pads slice to the left up to length l.
   125  func LeftPadBytes(slice []byte, l int) []byte {
   126  	if l <= len(slice) {
   127  		return slice
   128  	}
   129  
   130  	padded := make([]byte, l)
   131  	copy(padded[l-len(slice):], slice)
   132  
   133  	return padded
   134  }
   135  
   136  // TrimLeftZeroes returns a subslice of s without leading zeroes
   137  func TrimLeftZeroes(s []byte) []byte {
   138  	idx := 0
   139  	for ; idx < len(s); idx++ {
   140  		if s[idx] != 0 {
   141  			break
   142  		}
   143  	}
   144  	return s[idx:]
   145  }
   146  
   147  // TrimRightZeroes returns a subslice of s without trailing zeroes
   148  func TrimRightZeroes(s []byte) []byte {
   149  	idx := len(s)
   150  	for ; idx > 0; idx-- {
   151  		if s[idx-1] != 0 {
   152  			break
   153  		}
   154  	}
   155  	return s[:idx]
   156  }