//---------------------------------------------------------------------------
//  __________________    _________  _____            _____  .__         ._.
//  \______   \______ \  /   _____/ /     \          /  _  \ |__| ____   | |
//   |    |  _/|    |  \ \_____  \ /  \ /  \        /  /_\  \|  _/ __ \  | |
//   |    |   \|    `   \/        /    Y    \      /    |    |  \  ___/   \|
//   |______  /_______  /_______  \____|__  / /\   \____|__  |__|\___ |   __
//          \/        \/        \/        \/  )/           \/        \/   \/
//
// This file is part of libdsm. Copyright © 2014 VideoLabs SAS
//
// Author: Julien 'Lta' BALLET <contact@lta.io>
//
// This program is free software. It comes without any warranty, to the extent
// permitted by applicable law. You can redistribute it and/or modify it under
// the terms of the Do What The Fuck You Want To Public License, Version 2, as
// published by Sam Hocevar. See the COPYING file for more details.
//----------------------------------------------------------------------------

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>

#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <netinet/udp.h>
#include <arpa/inet.h>

#define NBT_UDP_PORT        138
#define NBT_TCP_PORT        139

#include "bdsm.h"
#include "bdsm/smb_trans2.h"

#include <getopt.h>

static int parse_options(int argc, char * argv[])
{
  /* *INDENT-OFF* */
  char usage_str[] = {
    "  -h, --help         Show this help screen.\n"
    "  -v, --version      Print the version and quit.\n"
  };
  /* *INDENT-ON* */

  struct option long_options[] = {
    {"help", no_argument, 0, 'h'},
    {"version", no_argument, 0, 'v'},
    {0, 0, 0, 0},
  };

  int c, opt_index = 0;

  char const *pname = ((pname = strrchr(argv[0], '/')) != NULL) ? pname + 1 : argv[0];

  while (0 < (c = getopt_long(argc, argv, "hv", long_options, &opt_index)) ) {
    switch (c) {

    case 'h':
      fprintf(stderr, "%s", usage_str);
      exit(0);

    case 'v':
      fprintf(stderr, "v%s\n", VERSION);
      exit(0);

    default:
      fprintf(stderr, "unknown option, %c, in getopt_long.\n", c);
      exit(-1);
    }
  }

  return optind;
}

int main(int ac, char **av)
{
  char const *pname = ((pname = strrchr(av[0], '/')) != NULL) ? pname + 1 : av[0];
  int const argoffset = parse_options(ac, av);

  if (argoffset >= ac) {
    fprintf(stderr, "usage: %s [options] host\n", pname);
    exit(-1);
  }

  char * hoststr = av[argoffset];

  struct sockaddr_in  addr;
  bdsm_context_t      *ctx;

  ctx = bdsm_context_new();
  assert(ctx);
  if (0 != netbios_ns_resolve(ctx->ns, hoststr, NETBIOS_FILESERVER, &addr.sin_addr.s_addr)) {
    exit(-1);
  }
  printf("%s's IP address is : %s\n", hoststr, inet_ntoa(addr.sin_addr));

  //netbios_ns_discover(ctx->ns);
  //exit(0);

  // netbios_session_t *session;
  // session = netbios_session_new(addr.sin_addr.s_addr);
  // if (netbios_session_connect(session, "Cerbere"))
  //   printf("A NetBIOS session with %s has been established\n", hoststr);
  // else
  // {
  //   printf("Unable to establish a NetBIOS session with %s\n", hoststr);
  //   exit(21);
  // }

  // netbios_session_destroy(session);

  smb_session_t *session;
  session = smb_session_new();
  if (smb_session_connect(session, hoststr, addr.sin_addr.s_addr))
    printf("Successfully connected to %s\n", hoststr);
  else
  {
    printf("Unable to connect to %s\n", hoststr);
    exit(42);
  }
  if (smb_negotiate(session))
  {
    fprintf(stderr, "Dialect/Security Mode negotation success.\n");
    fprintf(stderr, "Session key is 0x%x\n", session->srv.session_key);
    fprintf(stderr, "Challenge key is 0x%lx\n", session->srv.challenge);
  }
  else
  {
    printf("Unable to negotiate SMB Dialect\n");
    exit(42);
  }


  if (smb_authenticate(session, hoststr, av[2], av[3]))
  {
    if (session->guest)
      printf("Login FAILED but we were logged in as GUEST \n");
    else
      printf("Successfully logged in as %s\\%s\n", hoststr, av[2]);
  }
  else
  {
    printf("Authentication FAILURE.\n");
    exit(42);
  }

  smb_tid ipc  = smb_tree_connect(session, "IPC$");

  if (ipc == 0)
  {
    fprintf(stderr, "Unable to connect to IPC$ share\n");
    exit(42);
  }
  fprintf(stderr, "Connected to IPC$ share\n");

  smb_tid test = smb_tree_connect(session, "test");
  if (test)
    fprintf(stderr, "Connected to Test share\n");
  else
  {
    fprintf(stderr, "Unable to connect to Test share\n");
    exit(42);
  }

  // smb_fd fd = smb_fopen(session, test, "\\BDSM\\test.txt", SMB_MOD_RO);
  // if (fd)
  //   fprintf(stderr, "Successfully opened file: fd = 0x%.8x\n", fd);
  // else
  // {
  //   fprintf(stderr, "Unable to open file\n");
  //   exit(42);
  // }

  char              data[1024];
  smb_share_list_t  *share_list;
  smb_file_t        *files;


  // smb_fread(session, fd, data, 1024);
  // fprintf(stderr, "Read from file:\n%s\n", data);
  // smb_fclose(session, fd);

  if (!smb_share_list(session, &share_list))
  {
    fprintf(stderr, "Unable to list share for %s\n", hoststr);
    exit(42);
  }
  else
  {
    fprintf(stderr, "Share list : \n");
  }

  fprintf(stderr, "Let's find files at share's root :\n");
  files = smb_find(session, test, "\\*");
  if (files != NULL)
    while(files)
    {
      fprintf(stderr, "Found a file %s \n", files->name);
      files = files->next;
    }
  else
    fprintf(stderr, "Unable to list files\n");

  fprintf(stderr, "Query file info for path: %s\n", av[4]);
  files = smb_stat(session, test, av[4]);

  if (files)
    printf("File %s is %lu bytes long\n", av[4], files->size);


  smb_session_destroy(session);
  bdsm_context_destroy(ctx);

  return (0);
}