JezK

Edit File: process_cpmove

#!/usr/local/cpanel/3rdparty/bin/perl

#                                      Copyright 2025 WebPros International, LLC
#                                                           All rights reserved.
# copyright@cpanel.net                                         http://cpanel.net
# This code is subject to the cPanel license. Unauthorized copying is prohibited.

=encoding utf-8

=head1 NAME

process_cpmove - Wrapper script for cPanel & WHM's restorepkg binary

=head1 SYNOPSIS

process_cpmove [--help] [--username <username>] [restorepkg options]

=head1 DESCRIPTION

This script provides an easy to call wrapper for cPanel & WHM's restorepkg binary with the specific settings
needed for Comet Backup's agent. It reads a tar archive from stdin, saves it to a temporary
location, and then calls restorepkg to restore the cPanel & WHM specific account settings.

Make sure to call this script AFTER restoring the user's home directory and other files from Comet Backup.

All other arguments are passed through to restorepkg. See C<restorepkg --help> for available options.

=head1 OPTIONS

=over 4

=item B<--help>

Display this help message and exit.

=item B<--username> <username>

B<Required>. Specify the username to restore the account as. This will be passed to restorepkg as C<--newuser>.

=back

=head1 EXAMPLES

# Restore a backup from stdin
    cat /path/to/backup.tar | process_cpmove --username user

# Restore with additional restorepkg options
    cat /path/to/backup.tar | process_cpmove --username user --keep_local_cpuser_values PLAN,DNS

=cut

package scripts::process_cpmove;

use cPstrict;

use Getopt::Long ();
use Pod::Usage   ();
use File::Copy   ();
use File::Temp   ();

use FindBin;
use lib "$FindBin::Bin/../plugins/whm/comet/perl/usr/local/cpanel";

run(@ARGV) if !caller();

sub run {
    my (@args) = @_;

# Plugin policy: nothing in this process should be world-readable.
    umask 0027;

my $help;
    my $username;
    Getopt::Long::Configure('pass_through');
    Getopt::Long::GetOptionsFromArray(
        \@args,
        'help|h|?'   => \$help,
        'username=s' => \$username,
    ) or Pod::Usage::pod2usage(2);

if ($help) {
        Pod::Usage::pod2usage(
            -exitval => 0,
            -verbose => 2,
        );
    }

if ( !defined $username || !length $username ) {
        die "Error: --username is required\n";
    }

my $restorepkg_bin = '/usr/local/cpanel/bin/restorepkg';

if ( !-x $restorepkg_bin ) {
        die "Error: restorepkg binary not found or not executable at $restorepkg_bin\n";
    }

# Create a temporary file to store the incoming cpmove archive
    my $temp_fh = File::Temp->new(
        TEMPLATE => 'cpmove_XXXXXX',
        SUFFIX   => '.tar',
        TMPDIR   => 1,
        UNLINK   => 1,
    );

my $temp_file = $temp_fh->filename();

if ( !defined $temp_file ) {
        die "Error: Failed to create temporary file\n";
    }

# Read from STDIN and write to the temporary file
    binmode STDIN;
    binmode $temp_fh;

my $buffer;
    my $bytes_written = 0;
    while ( my $bytes_read = read( STDIN, $buffer, 65536 ) ) {
        print {$temp_fh} $buffer or die "Error: Failed to write to temporary file: $!\n";
        $bytes_written += $bytes_read;
    }

if ( !defined $buffer ) {
        die "Error: Failed to read from STDIN: $!\n";
    }

# Flush and close the filehandle to ensure all data is written
    close $temp_fh or die "Error: Failed to close temporary file: $!\n";

if ( $bytes_written == 0 ) {
        die "Error: No data received from STDIN\n";
    }

# Call restorepkg with the temporary file
    # We use system() instead of exec() to ensure the temp file isn't cleaned up
    # until after restorepkg completes
    my @cmd = ( $restorepkg_bin, $temp_file, '--newuser', $username, @args );

my $exit_code = system(@cmd);

if ( $exit_code == -1 ) {
        die "Error: Failed to execute restorepkg: $!\n";
    }
    elsif ( $exit_code & 127 ) {
        my $signal = $exit_code & 127;
        die sprintf( "Error: restorepkg died with signal %d\n", $signal );
    }
    else {
        my $status = $exit_code >> 8;
        if ( $status != 0 ) {
            die sprintf( "Error: restorepkg exited with status %d\n", $status );
        }
    }

# Temp file will be automatically cleaned up when $temp_fh goes out of scope
    return 0;
}