Perlでファイル転送(FTP)

#!/usr/bin/perl
# @(#)ftp.pl Copyright (C)2001 ASH. http://ash.jp/
#
# 簡易ファイル転送スクリプト（FTP）
#   Usage: ftp.pl [host [port]]
#

use strict;
use Socket;
use FileHandle;

require 'ftp_sub.pl'; # FTPプロトコル処理関数
require 'ftp_cmd.pl'; # FTPコマンド処理関数

&main();

#
# FTPプログラムメイン処理
#
sub main {
  my ($host, $port, $user, $pass);
  my ($in, $cmd, @opt, $buf);
  my ($comd);

  ($host, $port) = @ARGV;
  if ($host eq '') { $host = 'localhost'; }
  if ($port eq '') { $port = getservbyname('ftp', 'tcp'); }

  $comd = new FileHandle;

  # FTP接続開始
  &ftp_open($comd, $host, $port);
  &ftp_recv($comd);

  # ユーザ認証
  print "Username: "; chomp($user=<STDIN>);
  if ($user eq '') { $user = 'ftp'; }
  &ftp_send($comd, 'USER', $user);
  $buf = &ftp_recv($comd);

  # パスワード認証
  system("stty -echo >/dev/null 2>&1");
  print "Password: "; chomp($pass=<STDIN>);
  system("stty echo >/dev/null 2>&1");
  print "\n";
  &ftp_send($comd, 'PASS', $pass);
  $buf = &ftp_recv($comd);
  if ($buf =~ /^5/) { exit; }

  # SYSTコマンド送信
  &ftp_send($comd, 'SYST');
  $buf = &ftp_recv($comd);

  while (-1) {
    # FTPコマンド入力
    print "ftp> "; chomp($in=<STDIN>);
    ($cmd, @opt) = split(' ', $in);

    # コマンド呼出し処理
    if ($in =~ /^ls/i){ # ls files
      &cmd_ls($comd, $cmd, @opt);

    } elsif ($in =~ /^cd/i) { # cd directory
      # CWDコマンド送信
      &ftp_send($comd, 'CWD', @opt);
      $buf = &ftp_recv($comd);

    } elsif ($in =~ /^pw/i) { # pwd
      # PWDコマンド送信
      &ftp_send($comd, 'PWD');
      $buf = &ftp_recv($comd);

    } elsif ($in =~ /^bi/i) { # binary
      # TYPEコマンド送信
      &ftp_send($comd, 'TYPE', 'I');
      $buf = &ftp_recv($comd);

    } elsif ($in =~ /^as/i) { # ascii
      # TYPEコマンド送信
      &ftp_send($comd, 'TYPE', 'A');
      $buf = &ftp_recv($comd);

    } elsif ($in =~ /^get/i) { # get file
      &cmd_get($comd, $cmd, @opt);

    } elsif ($in =~ /^put/i) { # put file
      &cmd_put($comd, $cmd, @opt);

    } elsif ($in =~ /^!(.*)/) { # escape shell
      system("$1");

    } elsif ($in =~ /^bye|^quit/i) { # quit
      last;

    } else {
      print "? Invalid command\n";
    }
  }

  # FTP接続終了
  &ftp_send($comd, 'QUIT');
  &ftp_recv($comd);
  &ftp_close($comd);
}

#
# FTPプロトコル処理関数
#

# FTP接続開始
sub ftp_open {
  my ($sock, $host, $port) = @_;
  my ($ip, $sockaddr);

  # ソケット生成
  $ip = inet_aton($host) || die "host($host) not found.\n";
  $sockaddr = pack_sockaddr_in($port, $ip);
  socket($sock, PF_INET, SOCK_STREAM, 0) || die "socket error.\n";

  # 接続
  connect($sock, $sockaddr) || die "connect $host $port error.\n";
  autoflush $sock (1);
}

# FTP接続終了
sub ftp_close {
  my ($sock) = @_;

  close($sock);
}

