1 files changed, 108 insertions, 10 deletions

diff --git a/debian/perl-framework/t/modules/proxy_balancer.t b/debian/perl-framework/t/modules/proxy_balancer.t
index a16521b..ee31507 100644
--- a/debian/perl-framework/t/modules/proxy_balancer.t
+++ b/debian/perl-framework/t/modules/proxy_balancer.t
@@ -6,22 +6,120 @@ use Apache::TestRequest;
 use Apache::TestUtil;
 use Apache::TestConfig ();
 
-plan tests => 3, need 'proxy_balancer', 'proxy_http';
+my @echos = ('A'x8, 'A'x64, 'A'x2048, 'A'x4096);
+
+my $skipbodyfailover = !need_min_apache_version("2.4.42");
+my $referertest = 0;
+
+if (have_min_apache_version("2.4.41")) {
+  $referertest = 2;
+}
+
+plan tests => 6+(2*scalar @echos)+$referertest, need 'proxy_balancer', 'proxy_http';
 
 Apache::TestRequest::module("proxy_http_balancer");
 Apache::TestRequest::user_agent(requests_redirectable => 0);
 
+# Extract the nonce from response to the URL
+sub GetNonce {
+  my $url = shift;
+  my $balancer = shift;
+  my $r;
+  $r = GET($url);
+  my $NONCE;
+  foreach my $query ( split( /\?b=/, $r->content ) ){
+    if ($query =~ m/$balancer/) {
+      foreach my $var ( split( /&/, $query ) ){
+        if ($var =~ m/nonce=/) {
+          foreach my $nonce ( split( /nonce=/, $var ) ){
+            my $ind = index ($nonce, "\"");
+            $nonce = substr($nonce, 0, ${ind});
+            if ( $nonce =~ m/^[0-9a-fA-F-]+$/ ) {
+              $NONCE = $nonce;
+              last;
+            }
+          }
+          last;
+        }
+      }
+    last;
+    }
+  }
+  return $NONCE;
+}
+
+my $r;
+
+if (have_module('lbmethod_byrequests')) {
+    $r = GET("/baltest1/index.html");
+    ok t_cmp($r->code, 200, "Balancer did not die");
+} else {
+    skip "skipping tests without mod_lbmethod_byrequests" foreach (1..1);
+}
+
+if (have_module('lbmethod_bytraffic')) {
+    $r = GET("/baltest2/index.html");
+    ok t_cmp($r->code, 200, "Balancer did not die");
+} else {
+    skip "skipping tests without mod_lbmethod_bytraffic" foreach (1..1);
+}
+
+if (have_module('lbmethod_bybusyness')) {
+    $r = GET("/baltest3/index.html");
+    ok t_cmp($r->code, 200, "Balancer did not die");
+} else {
+    skip "skipping tests without mod_lbmethod_bybusyness" foreach (1..1);
+}
+
+if (have_module('lbmethod_heartbeat')) {
+    #$r = GET("/baltest4/index.html");
+    #ok t_cmp($r->code, 200, "Balancer did not die");
+} else {
+    #skip "skipping tests without mod_lbmethod_heartbeat" foreach (1..1);
+}
+
 
-my $r = GET("/baltest1/index.html");
-ok t_cmp($r->code, 200, "Balancer did not die");
 
-$r = GET("/baltest2/index.html");
-ok t_cmp($r->code, 200, "Balancer did not die");
+# PR63891
+foreach my $t (@echos) {
+    $r = POST "/baltest_echo_post", content => $t;
+    skip $skipbodyfailover, t_cmp($r->code, 200, "failed over");
+    skip $skipbodyfailover, t_cmp($r->content, $t, "response body echoed");
+}
+
+# test dynamic part
+$r = GET("/balancer-manager");
+ok t_cmp($r->code, 200, "Can't find balancer-manager");
+
+# get the nonce and add a worker
+my $result = GetNonce("/balancer-manager", "dynproxy");
 
-$r = GET("/baltest3/index.html");
-ok t_cmp($r->code, 200, "Balancer did not die");
+my $query = "b_lbm=byrequests&b_tmo=0&b_max=0&b_sforce=0&b_ss=&b_nwrkr=ajp%3A%2F%2F%5B0%3A0%3A0%3A0%3A0%3A0%3A0%3A1%5D%3A8080&b_wyes=1&b=dynproxy&nonce=" . $result;
+my @proxy_balancer_headers;
+my $vars   = Apache::Test::vars();
+push @proxy_balancer_headers, "Referer" => "http://" . $vars->{servername} . ":" . $vars->{port} . "/balancer-manager";
+
+# First try without the referer it should fail.
+if (have_min_apache_version("2.4.41")) {
+  $r = POST("/balancer-manager", content => $query);
+  ok t_cmp($r->code, 200, "request failed");
+  ok !t_cmp($r->content, qr/ajp/, "AJP worker created");
+}
 
-if (have_min_apache_version("2.3.0")) {
-    # $r = GET("/baltest4/index.html");
-    # ok t_cmp($r->code, 200, "Balancer did not die");
+# Try with the referer and http (byrequests)
+if (have_min_apache_version("2.4.49") && have_module('lbmethod_byrequests')) {
+  $r = GET("/dynproxy");
+  ok t_cmp($r->code, 503, "request should fail for /dynproxy");
+  # create it
+  $query = 'b_lbm=byrequests&b_tmo=0&b_max=0&b_sforce=0&b_ss=&b_nwrkr=http%3A%2F%2F' . $vars->{servername} . '%3A' . $vars->{port} . '&b_wyes=1&b=dynproxy&nonce=' . $result;
+  $r = POST("/balancer-manager", content => $query, @proxy_balancer_headers);
+  # enable it.
+  $query = 'w=http%3A%2F%2F' . $vars->{servername} . '%3A' . $vars->{port} . '&b=dynproxy&w_status_D=0&nonce=' . $result;
+  $r = POST("/balancer-manager", content => $query, @proxy_balancer_headers);
+  # make a query
+  $r = GET("/dynproxy");
+  ok t_cmp($r->code, 200, "request failed to /dynproxy");
+} else {
+    skip "skipping tests without lbmethod_byrequests";
+    skip "skipping tests without lbmethod_byrequests";
 }