Add valid-http-codes option

Add a new valid-http-codes option. By default only 2xx codes are OK now.
Be warned as 401 codes were allowed on previous versions.

Gemfile.lock

@@ -10,3 +10,6 @@ PLATFORMS
 DEPENDENCIES
   hpricot
   optiflag
+
+BUNDLED WITH
+   1.10.3

check_webpage.rb

@@ -145,6 +145,11 @@ module Example extend OptiFlagSet
     description "--graphite-prefix, prefix the graphite path (default to 'webpage')"
   end
 
+  optional_flag "vhc" do
+    long_form "valid-http-codes"
+    description "--valid-http-codes XXX,XXX (ex 401,404)"
+  end
+
   and_process!
 end 
 
@@ -195,6 +200,12 @@ module Example extend OptiFlagSet
   httpHeaders['Authorization'] = 'Basic '+Base64.encode64(ARGV.flags.a)
 end
 
+if ARGV.flags.vhc?
+  validHttpCodes = ARGV.flags.vhc.split(',')
+else
+  validHttpCodes = []
+end
+
 if ARGV.flags.p?
   postData = ARGV.flags.p
 else
@@ -526,11 +537,11 @@ def getInnerLinks (mainUrl, data, httpHeaders, reports, proxy)
 ## check main url return code
 ###############################################################
 if res.code =~ /[^2]../
-  if res.code == "401" && ARGV.flags.a.nil? # server respond but authentication required 
-    text = "OK: server respond but rcode is "
+  if validHttpCodes.include? res.code
+    text = "OK: server respond with http code "
     result = 0
   else
-    text = "Critical: main page rcode is "
+    text = "Critical: main page http code is "
     result = 2
   end
 

test/test.rb

@@ -58,7 +58,9 @@ def test_timeout
   end
 
   def test_auth_err
-    assert_equal(true, system( '../check_webpage.rb -u '+WEB_SERVER_URL+'/401' ) )
+    assert_equal(true,  system( '../check_webpage.rb -u '+WEB_SERVER_URL+'/401 -vhc 401') )
+    assert_equal(true,  system( '../check_webpage.rb -u '+WEB_SERVER_URL+'/401 -vhc 404,401') )
+    assert_equal(false, system( '../check_webpage.rb -u '+WEB_SERVER_URL+'/401' ) )
     assert_equal(false, system( '../check_webpage.rb -a test -u '+WEB_SERVER_URL+'/401' ) )
   end
 end