Merge pull request #7 from EGI-Federation/token

add support for bearer tokens

README.md

@@ -54,6 +54,8 @@ optional arguments:
   -E ENDPOINT, --endpoint ENDPOINT
                         Storage URL to test (Mandatory)
   -X X509, --x509 X509  location of x509 certificate proxy file
+  -to TOKEN, --token TOKEN
+                        BEARER TOKEN to be used
   --se-timeout SE_TIMEOUT
                         storage operations timeout
   -S, --skip-ls-dir     skip LSDir tests, needed for Object storage backend

nagios-plugins-storage.spec

@@ -5,7 +5,7 @@
 %define nagios_plugins_dir %{_libdir}/nagios/plugins
 
 Name:       nagios-plugins-storage
-Version:    0.1.1
+Version:    0.1.2
 Release:    1%{?dist}
 Summary:    Nagios probes to be run remotely against EGI Online endpoints
 License:    MIT
@@ -25,7 +25,6 @@ Requires:   gfal2-plugin-file
 Requires:   gfal2-plugin-xrootd
 Requires:   gfal2-plugin-http 
 
-
 %description
 This package provides the nagios probes for XRootD
 
@@ -48,6 +47,9 @@ rm -rf %{buildroot}
 %doc LICENSE README.md
 
 %changelog
+* Wed Jun 19 2024 Andrea Manzi <[email protected]> - 0.1.2-1
+- Add support for bearer tokens
+
 * Tue Jun 18 2024 Andrea Manzi <[email protected]> - 0.1.1-1
 - Add read-only option
 

plugins/storage_probe.py

@@ -26,6 +26,8 @@
 app = nap.core.Plugin(description="NAGIOS Storage probe", version=PROBE_VERSION)
 app.add_argument("-E", "--endpoint", help="base URL to test")
 app.add_argument("-X", "--x509", help="location of x509 certificate proxy file")
+app.add_argument("-to", "--token", help="BEARER TOKEN to be used")
+
 app.add_argument(
     "--se-timeout",
     dest="se_timeout",
@@ -67,7 +69,12 @@ def parse_args(args, io):
         gfal2.cred_set(ctx, "davs://", cred)
         gfal2.cred_set(ctx, "root://", cred)
         gfal2.cred_set(ctx, "xroot://", cred)
-
+    if args.token:
+        cred = gfal2.cred_new("BEARER", args.token)
+        gfal2.cred_set(ctx, "https://", cred)
+        gfal2.cred_set(ctx, "davs://", cred)
+        gfal2.cred_set(ctx, "root://", cred)
+        gfal2.cred_set(ctx, "xroot://", cred)
 
 @app.metric(seq=1, metric_name="LsDir", passive=True)
 def metricLsDir(args, io):