Sessions/SSHCopyIDSession.m
////////////////////////////////////////////////////////////////////////////////
//
// B L I N K
//
// Copyright (C) 2016-2018 Blink Mobile Shell Project
//
// This file is part of Blink.
//
// Blink is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Blink is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Blink. If not, see <http://www.gnu.org/licenses/>.
//
// In addition, Blink is also subject to certain additional terms under
// GNU GPL version 3 section 7.
//
// You should have received a copy of these additional terms immediately
// following the terms and conditions of the GNU General Public License
// which accompanied the Blink Source Code. If not, see
// <http://www.github.com/blinksh/blink>.
//
////////////////////////////////////////////////////////////////////////////////

#import "SSHCopyIDSession.h"
#import "BKPubKey.h"
#import "SSHSession.h"

static const char *copy_command =
"sh -c 'umask 077; test -d ~/.ssh || mkdir ~/.ssh ; cat >> .ssh/authorized_keys; test -x /sbin/restorecon && /sbin/restorecon .ssh .ssh/authorized_keys'";

static const char *usage_format =
  "Usage: ssh-copy-id identity_file [user@]host";

@implementation SSHCopyIDSession

- (int)main:(int)argc argv:(char **)argv
{
  if (argc != 3) {
    return [self dieMsg:@(usage_format)];
  }

  NSString *keyName = [NSString stringWithFormat:@"%s", argv[1]];
  BKPubKey *pkcard = [BKPubKey withID:keyName];

  if (!pkcard) {
    return [self dieMsg:@"ERROR: No identities found."];
  }
  const char *public_key = [[pkcard publicKey] UTF8String];

  SSHSession *sshSession = [[SSHSession alloc] initWithDevice:_device andParams:nil];
  
  // Pipe public key
  int pinput[2];
  pipe(pinput);
  FILE *inputr = fdopen(pinput[0], "r");
  fclose(sshSession.stream.in);
  sshSession.stream.in = inputr;

  write(pinput[1], public_key, strlen(public_key));
  write(pinput[1], "\n", 1);
  close(pinput[1]);

  NSString *ssh_command = [NSString stringWithFormat:@"ssh -v %s -- %s", argv[2], copy_command];
  [sshSession executeAttachedWithArgs:ssh_command];

  close(pinput[0]);

  return 0;
}

- (int)dieMsg:(NSString *)msg
{
  fprintf(_stream.out, "%s\n", [msg UTF8String]);
  return -1;
}

@end
