# User Defined Reflective Loader Kit Aggressor Script

# Enable Debug of PE content
# The generated PE content will be displayed in the script console if debug is true

#$debug = "true";
$debug = "true";


# ===========================================================================
# 'case insensitive sort' from sleep manual...
# ===========================================================================
sub caseInsensitiveCompare
{
    $a = lc($1);
    $b = lc($2);
    return $a cmp $b;
}

# ===========================================================================
# Dump PE Information
# $1 = Beacon DLL content
# ===========================================================================
sub dump_my_pe {
    local('$out $key $val %pemap @sorted_keys');

    %pemap = pedump($1);

    # ---------------------------------------------------
    # Example listing all items from hash/map...
    # ---------------------------------------------------
    @sorted_keys = sort(&caseInsensitiveCompare, keys(%pemap));
    foreach $key (@sorted_keys)
    {
        $out = "$[50]key";
        foreach $val (values(%pemap, @($key)))
        {
            $out .= " $val";
            println($out);
        }
    }

    # ---------------------------------------------------
    # Example of grabbing specific items from hash/map...
    # ---------------------------------------------------
    local('@loc_cs @val_cs');
    @loc_cs = values(%pemap, @("CheckSum."));
    @val_cs = values(%pemap, @("CheckSum."));

    println("");
    println("My DLL CheckSum Location: " . @loc_cs);
    println("My DLL CheckSum Value: " . @val_cs);
    println("");
}

sub generate_my_dll {

	local('$handle $data $loader $temp_dll');

	# ---------------------------------------------------------------------
	# Load a object file that contains a Reflective Loader.
	# The architecture ($3) is used in the path.
	# ---------------------------------------------------------------------
    $handle = openf(script_resource("/bin/ReflectiveLoader. $+ $3 $+ .o"));
	$data   = readb($handle, -1);
	closef($handle);

	warn("Loaded Length: " . strlen($data));

	if (strlen($data) eq 0) {
		warn("Error loading reflective loader object file.");
		return $null;
	}

	# ---------------------------------------------------------------------
	# extract loader ($loader) from the object file data ($data).
	# ---------------------------------------------------------------------
	$loader = extract_reflective_loader($data);

	warn("Extracted Length: " . strlen($loader));

	if (strlen($loader) eq 0) {
		warn("Error extracting reflective loader.");
		return $null;
	}

	# ---------------------------------------------------------------------
	# Setup the reflective loader ($loader) in the beacon ($2).
	# ---------------------------------------------------------------------
	$temp_dll = setup_reflective_loader($2, $loader);

	# ---------------------------------------------------------------------
	# OPTIONAL: Additional Customization of the PE...
	# 	- Use 'pedump' function to get information for the updated DLL.
	# 	- Use these convenience functions to perform transformations on the DLL:
	# 		pe_remove_rich_header
	# 		pe_insert_rich_header
	# 		pe_set_compile_time_with_long
	# 		pe_set_compile_time_with_string
	# 		pe_set_export_name
	# 		pe_update_checksum
	# 	- Use these basic functions to perform transformations on the DLL:
	# 		pe_mask
	# 		pe_mask_section
	# 		pe_mask_string
	# 		pe_patch_code
	# 		pe_set_string
	# 		pe_set_stringz
	# 		pe_set_long
	# 		pe_set_short
	# 		pe_set_value_at
	# 		pe_stomp
	# ---------------------------------------------------------------------

	println($debug);
	if ($debug eq "true") {
		dump_my_pe($temp_dll);
	}

	# ---------------------------------------------------------------------
	# Give back the updated beacon DLL.
	# ---------------------------------------------------------------------
	return $temp_dll;
}

# ------------------------------------
# $1 = DLL file name
# $2 = DLL content
# $3 = arch
# ------------------------------------
set BEACON_RDLL_GENERATE {
	warn("========== Running 'BEACON_RDLL_GENERATE' for DLL " . $1 . " with architecture " . $3 . " ==========");
	return generate_my_dll($1, $2, $3);
}

# ------------------------------------
# $1 = DLL file name
# $2 = DLL content
# $3 = arch
# $4 = parent Beacon ID
# $5 = GetModuleHandleA pointer
# $6 = GetProcAddress pointer
# ------------------------------------
set BEACON_RDLL_GENERATE_LOCAL {
	warn("========== Running 'BEACON_RDLL_GENERATE_LOCAL' for DLL " . $1 . " with architecture " . $3 . " Beacon ID " . $4 . " GetModuleHandleA " . $5 . " GetProcAddress " . $6 . " ==========");
	return generate_my_dll($1, $2, $3);
}