/*
* Copyright (c) 2014 LastPass. All Rights Reserved.
*
*
*/
#include "cmd.h"
#include "util.h"
#include "config.h"
#include "terminal.h"
#include "kdf.h"
#include "endpoints.h"
#include "clipboard.h"
#include <getopt.h>
#include <stdio.h>
#include <string.h>
#define ALL_CHARS_LEN 94
#define NICE_CHARS_LEN 62
static char *chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890`~!@#$%^&*()-_=+[{]}\\|;:'\",<.>/?";
int cmd_generate(int argc, char **argv)
{
unsigned char key[KDF_HASH_LEN];
struct session *session = NULL;
struct blob *blob = NULL;
static struct option long_options[] = {
{"sync", required_argument, NULL, 'S'},
{"username", required_argument, NULL, 'U'},
{"url", required_argument, NULL, 'L'},
{"no-symbols", no_argument, NULL, 'X'},
{"clip", no_argument, NULL, 'c'},
{0, 0, 0, 0}
};
char option;
int option_index;
char *username = NULL;
char *url = NULL;
bool no_symbols = false;
unsigned long length;
char *name;
enum blobsync sync = BLOB_SYNC_AUTO;
_cleanup_free_ char *password = NULL;
struct account *new = NULL, *found;
struct account *notes_expansion, *notes_collapsed = NULL;
bool clip = false;
while ((option = getopt_long(argc, argv, "c", long_options, &option_index)) != -1) {
switch (option) {
case 'S':
sync = parse_sync_string(optarg);
break;
case 'U':
username = xstrdup(optarg);
break;
case 'L':
url = xstrdup(optarg);
break;
case 'X':
no_symbols = true;
break;
case 'c':
clip = true;
break;
case '?':
default:
die_usage(cmd_generate_usage);
}
}
if (argc - optind != 2)
die_usage(cmd_generate_usage);
name = argv[optind];
length = strtoul(argv[optind + 1], NULL, 10);
if (!length)
die_usage(cmd_generate_usage);
init_all(sync, key, &session, &blob);
password = xcalloc(length + 1, 1);
for (size_t i = 0; i < length; ++i)
password[i] = chars[range_rand(0, no_symbols ? NICE_CHARS_LEN : ALL_CHARS_LEN)];
found = find_unique_account(blob, name);
if (found) {
if (found->share && found->share->readonly)
die("%s is a readonly shared entry from %s. It cannot be edited.", found->fullname, found->share->name);
notes_expansion = notes_expand(found);
if (notes_expansion) {
notes_collapsed = found;
found = notes_expansion;
}
account_set_password(found, xstrdup(password), key);
if (username)
account_set_username(found, username, key);
if (url) {
free(found->url);
found->url = url;
}
if (notes_expansion && notes_collapsed) {
found = notes_collapsed;
notes_collapsed = notes_collapse(notes_expansion);
account_free(notes_expansion);
account_set_note(found, xstrdup(notes_collapsed->note), key);
account_free(notes_collapsed);
}
} else {
new = new0(struct account, 1);
new->id = xstrdup("0");
account_set_password(new, xstrdup(password), key);
account_set_fullname(new, xstrdup(name), key);
account_set_username(new, username ? username : xstrdup(""), key);
account_set_note(new, xstrdup(""), key);
new->url = url ? url : xstrdup("");
new->next = blob->account_head;
blob->account_head = new;
}
lastpass_update_account(sync, key, session, found ? found : new, blob);
blob_save(blob, key);
if (clip)
clipboard_open();
printf("%s\n", password);
session_free(session);
blob_free(blob);
return 0;
}