Switch to poetry build system (#687)

Author: SR4ven

.github/workflows/deploy.yml

@@ -17,11 +17,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install setuptools wheel twine
+        pip install poetry
     - name: Build and publish
-      env:
-        TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}
-        TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
       run: |
-        python setup.py sdist bdist_wheel
-        twine upload dist/*
+        poetry publish --build -u ${{ secrets.PYPI_USERNAME }} -p ${{ secrets.PYPI_PASSWORD }}

.gitignore

@@ -48,6 +48,7 @@ ENV/
 env.bak/
 venv.bak/
 Pipfile
+poetry.lock
 
 # Sphinx documentation
 docs/_build/

CHANGELOG.rst

@@ -16,11 +16,12 @@ Fixes
 - Specified encoding on file write rather than assuming default encoding
 - Changed type of `default_value` from string to bytes for `FromFile`.
 - `s_update` primitive was out of date.
-- The minimum supported Python version is now 3.7.
+- The minimum supported Python version is now 3.8.
 - Removed duplicates from `BitField` primitive.
 - Fixed unwanted deprecation warning when using `Session.fuzz(name=name)`.
 - Changed type of `dep_value` argument of `Block` to bytes and added type checks.
 - Split sessions.py into multiple files.
+- Using poetry as package build system.
 
 v0.4.1
 ------

CONTRIBUTING.rst

@@ -18,7 +18,7 @@ See installation instructions for details on installing boofuzz from source with
 Pull Request Checklist
 ----------------------
 
-1. Install python version 3.7+
+1. Install python version 3.8+
 
 2. Verify tests pass:
 
@@ -75,7 +75,7 @@ Prep
 
 2. Increment version number from last release according to PEP 0440 and roughly according to the Semantic Versioning guidelines.
 
-    1. In ``boofuzz/__init__.py``.
+    1. In ``pyproject.toml``.
 
     2. In ``docs/conf.py``.
 

INSTALL.rst

@@ -58,20 +58,38 @@ environment beforehand.
 
 From Source
 -----------
-
-
 1. Like above, it is recommended to set up a virtual environment. Depending on your
    concrete setup, this is largely equivalent to the steps outlined above. Make sure
-   to upgrade ``setuptools`` and ``pip``.
+   to upgrade ``setuptools`` and ``pip`` or ``poetry``.
 2. Download the source code. You can either grab a zip from https://github.com/jtpereyda/boofuzz
    or directly clone it with git:
 
    .. code-block:: bash
 
       $ git clone https://github.com/jtpereyda/boofuzz.git
 
-3. Install. Run ``pip`` from within the boofuzz directory after activating the virtual
-   environment:
+Install with Poetry
+~~~~~~~~~~~~~~~~~~~
+Poetry will automatically create a virtual environment for you and install the required dependencies. The installation
+will be editable by default, meaning that changes to the source code will be seen directly without reinstalling.
+
+Simply execute the following command inside the boofuzz source dir:
+
+   .. code-block:: bash
+
+       $ poetry install
+
+To install with extra dependencies like `dev` or `docs`, specify them in one of the following ways:
+
+   .. code-block:: bash
+
+       $ poetry install --extras "dev"
+       $ poetry install -E docs
+       $ poetry install --all-extras
+
+Install with Pip
+~~~~~~~~~~~~~~~~
+Run ``pip`` from within the boofuzz directory after activating the virtual environment:
 
    .. code-block:: bash
 

MANIFEST.in

@@ -187,8 +187,6 @@
     "Word",
 ]
 
-__version__ = "0.4.1"
-
 
 # REQUEST MANAGEMENT
 def s_get(name=None):

boofuzz/__init__.py

@@ -106,7 +106,7 @@ def __init__(
         self._current_num_mutations = 0
 
         self._format_raw_bytes = bytes_to_str
-        self._version = helpers.get_boofuzz_version(helpers)
+        self._version = helpers.get_boofuzz_version()
 
         # Resize console to minimum size
         self._width, self._height = get_terminal_size()

boofuzz/fuzz_logger_curses.py

@@ -1,4 +1,5 @@
 import errno
+import importlib.metadata
 import os
 import re
 import signal
@@ -424,22 +425,14 @@ def path_exists(path):
     return os.path.exists(path)
 
 
-def get_boofuzz_version(boofuzz_class):
+def get_boofuzz_version():
     """
-    Parses __init__.py for a version string and returns it like 'v0.0.0'
-
-    :type  boofuzz_class: class
-    :param boofuzz_class: Any boofuzz class in the same dir as the __init__ class.
+    Gets the currently installed boofuzz version
 
     :rtype: str
     :return: Boofuzz version as string
     """
-    path = os.path.dirname(boofuzz_class.__file__)
-    with open(os.path.join(path, "__init__.py")) as search:
-        for line in search:
-            if line.find("__version__ = ") != -1:
-                return "v" + re.search(r'"(.*?)"', line).group(1)  # pytype: disable=attribute-error
-    return "v-.-.-"
+    return "v" + importlib.metadata.version("boofuzz")
 
 
 def str_to_bytes(value, encoding="utf-8", errors="replace"):

boofuzz/helpers.py

@@ -1,2 +1,116 @@
+[tool.poetry]
+name = "boofuzz"
+version = "0.4.1"
+description = "A fork and successor of the Sulley Fuzzing Framework"
+authors = ["Joshua Pereyda <joshua.t.pereyda@gmail.com>"]
+license = "GPL-2.0-only"
+readme = ["README.rst", "CHANGELOG.rst"]
+repository = "https://github.com/jtpereyda/boofuzz"
+documentation = "https://boofuzz.readthedocs.io/"
+keywords = ["security", "fuzzing"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Environment :: Console",
+    "Environment :: Console :: Curses",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    "Natural Language :: English",
+    "Operating System :: OS Independent",
+    "Topic :: Security",
+    "Topic :: System :: Networking",
+    "Topic :: Software Development :: Testing :: Traffic Generation",
+]
+
+include = [
+    { path = "*.py", format = "sdist" },
+    { path = "*.rst", format = "sdist" },
+    { path = "*.toml", format = "sdist" },
+    { path = "*.txt", format = "sdist" },
+    { path = "tox.ini", format = "sdist" },
+    { path = "_static", format = "sdist" },
+    { path = "artwork", format = "sdist" },
+    { path = "docs", format = "sdist" },
+    { path = "examples", format = "sdist" },
+    { path = "request_definitions", format = "sdist" },
+    { path = "unit_tests", format = "sdist" },
+    { path = "utils", format = "sdist" },
+]
+
+[tool.poetry.dependencies]
+attrs = "*"
+click = "*"
+colorama = "*"
+Flask = "*"
+funcy = "*"
+psutil = "*"
+pydot = "*"
+pyserial = "*"
+python = "^3.8"
+tornado = "*"
+
+# dev extras
+black = { version = "*", optional = true }
+flake8 = { version = "*", optional = true }
+ipaddress = { version = "*", optional = true }
+mock = { version = "*", optional = true }
+netifaces = { version = "*", optional = true }
+pytest = { version = "*", optional = true }
+pytest-bdd = { version = "*", optional = true }
+pytest-cov = { version = "*", optional = true }
+tox = { version = "*", optional = true }
+wheel = { version = "*", optional = true }
+
+# docs extras
+pygments = { version = ">=2.4.0", optional = true }
+sphinx = { version = "*", optional = true }
+sphinx_rtd_theme = { version = "*", optional = true }
+
+[tool.poetry.extras]
+dev = [
+    "black",
+    "flake8",
+    "ipaddress",
+    "mock",
+    "netifaces",
+    "pygments",
+    "pytest",
+    "pytest-bdd",
+    "pytest-cov",
+    "sphinx",
+    "sphinx_rtd_theme",
+    "tox",
+    "wheel",
+]
+docs = [
+    "sphinx",
+    "sphinx_rtd_theme",
+    "pygments",
+]
+
+[tool.poetry.scripts]
+boo = 'boofuzz.cli:main'
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"
+
 [tool.black]
 line-length = 120
+
+[tool.pytest.ini_options]
+testpaths = ["unit_tests"]
+filterwarnings = [
+    "ignore:SocketConnection is deprecated:FutureWarning",
+]
+
+[tool.pytype]
+disable = [
+    "import-error",
+]
+exclude = [
+    "**/ida_fuzz_library_extender.py",
+    "examples/*.py",
+    "**/*_test_*.py",
+    "request_definitions/*.py",
+    "utils/*.py",
+]