From 21e68a8a336c6cd0b478631863a3b0d253c4808a Mon Sep 17 00:00:00 2001
From: Dietmar Maurer <dietmar@proxmox.com>
Date: Wed, 28 Dec 2016 11:35:30 +0100
Subject: [PATCH] perftest1.pl: another example

---
 Makefile              |  1 +
 debian/changelog      |  2 +-
 examples/perftest1.pl | 96 +++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 98 insertions(+), 1 deletion(-)
 create mode 100755 examples/perftest1.pl

diff --git a/Makefile b/Makefile
index 08b1ce4..7947880 100644
--- a/Makefile
+++ b/Makefile
@@ -24,6 +24,7 @@ install:
 	install -d -m 755 ${DOCDIR}/examples
 	install -m 0755 examples/example1.pl ${DOCDIR}/examples
 	install -m 0755 examples/example2.pl ${DOCDIR}/examples
+	install -m 0755 examples/perftest1.pl ${DOCDIR}/examples
 
 .PHONY: upload
 upload: ${DEB}
diff --git a/debian/changelog b/debian/changelog
index 81d3dea..bd9cb3c 100644
--- a/debian/changelog
+++ b/debian/changelog
@@ -1,6 +1,6 @@
 libpve-apiclient-perl (1.0-2) unstable; urgency=medium
 
-  * add simple example code: example1.pl and example2.pl
+  * add simple example code: example1.pl, example2.pl and perftest1.pl
 
  -- Proxmox Support Team <support@proxmox.com>  Wed, 28 Dec 2016 10:46:52 +0100
 
diff --git a/examples/perftest1.pl b/examples/perftest1.pl
new file mode 100755
index 0000000..17f1a91
--- /dev/null
+++ b/examples/perftest1.pl
@@ -0,0 +1,96 @@
+#!/usr/bin/perl
+
+use strict;
+use warnings;
+
+# NOTE: you need to run this on a PVE host, or modify the source to
+# provide username/password/hostname from somewhere else.
+
+use Time::HiRes qw( usleep ualarm gettimeofday tv_interval );
+
+use PVE::APIClient::LWP;
+use PVE::AccessControl;
+use PVE::INotify;
+use JSON;
+
+# normally you use username/password,
+# but we can simply create a ticket and CRSF token if we are root
+# running on a pve host
+
+my $hostname = PVE::INotify::read_file("hostname");
+my $ticket = PVE::AccessControl::assemble_ticket('root@pam');
+my $csrftoken = PVE::AccessControl::assemble_csrf_prevention_token('root@pam');
+
+my $wcount = 10;
+my $qcount = 100;
+
+sub get_local_cert_fingerprint { my ($node) = @_; my $cert_path =
+    "/etc/pve/nodes/$node/pve-ssl.pem"; my $custom_cert_path =
+    "/etc/pve/nodes/$node/pveproxy-ssl.pem";
+
+    $cert_path = $custom_cert_path if -f $custom_cert_path;
+
+    my $bio = Net::SSLeay::BIO_new_file($cert_path, 'r'); my $cert =
+    Net::SSLeay::PEM_read_bio_X509($bio); Net::SSLeay::BIO_free($bio);
+
+    my $fp = Net::SSLeay::X509_get_fingerprint($cert, 'sha256'); die
+    "got empty fingerprint" if !defined($fp) || ($fp eq '');
+
+    return $fp; }
+
+my $local_fingerprint = get_local_cert_fingerprint($hostname);
+
+sub test_rpc {
+    my ($host) = @_;
+
+    my $conn = PVE::APIClient::LWP->new(
+	#username => 'root@pam',
+	#password => 'yourpassword',
+	ticket => $ticket,
+	csrftoken => $csrftoken,
+	host => $host,
+	# add local hosts cert fingerprint
+	cached_fingerprints => {
+	    $local_fingerprint => 1,
+    });
+
+    for (my $i = 0; $i < $qcount; $i++) {
+	eval {
+	    my $res = $conn->get("/", {});
+	};
+	if (my $err = $@) {
+	    print "ERROR: $err\n";
+	    last;
+	}
+    }
+}
+
+sub run_tests {
+    my ($host) = @_;
+    
+    my $workers;
+
+    my $starttime = [gettimeofday];
+
+    for (my $i = 0; $i < $wcount; $i++) {
+	if (my $pid = fork ()) {
+	    $workers->{$pid} = 1;
+	} else {
+	    test_rpc($host);
+	    exit (0);
+	}
+    }
+
+    # wait for children
+    1 while (wait > 0);
+
+    my $elapsed = int(tv_interval ($starttime) * 1000);
+
+    my $tpq = $elapsed / ($wcount*$qcount);
+
+    print "$host: $tpq ms per query\n";
+}
+
+run_tests("localhost"); # test 'pvedaemon'
+
+run_tests($hostname); # test 'pveproxy'
-- 
2.39.2