connaccount.c

Go to the documentation of this file.

 
#include "config.h"
#include <assert.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include "mutt/lib.h"
#include "gui/lib.h"
#include "mutt.h"
#include "connaccount.h"
#include "enter/lib.h"
#include "accountcmd.h"
#include "globals.h"
 
int mutt_account_getuser(struct ConnAccount *cac)
{
  if (cac->flags & MUTT_ACCT_USER)
    return 0;
  if (!cac->get_field)
    return -1;
 
  const char *user = cac->get_field(MUTT_CA_USER, cac->gf_data);
  if (user)
  {
    mutt_str_copy(cac->user, user, sizeof(cac->user));
  }
  else if (mutt_account_call_external_cmd(cac) != MUTT_ACCT_NO_FLAGS)
  {
    /* The external command might interact with the screen */
    if (!OptNoCurses)
      mutt_need_hard_redraw();
    return 0;
  }
  else if (OptNoCurses)
  {
    return -1;
  }
  else
  {
    /* prompt (defaults to unix username), copy into cac->user */
    char prompt[256] = { 0 };
    /* L10N: Example: Username at myhost.com */
    snprintf(prompt, sizeof(prompt), _("Username at %s: "), cac->host);
    mutt_str_copy(cac->user, Username, sizeof(cac->user));
 
    struct Buffer *buf = mutt_buffer_pool_get();
    const int rc = mutt_buffer_get_field(prompt, buf, MUTT_COMP_UNBUFFERED,
                                         false, NULL, NULL, NULL);
    mutt_str_copy(cac->user, mutt_buffer_string(buf), sizeof(cac->user));
    mutt_buffer_pool_release(&buf);
    if (rc != 0)
      return -1;
  }
 
  cac->flags |= MUTT_ACCT_USER;
  return 0;
}
 
int mutt_account_getlogin(struct ConnAccount *cac)
{
  if (cac->flags & MUTT_ACCT_LOGIN)
    return 0;
  if (!cac->get_field)
    return -1;
 
  const char *login = cac->get_field(MUTT_CA_LOGIN, cac->gf_data);
  if (!login && (mutt_account_getuser(cac) == 0))
  {
    login = cac->user;
  }
 
  if (!login)
  {
    mutt_debug(LL_DEBUG1, "Couldn't get user info\n");
    return -1;
  }
 
  mutt_str_copy(cac->login, login, sizeof(cac->login));
  cac->flags |= MUTT_ACCT_LOGIN;
  return 0;
}
 
int mutt_account_getpass(struct ConnAccount *cac)
{
  if (cac->flags & MUTT_ACCT_PASS)
    return 0;
  if (!cac->get_field)
    return -1;
 
  const char *pass = cac->get_field(MUTT_CA_PASS, cac->gf_data);
  if (pass)
  {
    mutt_str_copy(cac->pass, pass, sizeof(cac->pass));
  }
  else if (mutt_account_call_external_cmd(cac) != MUTT_ACCT_NO_FLAGS)
  {
    /* The external command might interact with the screen */
    if (!OptNoCurses)
      mutt_need_hard_redraw();
    return 0;
  }
  else if (OptNoCurses)
  {
    return -1;
  }
  else
  {
    char prompt[256] = { 0 };
    snprintf(prompt, sizeof(prompt), _("Password for %s@%s: "),
             (cac->flags & MUTT_ACCT_LOGIN) ? cac->login : cac->user, cac->host);
    cac->pass[0] = '\0';
 
    struct Buffer *buf = mutt_buffer_pool_get();
    const int rc = mutt_buffer_get_field(prompt, buf, MUTT_COMP_PASS | MUTT_COMP_UNBUFFERED,
                                         false, NULL, NULL, NULL);
    mutt_str_copy(cac->pass, mutt_buffer_string(buf), sizeof(cac->pass));
    mutt_buffer_pool_release(&buf);
    if (rc != 0)
      return -1;
  }
 
  cac->flags |= MUTT_ACCT_PASS;
  return 0;
}
 
void mutt_account_unsetpass(struct ConnAccount *cac)
{
  cac->flags &= ~MUTT_ACCT_PASS;
  memset(cac->pass, 0, sizeof(cac->pass));
}
 
char *mutt_account_getoauthbearer(struct ConnAccount *cac, bool xoauth2)
{
  if (!cac || !cac->get_field)
    return NULL;
 
  /* The oauthbearer token includes the login */
  if (mutt_account_getlogin(cac))
    return NULL;
 
  const char *cmd = cac->get_field(MUTT_CA_OAUTH_CMD, cac->gf_data);
  if (!cmd)
  {
    /* L10N: You will see this error message if (1) you have "oauthbearer" in
       one of your $*_authenticators and (2) you do not have the corresponding
       $*_oauth_refresh_command defined. So the message does not mean "None of
       your $*_oauth_refresh_command's are defined." */
    mutt_error(_("No OAUTH refresh command defined"));
    return NULL;
  }
 
  FILE *fp = NULL;
  pid_t pid = filter_create(cmd, NULL, &fp, NULL);
  if (pid < 0)
  {
    mutt_perror(_("Unable to run refresh command"));
    return NULL;
  }
 
  size_t token_size = 0;
  char *token = mutt_file_read_line(NULL, &token_size, fp, NULL, MUTT_RL_NO_FLAGS);
  mutt_file_fclose(&fp);
  filter_wait(pid);
 
  /* The refresh cmd in some cases will invoke gpg to decrypt a token */
  if (!OptNoCurses)
    mutt_need_hard_redraw();
 
  if (!token || (*token == '\0'))
  {
    mutt_error(_("Command returned empty string"));
    FREE(&token);
    return NULL;
  }
 
  if ((!xoauth2 && (token_size > 512)) || (xoauth2 && (token_size > 4096)))
  {
    mutt_error(_("OAUTH token is too big: %ld"), token_size);
    FREE(&token);
    return NULL;
  }
 
  /* 4500 is chosen to allow for both a token that is 4096-long plus a
   * username that can be up to 320-long. */
  char oauthbearer[4500] = { 0 };
  int oalen = 0;
  if (xoauth2)
  {
    oalen = snprintf(oauthbearer, sizeof(oauthbearer),
                     "user=%s\001auth=Bearer %s\001\001", cac->login, token);
  }
  else
  {
    oalen = snprintf(oauthbearer, sizeof(oauthbearer),
                     "n,a=%s,\001host=%s\001port=%d\001auth=Bearer %s\001\001",
                     cac->login, cac->host, cac->port, token);
  }
  FREE(&token);
 
  size_t encoded_len = oalen * 4 / 3 + 10;
  assert(encoded_len < 6010); // Assure LGTM that we won't overflow
 
  char *encoded_token = mutt_mem_malloc(encoded_len);
  mutt_b64_encode(oauthbearer, oalen, encoded_token, encoded_len);
 
  return encoded_token;
}