One Hat Cyber Team

View File Name : distribute-type-lib.pl

# Functions for talking to the Distribute.IT API
use strict;
no strict 'refs';
use warnings;
our (%text);
our $account;

my $distribute_api_url = "https://www.distributeit.com.au/api/";
my %distribute_error_map = (
	"100", "Missing parameters",
	"101", "API Site not currently functioning",
	"102", "Authentication Failure",
	"103", "Account has been disabled",
	"104", "User has been disabled",
	"105", "Request coming from incorrect IP address",
	"108", "Account Balance could not be obtained as account is Invoice based",
	"201", "Invalid or not supplied 'Type' parameter",
	"202", "Your Account has not been enabled for this 'Type'",
	"203", "Invalid or not supplied Action/Object parameters",
	"301", "Invalid Order ID",
	"302", "Domain not supplied",
	"303", "Domain Pricing table not set up for your account",
	"304", "Domain not available for Registration",
	"305", "Domain is not renewable",
	"306", "Domain is not transferable",
	"307", "Incorrect Domain Password",
	"308", "Domain UserID or Password not supplied",
	"309", "Invalid Domain Extension",
	"310", "Domain does not exist, has been deleted or transferred away",
	"311", "Domain does not exist in your reseller profile",
	"312", "Supplied UserID and Password do not match the domain",
	"401", "Connection to Registry failed - retry",
	"500", "Pre-Paid balance is not enough to cover order cost",
	"501", "Invalid credit card type. See Appendix G",
	"502", "Invalid credit card number",
	"503", "Invalid credit card expiry date",
	"504", "Credit Card amount plus the current pre-paid balance is not sufficient to cover the cost of the order",
	"505", "Error with credit card transaction at bank",
	"600", "Error with one or more fields when creating a Domain Contact",
	"601", "Error with one or more fields when creating, renewing or transferring a Domain",
	"602", "Error with one or more fields associated with a Host",
	"603", "Error with one or more fields associated with Eligibility fields",
	"604", "Error with one or more fields associated with a Nameserver",
	"610", "Error connecting to registry",
	"611", "Domain cannot be Renewed or Transferred",
	"612", "Locking is not available for this domain",
	"613", "Domain Status prevents changing of domain lock",
	);

# Returns the name of this registrar
sub type_distribute_desc
{
return $text{'type_distribute'};
}

# type_distribute_domains(&account)
# Returns a list of TLDs that can be used with Distribute IT. Hard-coded for
# now, as I don't know of any programatic way to get this.
sub type_distribute_domains
{
return (".com", ".net", ".org", ".biz", ".info", ".com.au", ".net.au",
	".org.au", ".asn.au", ".id.au", ".co.uk", ".org.uk", ".co.nz",
	".ac.nz", ".gen.nz", ".geek.nz", ".maori.nz", ".net.nz", ".org.nz",
	".school.nz", ".cn", ".tv", ".ca", ".cc", ".name", ".vic.au", ".mobi",
	".asia", ".hk", ".com.my", ".net.my", ".org.my", ".vn", ".com.vn",
	".net.vn", ".biz.vn", ".org.vn", ".info.vn", ".name.vn", ".sg",
	".ph", ".com.ph", ".net.ph", ".org.ph", ".jp", ".in", ".co.in",
	".kr", ".co.kr", ".tw", ".md");
}

# type_distribute_edit_inputs(&account, new?)
# Returns table fields for entering the account login details
sub type_distribute_edit_inputs
{
my ($account, $new) = @_;
my $rv;
$rv .= &ui_table_row($text{'distribute_account'},
	&ui_textbox("distribute_account",
		    $account->{'distribute_account'}, 10));
$rv .= &ui_table_row($text{'distribute_user'},
	&ui_textbox("distribute_user", $account->{'distribute_user'}, 30));
$rv .= &ui_table_row($text{'distribute_pass'},
	&ui_textbox("distribute_pass", $account->{'distribute_pass'}, 30));
$rv .= &ui_table_row($text{'rcom_years'},
	&ui_opt_textbox("distribute_years", $account->{'distribute_years'},
			4, $text{'rcom_yearsdef'}));
$rv .= &ui_table_row($text{'distribute_dom'},
	&ui_textbox("distribute_dom", $account->{'distribute_dom'}, 40));
return $rv;
}

