Unverified Commit 54769c6f authored by Andrew Marshall's avatar Andrew Marshall Committed by GitHub
Browse files

hashcat: add CUDA support (#240735) 

Previously, hashcat was unable to use CUDA at runtime, and would warn:

> Failed to initialize the NVIDIA main driver CUDA runtime library.
> Failed to initialize NVIDIA RTC library.
> * Device #1: CUDA SDK Toolkit not installed or incorrectly installed.
>              CUDA SDK Toolkit required for proper device support and utilization.
>              Falling back to OpenCL runtime.

This remedies that, at least on NixOS.
parent e8f97f0f

pkgs/tools/security/hashcat/default.nix

+24 −3
Original line number Diff line number Diff line 
{ lib, stdenv

, addOpenGLRunpath

, config

, cudaPackages ? {}

, cudaSupport ? config.cudaSupport or false

, fetchurl

, makeWrapper

, opencl-headers
@@ -15,7 +19,12 @@ stdenv.mkDerivation rec { 
    sha256 = "sha256-sl4Qd7zzSQjMjxjBppouyYsEeyy88PURRNzzuh4Leyo=";

  };



  nativeBuildInputs = [ makeWrapper ];

  nativeBuildInputs = [

    makeWrapper

  ] ++ lib.optionals cudaSupport [

    addOpenGLRunpath

  ];



  buildInputs = [ opencl-headers xxHash ];



  makeFlags = [
@@ -34,8 +43,20 @@ stdenv.mkDerivation rec { 
    done

  '';



  postFixup = ''

    wrapProgram $out/bin/hashcat --prefix LD_LIBRARY_PATH : ${ocl-icd}/lib

  postFixup = let

    LD_LIBRARY_PATH = builtins.concatStringsSep ":" ([

      "${ocl-icd}/lib"

    ] ++ lib.optionals cudaSupport [

      "${cudaPackages.cudatoolkit}/lib"

    ]);

  in ''

    wrapProgram $out/bin/hashcat \

      --prefix LD_LIBRARY_PATH : ${lib.escapeShellArg LD_LIBRARY_PATH}

  '' + lib.optionalString cudaSupport ''

    for program in $out/bin/hashcat $out/bin/.hashcat-wrapped; do

      isELF "$program" || continue

      addOpenGLRunpath "$program"

    done

  '';



  meta = with lib; {