/*

 * Dropbear - a SSH2 server

 *

 * Copyright (c) 2002,2003 Matt Johnston

 * All rights reserved.

 *

 * Permission is hereby granted, free of charge, to any person obtaining a copy

 * of this software and associated documentation files (the "Software"), to deal

 * in the Software without restriction, including without limitation the rights

 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

 * copies of the Software, and to permit persons to whom the Software is

 * furnished to do so, subject to the following conditions:

 *

 * The above copyright notice and this permission notice shall be included in

 * all copies or substantial portions of the Software.

 *

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

 * SOFTWARE. */

#include "includes.h"

#include "runopts.h"

#include "signkey.h"

#include "buffer.h"

#include "dbutil.h"

#include "auth.h"

#include "algo.h"

#include "dbrandom.h"

runopts opts; /* GLOBAL */

/* returns success or failure, and the keytype in *type. If we want

 * to restrict the type, type can contain a type to return */

int readhostkey(const char * filename, sign_key * hostkey,

  enum signkey_type *type) {

  int ret = DROPBEAR_FAILURE;

  buffer *buf;

  buf = buf_new(MAX_PRIVKEY_SIZE);

  if (buf_readfile(buf, filename) == DROPBEAR_FAILURE) {

    goto out;

  }

  buf_setpos(buf, 0);

  addrandom(buf_getptr(buf, buf->len), buf->len);

  if (buf_get_priv_key(buf, hostkey, type) == DROPBEAR_FAILURE) {

    goto out;

  }

  ret = DROPBEAR_SUCCESS;

out:

  buf_burn_free(buf);

  return ret;

}

#if DROPBEAR_USER_ALGO_LIST

void

parse_ciphers_macs() {

  int printed_help = 0;

  if (opts.cipher_list) {

    if (strcmp(opts.cipher_list, "help") == 0) {

      char *ciphers = algolist_string(sshciphers);

      dropbear_log(LOG_INFO, "Available ciphers: %s", ciphers);

      m_free(ciphers);

      printed_help = 1;

    } else {

      if (check_user_algos(opts.cipher_list, sshciphers, "cipher") == 0) {

        dropbear_exit("No valid ciphers specified for '-c'");

      }

    }

  }

  if (opts.mac_list) {

    if (strcmp(opts.mac_list, "help") == 0) {

      char *macs = algolist_string(sshhashes);

      dropbear_log(LOG_INFO, "Available MACs: %s", macs);

      m_free(macs);

      printed_help = 1;

    } else {

      if (check_user_algos(opts.mac_list, sshhashes, "MAC") == 0) {

        dropbear_exit("No valid MACs specified for '-m'");

      }

    }

  }

  if (printed_help) {

    dropbear_exit(".");

  }

}

#endif

void print_version() {

  fprintf(stderr, "Dropbear v%s\n", DROPBEAR_VERSION);

}

void parse_recv_window(const char* recv_window_arg) {

  int ret;

  unsigned int rw;

  ret = m_str_to_uint(recv_window_arg, &rw);

  if (ret == DROPBEAR_FAILURE || rw == 0 || rw > MAX_RECV_WINDOW) {

    if (rw > MAX_RECV_WINDOW) {

      opts.recv_window = MAX_RECV_WINDOW;

    }

    dropbear_log(LOG_WARNING, "Bad recv window '%s', using %d",

      recv_window_arg, opts.recv_window);

  } else {

    opts.recv_window = rw;

  }

}

/* Splits addr:port. Handles IPv6 [2001:0011::4]:port style format.

   Returns first/second parts as malloced strings, second will

   be NULL if no separator is found.

   :port  ->  (NULL, "port")

   port  ->   (port, NULL)

   addr:port  (addr, port)

   addr: ->   (addr, "")

   Returns DROPBEAR_SUCCESS/DROPBEAR_FAILURE */

int split_address_port(const char* spec, char **first, char ** second) {

  char *spec_copy = NULL, *addr = NULL, *colon = NULL;

  int ret = DROPBEAR_FAILURE;

  *first = NULL;

  *second = NULL;

  spec_copy = m_strdup(spec);

  addr = spec_copy;

  if (*addr == '[') {

    addr++;

    colon = strchr(addr, ']');

    if (!colon) {

      dropbear_log(LOG_WARNING, "Bad address '%s'", spec);

      goto out;

    }

    *colon = '\0';

    colon++;

    if (*colon == '\0') {

      /* No port part */

      colon = NULL;

    } else if (*colon != ':') {

      dropbear_log(LOG_WARNING, "Bad address '%s'", spec);

      goto out;

    }

  } else {

    /* search for ':', that separates address and port */

    colon = strrchr(addr, ':');

  }

  /* colon points to ':' now, or is NULL */

  if (colon) {

    /* Split the address/port */

    *colon = '\0';

    colon++;

    *second = m_strdup(colon);

  }

  if (strlen(addr)) {

    *first = m_strdup(addr);

  }

  ret = DROPBEAR_SUCCESS;

out:

  m_free(spec_copy);

  return ret;

}