From d900bca110121132f4e33f317c1cd666bed1c286 Mon Sep 17 00:00:00 2001
From: Tomas Plesnik <plesnik@ics.muni.cz>
Date: Tue, 14 Oct 2014 14:23:04 +0200
Subject: [PATCH] pridana kontrola parametru client_id jako integer; smazana
 kontrola parametru client_id jestli je definovan, jelikoz kontrolu zajisti
 odchytavani poctu argumentu skriptu; lepsi prehlednost kodu (nahrazeni
 negovanych ifu za unless); uprava odsazeni

---
 src/warden-server/bin/unregisterClient.pl | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/warden-server/bin/unregisterClient.pl b/src/warden-server/bin/unregisterClient.pl
index 1fe3778..db4d62b 100755
--- a/src/warden-server/bin/unregisterClient.pl
+++ b/src/warden-server/bin/unregisterClient.pl
@@ -95,17 +95,18 @@ if ($help) {
 
 # catch version option
 if ($script_version) {
-	  version;
+  version;
 }
 
 # check parameters definition
-if (!defined $client_id) {
-  WardenCommon::errMsg("Parameter 'client_id' is not defined!");
+if ($client_id !~ /^\d+\z/) {
+  WardenCommon::errMsg("Parameter '-i <client_id>' is not a integer");
 }
 
 # connect to DB
 my $dbh = WardenCommon::connectDB($WardenCommon::DB_NAME, $WardenCommon::DB_HOST, $WardenCommon::DB_USER, $WardenCommon::DB_PASS);
 
+# obtain a server hostname
 my $warden_server = WardenCommon::trim(`hostname -f`);
 
 # check if client has been already registered
@@ -114,7 +115,7 @@ $sth->execute($client_id) or die "Cannot execute statement: " . $sth->errstr;
 my ($id, $hostname, $service, $client_type, $type) = $sth->fetchrow();
 
 # delete registered client
-if (!defined $id) {
+unless (defined $id) {
   sendMsg("err", "Attempt to delete unregister client [Client_id: '$client_id'] from Warden server '$warden_server'");
   WardenCommon::errMsg("Client (#$client_id) is not registered at Warden server '$warden_server'");
 } else {
-- 
GitLab