envlist.c

Go to the documentation of this file.

 
#include "config.h"
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include "envlist.h"
#include "memory.h"
#include "string2.h"

void envlist_free(char ***envp)
{
  if (!envp || !*envp)
    return;
 
  for (char **p = *envp; p && *p; p++)
    FREE(p);
 
  FREE(envp);
}

char **envlist_init(char **envp)
{
  if (!envp)
    return NULL;
 
  char **src = NULL;
  char **dst = NULL;
  int count = 0;
  for (src = envp; src && *src; src++)
    count++;
 
  char **env_copy = MUTT_MEM_CALLOC(count + 1, char *);
  for (src = envp, dst = env_copy; src && *src; src++, dst++)
    *dst = mutt_str_dup(*src);
 
  return env_copy;
}

bool envlist_set(char ***envp, const char *name, const char *value, bool overwrite)
{
  if (!envp || !*envp || !name || (name[0] == '\0'))
    return false;
 
  // Find a matching entry
  int count = 0;
  int match = -1;
  char *str = NULL;
  for (; (str = (*envp)[count]); count++)
  {
    size_t len = mutt_str_startswith(str, name);
    if ((len != 0) && (str[len] == '='))
    {
      if (!overwrite)
        return false;
      match = count;
      break;
    }
  }
 
  // Format var=value string dynamically to avoid truncation
  const char *val = NONULL(value);
  size_t nlen = mutt_str_len(name);
  size_t vlen = mutt_str_len(val);
  char *work = MUTT_MEM_MALLOC(nlen + 1 + vlen + 1, char);
  snprintf(work, nlen + 1 + vlen + 1, "%s=%s", name, val);
 
  if (match >= 0)
  {
    // match found, overwrite
    FREE(&(*envp)[match]);
    (*envp)[match] = work;
  }
  else
  {
    // not found, add a new entry
    MUTT_MEM_REALLOC(envp, count + 2, char *);
    (*envp)[count] = work;
    (*envp)[count + 1] = NULL;
  }
 
  return true;
}

bool envlist_unset(char ***envp, const char *name)
{
  if (!envp || !*envp || !name || (name[0] == '\0'))
    return false;
 
  int count = 0;
  for (; (*envp)[count]; count++)
    ; // do nothing
 
  char *str = NULL;
  for (int match = 0; (str = (*envp)[match]); match++)
  {
    size_t len = mutt_str_startswith(str, name);
    if ((len != 0) && (str[len] == '='))
    {
      FREE(&(*envp)[match]);
      // Move down the later entries
      memmove(&(*envp)[match], &(*envp)[match + 1], (count - match) * sizeof(char *));
      // Shrink the array
      MUTT_MEM_REALLOC(envp, count, char *);
      return true;
    }
  }
  return false;
}