Changeset 228

Timestamp:

03/09/07 15:04:44 (1 year ago)

Author:

kindlund

Message:

sc: merging branch using tags svn+ssh://kindlund@www.honeyclient.org/home/svn/honeyclient/honeyclient/tags/exp/PRE-flindiakos-multi_urls and svn+ssh://kindlund@www.honeyclient.org/home/svn/honeyclient/honeyclient/branches/exp/flindiakos-multi_urls

Files:

• honeyclient/trunk/bin/StartManager.pl (modified) (1 diff)

honeyclient/trunk/bin/StartManager.pl

@@ -15 +15 @@
-use MIME::Base64 qw(encode_base64 decode_base64);
-
+# Include Getopt Parser
+use Getopt::Long;
+
-use HoneyClient::Manager;
-
-# We expect that the user will supply a single argument to this script.
-# Namely, the initial URL that they want the Agent to use.
+# They can however supply multiple urls which will be processed in order
+
+my $driver = "IE";
+my $config = "/vm/Agent.Master-7/winXPPro.cfg";
+my $maxrel = 5;
+my $nexturl = "";
+my $urllist= "";
+
+# TODO: Need --help option, along with sanity checking.
+# TODO: Also need a decent POD for this code.
+GetOptions('driver=s'             => \$driver,
+           'master_vm_config=s'   => \$config,
+           'url_list=s'           => \$urllist,
+           'max_relative_links:i' => \$maxrel);
+
+# Go through the list of urls to create the array
+# Anything not associated with an option is a URL
+# Grab those first and then get the ones from the file specified
+my @urls;
+push( @urls, @ARGV ); 
+if( -e $urllist ){
+    open URL, $urllist;
+    push(@urls, <URL>);
+}
+
+# Get the first url from the list
+# Create a hashtable in the form: url => 1 for links_to_visit 
+chomp @urls;
+my $firsturl = shift @urls;
+my %remaining_urls;
+foreach(@urls){
+    # We assign our initial list of URLs a priority of 1000, so that
+    # they'll be (likely to be) selected first, before going to any other
+    # external URLs found from subsequent drive operations.
+    $remaining_urls{$_} = 1000;
+}
-
-my $agentState = HoneyClient::Manager->run(
-'IE'
-'/vm/Agent.Master-7/winXPPro.cfg'
+$driver
+$config
-                    agent_state      => encode_base64(nfreeze({
-IE
-ARGV[0]
+$driver
+firsturl
-                            # Enable this line, if you want to only go to the
-                            # first 5 links for each domain.
-
+
+
-                         },
-                    })),