29c269ac8d
only install 'noiseprotocol' (which is necessary for dilation to work) if the "dilate" feature is requested (e.g. `pip install magic-wormhole[dilate]`)
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
# No unicode_literals
|
|
import json
|
|
import os
|
|
import unicodedata
|
|
from binascii import hexlify, unhexlify
|
|
from hkdf import Hkdf
|
|
|
|
|
|
def HKDF(skm, outlen, salt=None, CTXinfo=b""):
|
|
return Hkdf(salt, skm).expand(CTXinfo, outlen)
|
|
|
|
def to_bytes(u):
|
|
return unicodedata.normalize("NFC", u).encode("utf-8")
|
|
|
|
def to_unicode(any):
|
|
if isinstance(any, type(u"")):
|
|
return any
|
|
return any.decode("ascii")
|
|
|
|
def bytes_to_hexstr(b):
|
|
assert isinstance(b, type(b""))
|
|
hexstr = hexlify(b).decode("ascii")
|
|
assert isinstance(hexstr, type(u""))
|
|
return hexstr
|
|
|
|
|
|
def hexstr_to_bytes(hexstr):
|
|
assert isinstance(hexstr, type(u""))
|
|
b = unhexlify(hexstr.encode("ascii"))
|
|
assert isinstance(b, type(b""))
|
|
return b
|
|
|
|
|
|
def dict_to_bytes(d):
|
|
assert isinstance(d, dict)
|
|
b = json.dumps(d).encode("utf-8")
|
|
assert isinstance(b, type(b""))
|
|
return b
|
|
|
|
|
|
def bytes_to_dict(b):
|
|
assert isinstance(b, type(b""))
|
|
d = json.loads(b.decode("utf-8"))
|
|
assert isinstance(d, dict)
|
|
return d
|
|
|
|
|
|
def estimate_free_space(target):
|
|
# f_bfree is the blocks available to a root user. It might be more
|
|
# accurate to use f_bavail (blocks available to non-root user), but we
|
|
# don't know which user is running us, and a lot of installations don't
|
|
# bother with reserving extra space for root, so let's just stick to the
|
|
# basic (larger) estimate.
|
|
try:
|
|
s = os.statvfs(os.path.dirname(os.path.abspath(target)))
|
|
return s.f_frsize * s.f_bfree
|
|
except AttributeError:
|
|
return None
|