path: root/pkgs/development/python-modules/telfhash/default.nix

{ lib
, buildPythonPackage
, fetchFromGitHub
, capstone
, pyelftools
, tlsh
, nose
}:
buildPythonPackage {
  pname = "telfhash";
  version = "unstable-2021-01-29";

  src = fetchFromGitHub {
    owner = "trendmicro";
    repo = "telfhash";
    rev = "b5e398e59dc25a56a28861751c1fccc74ef71617";
    sha256 = "jNu6qm8Q/UyJVaCqwFOPX02xAR5DwvCK3PaH6Fvmakk=";
  };

  patches = [ ./telfhash-new-tlsh-hash.patch ];

  # The tlsh library's name is just "tlsh"
  postPatch = ''
    substituteInPlace requirements.txt --replace "python-tlsh" "tlsh"
  '';

  propagatedBuildInputs = [
    capstone
    pyelftools
    tlsh
  ];

  checkInputs = [
    nose
  ];

  checkPhase = ''
    nosetests
  '';

  pythonImportsCheck = [
    "telfhash"
  ];

  meta = with lib; {
    description = "Symbol hash for ELF files";
    homepage = "https://github.com/trendmicro/telfhash";
    license = licenses.asl20;
    maintainers = teams.determinatesystems.members;
  };
}