# type_distribute_edit_parse(&account, new?, &in)
# Updates the account object with parsed inputs. Returns undef on success or
# an error message on failure.
sub type_distribute_edit_parse
{
my ($account, $new, $in) = @_;
$in->{'distribute_account'} =~ /^\d+$/ || return $text{'distribute_eaccount'};
$account->{'distribute_account'} = $in->{'distribute_account'};
$in->{'distribute_user'} =~ /^\S+$/ || return $text{'distribute_euser'};
$account->{'distribute_user'} = $in->{'distribute_user'};
$in->{'distribute_pass'} =~ /^\S+$/ || return $text{'distribute_epass'};
$account->{'distribute_pass'} = $in->{'distribute_pass'};
if ($in->{'distribute_years_def'}) {
	delete($account->{'distribute_years'});
	}
else {
	$in->{'distribute_years'} =~ /^\d+$/ && $in->{'distribute_years'} > 0 &&
	  $in->{'distribute_years'} <= 10 || return $text{'rcom_eyears'};
	$account->{'distribute_years'} = $in->{'distribute_years'};
	}
$in->{'distribute_dom'} =~ /^[a-z0-9\.\-\_]+$/i ||
	return $text{'distribute_edom'};
$account->{'distribute_dom'} = $in->{'distribute_dom'};
return undef;
}

# type_distribute_renew_years(&account, &domain)
# Returns the number of years by default to renew a domain for
sub type_distribute_renew_years
{
my ($account, $d) = @_;
return $account->{'distribute_years'} || 2;
}

# type_distribute_validate(&account)
# Checks if an account's details are vaid. Returns undef if OK or an error
# message if the login or password are wrong.
sub type_distribute_validate
{
my ($account) = @_;

# Try to login
my ($ok, $sid) = &connect_distribute_api($account, 1);
return $sid if (!$ok);

# Validate the template domain
($ok, my $out) = &call_distribute_api(
	$sid, "query", { 'Type' => 'Domains',
			 'Object' => 'Domain',
			 'Action' => 'Details',
			 'Domain' => $account->{'distribute_dom'} });
$ok || return &text('distribute_edom2', $out);

return undef;
}

# type_distribute_check_domain(&account, domain)
# Checks if some domain name is available for registration, returns undef if
# yes, or an error message if not.
sub type_distribute_check_domain
{
my ($account, $dname) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);
($ok, my $out) = &call_distribute_api(
	$sid, "query", { 'Type' => 'Domains',
			 'Object' => 'Domain',
			 'Action' => 'Availability',
			 'Domain' => $dname });
return &text('distribute_taken', "$1") if (!$ok && $out =~ /304,(.*)/);
return &text('distribute_error', $out) if (!$ok);
return undef;
}

# type_distribute_owned_domain(&account, domain, [id])
# Checks if some domain is owned by the given account. If so, returns the
# 1 and the registrar's ID - if not, returns 1 and undef, on failure returns
# 0 and an error message.
sub type_distribute_owned_domain
{
my ($account, $dname, $id) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);
my $out;
($ok, $out) = &call_distribute_api(
	 $sid, "query", { 'Type' => 'Domains',
                          'Object' => 'Domain',
                          'Action' => 'Details',
			  'Domain' => $dname });
return !$ok && $out =~ /310/ ? (1, undef) :
       !$ok ? (0, $out) : (1, $dname);
}

# type_distribute_create_domain(&account, &domain)
# Actually register a domain, if possible. Returns 0 and an error message if
# it failed, or 1 and an ID for the domain on success.
sub type_distribute_create_domain
{
my ($account, $d) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);

# NS records come from the DNS domain
my $nss = &get_domain_nameservers($account, $d);
if (!ref($nss)) {
	return (0, $nss);
	}
elsif (!@$nss) {
	return (0, $text{'rcom_ensrecords'});
	}
elsif (@$nss < 2) {
	return (0, &text('distribute_enstwo', 2, $nss->[0]));
	}

# Get contact ID from the base domain
my ($ownerid, $adminid, $techid, $billingid);
($ok, $ownerid, $adminid, $techid, $billingid) =
	&get_distribute_contact_ids($account, $sid);
$ok || return (0, $ownerid);

# Create parameters
my %params = ( 'Type' => 'Domains',
		  'Object' => 'Domain',
		  'Action' => 'Create',
		  'Domain' => $d->{'dom'},
		  'UserID' => &distribute_username($d),
		  'Password' => &distribute_password($d),
		  'Host' => $nss,
		  'OwnerContactID' => $ownerid,
		  'AdministrationContactID' => $adminid,
		  'TechnicalContactID' => $techid,
		  'BillingContactID' => $billingid,
		);
if ($d->{'registrar_years'}) {
	$params{'Period'} = $d->{'registrar_years'};
	}
elsif ($account->{'distribute_years'}) {
	$params{'Period'} = $account->{'distribute_years'};
	}

# Create it
my $out;
($ok, $out) = &call_distribute_api($sid, "order", \%params);
if ($ok) {
	# Done, and got order ID
	return (1, $out);
	}
else {
	return (0, $out);
	}
}

