Weights & Biases Detection

#%NASL_MIN_LEVEL 80900
##
# (C) Tenable, Inc.
##

include('compat.inc');

if (description)
{
  script_id(208133);
  script_version("1.6");
  script_set_attribute(attribute:"plugin_modification_date", value:"2026/03/18");

  script_name(english:"Weights & Biases Detection");

  script_set_attribute(attribute:"synopsis", value:
"A Weights & Biases Python library is installed on the remote host.");
  script_set_attribute(attribute:"description", value:
"A Weights & Biases Python library is installed on the remote host. 

Note that Nessus has relied upon on the application's self-reported version
number.");
  script_set_attribute(attribute:"see_also", value:"https://github.com/wandb/wandb");
  script_set_attribute(attribute:"solution", value: "n/a");
  script_set_attribute(attribute:"risk_factor", value:"None");

  script_set_attribute(attribute:"plugin_publication_date", value:"2024/10/04");

  script_set_attribute(attribute:"plugin_type", value:"local");
  script_set_attribute(attribute:"cpe", value:"cpe:/a:wandb:wandb");
  script_set_attribute(attribute:"asset_inventory", value:"True");
  script_set_attribute(attribute:"asset_inventory_category", value:"software_enumeration");
  script_set_attribute(attribute:"asset_categories", value:"component");
  script_set_attribute(attribute:"thorough_tests", value:"true");

  script_set_attribute(attribute:"agent", value:"all");
  script_end_attributes();

  script_category(ACT_GATHER_INFO);
  script_family(english:"Artificial Intelligence");

  script_copyright(english:"This script is Copyright (C) 2024-2026 and is owned by Tenable, Inc. or an Affiliate thereof.");

  script_dependencies("os_fingerprint.nasl", "python_packages_installed_nix.nbin", "python_packages_win_installed.nbin");
  script_require_ports("Host/nix/Python/Packages/Enumerated", "Host/win/Python/Packages/Enumerated");

  exit(0);
}

include('python.inc');

var host_os = get_kb_item_or_exit('Host/OS');
var os = NULL;

if('windows' >< tolower(host_os))
{
  os = 'win';
  get_kb_item_or_exit("Host/win/Python/Packages/Enumerated");
}
else 
{
  os = 'nix';
  get_kb_item_or_exit("Host/nix/Python/Packages/Enumerated");
}
var app_name = 'WandB';

var found_lib = python::query_python_package(os:os, pkg_name:app_name);

if (empty_or_null(found_lib))
  audit(AUDIT_NOT_DETECT, app_name);

var extra_no_report,extra,extra_info;

foreach (var found in found_lib)
{
  extra_no_report = {};
  extra = {};

  extra_info = python::get_extra_from_association(assoc_str:found.association);
  if (!empty_or_null(extra_info))
  {
    extra = extra_info[0];
    extra_no_report = extra_info[1];
  }

  register_install(
    app_name        : app_name,
    vendor          : app_name,
    product         : app_name,
    path            : found.path,
    version         : found.version,
    cpe             : 'cpe:/a:wandb:wandb',
    extra_no_report : extra_no_report,
    extra           : extra
  );

}

report_installs(app_name:app_name);