# FTPコマンド送信
sub ftp_send {
  my ($sock, $cmd, @opt) = @_;

  if ($#opt) {
    print "---> $cmd\n";
    print $sock "$cmd\n";
  } else {
    if ($cmd eq 'PASS') {
      print "---> $cmd XXXX\n";
    } else {
      print "---> $cmd @opt\n";
    }
    print $sock "$cmd @opt\n";
  }
}

# FTPレスポンス受信
sub ftp_recv {
  my ($sock) = @_;
  my ($buf, $rc, $cont, $msg);

  $rc = 0;
  while (chomp($buf=<$sock>)) {
    print "$buf\n";
    $buf =~ /^(\d\d\d)([ |-])(.*)/;
    $rc = $1; $cont = $2; $msg = $3;
    if ($cont ne '-') { last; }
  }
  return($buf);
}

sub get_pasv_port {
  my ($buf) = @_;
  my ($host, $port);

  $buf =~ /^2\d\d .*\((\d+,\d+,\d+,\d+),(\d+),(\d+)\)/;
  $host = $1;
  $port = $2 * 256 + $3;
  $host =~ s/,/\./g;

  return($host, $port);
}
1;

#
# FTPコマンド処理関数
#

# lsコマンド処理
sub cmd_ls {
  my ($sock, $cmd, @opt) = @_;
  my ($host, $port, $buf, $pid);
  my ($data);

  # PASVモードに変更
  &ftp_send($sock, 'PASV');
  $buf = &ftp_recv($sock);
  ($host, $port) = &get_pasv_port($buf);

  # LISTコマンド送信
  &ftp_send($sock, 'LIST', @opt);
  if ($pid = fork()) { # 親
    $buf = &ftp_recv($sock);
    if ($buf =~ /^5/) { kill 'TERM', $pid; return; }
    wait;
    $buf = &ftp_recv($sock);

  } else { # 子
    $data = new FileHandle;
    &ftp_open($data, $host, $port);
    while (chomp($buf=<$data>)) {
      print "$buf\n";
    }
    &ftp_close($data);
    exit(0);
  }
}

# getコマンド処理
sub cmd_get {
  my ($sock, $cmd, @opt) = @_;
  my ($host, $port, $buf, $pid, $file);
  my ($data);

  # PASVモードに変更
  &ftp_send($sock, 'PASV');
  $buf = &ftp_recv($sock);
  ($host, $port) = &get_pasv_port($buf);

  # ファイル名取得
  ($file) = @opt;

  # RETRコマンド送信
  &ftp_send($sock, 'RETR', $file);
  if ($pid = fork()) { # 親
    $buf = &ftp_recv($sock);
    if ($buf =~ /^5/) { kill 'TERM', $pid; return; }
    wait;
    $buf = &ftp_recv($sock);

  } else { # 子
    $data = new FileHandle;
    &ftp_open($data, $host, $port);

    open (FILE, ">$file") || exit(1);
    while (<$data>) {
      print FILE $_;
    }
    close (FILE);

    &ftp_close($data);
    exit(0);
  }
}

# putコマンド処理
sub cmd_put {
  my ($sock, $cmd, @opt) = @_;
  my ($host, $port, $buf, $pid, $file);
  my ($data);

  # PASVモードに変更
  &ftp_send($sock, 'PASV');
  $buf = &ftp_recv($sock);
  ($host, $port) = &get_pasv_port($buf);

  # ファイル名取得
  ($file) = @opt;

  # STORコマンド送信
  &ftp_send($sock, 'STOR', $file);
  if ($pid = fork()) { # 親
    $buf = &ftp_recv($sock);
    if ($buf =~ /^5/) { kill 'TERM', $pid; return; }
    wait;
    $buf = &ftp_recv($sock);

  } else { # 子
    $data = new FileHandle;
    &ftp_open($data, $host, $port);

    open (FILE, "$file") || exit(1);
    while (<FILE>) {
      print $data $_;
    }
    close (FILE);

    &ftp_close($data);
    exit(0);
  }
}
1;