# get_distribute_contact_ids(&account, sid)
# Returns the default owner, admin, tech and contact IDs for some account,
# or an error message
sub get_distribute_contact_ids
{
my ($account, $sid) = @_;
my ($ok, $out) = &call_distribute_api(
	$sid, "query", { 'Type' => 'Domains',
			 'Object' => 'Domain',
			 'Action' => 'Details',
			 'Domain' => $account->{'distribute_dom'} });
$ok || return (0, &text('distribute_ebase', $account->{'distribute_dom'}));
my ($ownerid, $adminid, $techid, $billingid);
if ($out =~ /Owner-ContactID=(\S+)/) {
	$ownerid = $1;
	}
else {
	return (0, &text('distribute_eowner', $account->{'distribute_dom'}));
	}
$adminid = $out =~ /Administration-ContactID=(\S+)/ ? $1 : $ownerid;
$techid = $out =~ /Technical-ContactID=(\S+)/ ? $1 : $ownerid;
$billingid = $out =~ /Billing-ContactID=(\S+)/ ? $1 : $ownerid;
$ownerid =~ s/^DIT-//i;
$adminid =~ s/^DIT-//i;
$techid =~ s/^DIT-//i;
$billingid =~ s/^DIT-//i;
return (1, $ownerid, $adminid, $techid, $billingid);
}

# type_distribute_set_nameservers(&account, &domain, [&nameservers])
# Updates the nameservers for a domain to match DNS. Returns undef on success
# or an error message on failure.
sub type_distribute_set_nameservers
{
my ($account, $d, $nss) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);

# Get nameservers in DNS
$nss ||= &get_domain_nameservers($account, $d);
if (!ref($nss)) {
	return $nss;
	}
elsif (!@$nss) {
	return $text{'rcom_ensrecords'};
	}
elsif (@$nss < 2) {
	return (0, &text('distribute_enstwo', 2, $nss->[0]));
	}

# Set nameservers
($ok, my $out) = &call_distribute_api(
	$sid, "order", { 'Type' => 'Domains',
                          'Object' => 'Domain',
                          'Action' => 'UpdateHosts',
			  'Domain' => $d->{'dom'},
			  'AddHost' => $nss,
			  'RemoveHost' => 'ALL' });
return $ok ? undef : $out;
}

# type_gandi_get_nameservers(&account, &domain)
# Returns a array ref list of nameserver hostnames for some domain, or
# an error message on failure.
sub type_distribute_get_nameservers
{
my ($account, $d, $nss) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);

($ok, my $out) = &call_distribute_api(
	$sid, "query", { 'Type' => 'Domains',
			 'Object' => 'Domain',
			 'Action' => 'Details',
			 'Domain' => $d->{'dom'} });
$ok || return $out;
my @rv;
foreach my $l (split(/\r?\n/, $out)) {
	if ($l =~ /^Nameserver=(\S+)/) {
		push(@rv, $1);
		}
	}
return \@rv;
}

# type_distribute_delete_domain(&account, &domain)
# Deletes a domain previously created with this registrar
sub type_distribute_delete_domain
{
my ($account, $d) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);

($ok, my $out) = &call_distribute_api(
	$sid, "order", { 'Type' => 'Domains',
			 'Object' => 'Domain',
			 'Action' => 'Cancel',
			 'OrderID' => $d->{'registrar_id'} });
return ($ok, $out);
}

# type_distribute_get_expiry(&account, &domain)
# Returns either 1 and the expiry time (unix) for a domain, or 0 and an error
# message.
sub type_distribute_get_expiry
{
my ($account, $d) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);
my $os = &check_distribute_order_status($sid, $d->{'registrar_id'});
return (0, $os) if ($os);
($ok, my $out) = &call_distribute_api(
	 $sid, "query", { 'Type' => 'Domains',
                          'Object' => 'Domain',
                          'Action' => 'Details',
			  'Domain' => $d->{'dom'} });
if (!$ok) {
	# Failed completed
	return (0, $out);
	}
elsif ($out =~ /ExpiryDate=(\d+)\-(\d+)\-(\d+)/i) {
	# Got a date
	return (1, timelocal(0, 0, 0, $3, $2-1, $1-1900));
	}
else {
	# Unknown output??
	return (0, "Unknown expiry information : $out");
	}
}

# type_distribute_renew_domain(&account, &domain, years)
# Attempts to renew a domain for the specified period. Returns 1 and the
# registrars confirmation code on success, or 0 and an error message on
# failure.
sub type_distribute_renew_domain
{
my ($account, $d, $years) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);
my $os = &check_distribute_order_status($sid, $d->{'registrar_id'});
return (0, $os) if ($os);
($ok, my $out) = &call_distribute_api(
	$sid, "order", { 'Type' => 'Domains',
                          'Object' => 'Domain',
                          'Action' => 'Renewal',
			  'Domain' => $d->{'dom'},
			  'Period' => $years });
return ($ok, $out);
}

# type_distribute_transfer_domain(&account, &domain, key, [years])
# Transfer a domain from whatever registrar it is currently hosted with to
# this Distribute.IT account. Returns 1 and an order ID on succes, or 0
# and an error mesasge on failure. If a number of years is given, also renews
# the domain for that period.
sub type_distribute_transfer_domain
{
my ($account, $d, $key, $years) = @_;
my ($ok, $sid) = &connect_distribute_api($account, 1);
return &text('distribute_error', $sid) if (!$ok);

# Get contact ID from the base domain
my ($ownerid, $adminid, $techid, $billingid);
($ok, $ownerid, $adminid, $techid, $billingid) =
	&get_distribute_contact_ids($account, $sid);
$ok || return (0, $ownerid);

($ok, my $out) = &call_distribute_api(
	$sid, "order", { 'Type' => 'Domains',
                         'Object' => 'Domain',
                         'Action' => 'TransferRequest',
			 'Domain' => $d->{'dom'},
			 $years ? ( 'Period' => $years ) : ( ),
			 'DomainPassword' => $key,
			 'UserID' => &distribute_username($d),
			 'Password' => &distribute_password($d),
		         'OwnerContactID' => $ownerid,
		         'AdministrationContactID' => $adminid,
		         'TechnicalContactID' => $techid,
		         'BillingContactID' => $billingid, });
return ($ok, $out);
}

# check_distribute_order_status(sid, order-id)
# Returns an error message if a domain order is pending, undef if not
sub check_distribute_order_status
{
my ($sid, $orderid) = @_;
return undef if ($orderid !~ /^\d+$/);		# Don't know it
my ($ok, $out) = &call_distribute_api(
         $sid, "query", { 'Type' => 'Domains',
                          'Object' => 'Order',
			  'Action' => 'OrderStatus',
			  'OrderID' => $orderid });
$out =~ s/,\s*$//;
return !$ok ? &text('distribute_estatus2', $out) :
       $out =~ /^Complete/i ? undef :
       &text('distribute_estatus', $out);
}

# connect_distribute_api(&account, return-error)
# Login to the API, and return 1 and a session ID or 0 and an error message
sub connect_distribute_api
{
my ($account, $reterr) = @_;
my ($ok, $sid) = &call_distribute_api(
	undef, "auth", { 'AccountNo' => $account->{'distribute_account'},
			 'UserId' => $account->{'distribute_user'},
			 'Password' => $account->{'distribute_pass'} });
&error("Distribute IT login failed : $sid") if (!$ok && !$reterr);
return ($ok, $sid);
}

# call_distribute_api(session-id, program, &params)
# Calls the API via an HTTP request, and returns either 1 and the output or
# 0 and an error message
sub call_distribute_api
{
my ($sid, $prog, $params) = @_;
my ($host, $port, $page, $ssl) = &parse_http_url($distribute_api_url);
$params ||= { };
if ($sid) {
	$params->{'SessionID'} = $sid;
	}
$page .= $prog.".pl";
my @params;
foreach my $k (keys %$params) {
	my $v = $params->{$k};
	foreach my $vv (ref($v) ? @$v : ( $v )) {
		push(@params, &urlize($k)."=".&urlize($vv));
		}
	}
$page .= "?".join("&", @params);
my ($out, $err);
&http_download($host, $port, $page, \$out, \$err, undef, $ssl);
if ($err =~ /403/) {
	# Bad IP .. warn specifically
	return (0, $text{'distribute_eip'});
	}
elsif ($err) {
	# Some other HTTP error
	return (0, $err);
	}
if ($out =~ /^((\S+):\s+)?OK:\s*([\000-\377]*)/) {
	# Valid response
	return (1, $3, $2);
	}
elsif ($out =~ /^((\S+):\s+)?ERR:\s*([\000-\377]*)/) {
	# Valid error
	my ($ecode, $dname) = ($3, $2);
	if ($ecode =~ /^(\d+)(.*)$/ && $distribute_error_map{$1}) {
		$ecode = $1." - ".$distribute_error_map{$1}.$2;
		}
	return (0, $ecode, $dname);
	}
else {
	# Some other output??
	return (0, "Unknown error : $out");
	}
}

sub distribute_username
{
my ($d) = @_;
my $rv = $d->{'dom'};
$rv =~ s/[^a-z0-9]//gi;
$rv = substr($rv, 0, 16) if (length($rv) > 16);
return $rv;
}

sub distribute_password
{
my ($d) = @_;
my $rv = $d->{'pass'} || &virtual_server::random_password(8);
return $rv;
}

1;