probotic/src/parse.c

/* parse.c

    Probotic is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License version 3 only as
    published by the Free Software Foundation.

    Probotic is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License version 3 for more details.

    You should have received a copy of the GNU General Public License
    version 3 along with Probotic.

*/

#define PARAMS_COUNT 6

VARDECL creds_t creds =
{
  .username = NULL,
  .password = NULL,
  .channel  = NULL,
  .server   = NULL, /* localhost? */
#ifndef IRC_SSL_SUPPORT
  .port     = 6667
#else
  .port     = 6669
#endif /* IRC_SSL_SUPPORT */
};



#if 0

DECL char **
str_split(char const * s, char c)
{
  char ** ret = NULL;
  size_t i = 0;

  size_t current_token_i = 0;

  size_t token_start_i = 0;
  size_t tokens_q = 0;

  /* count tokens */
  for (i = 1; s[i]; ++i)
  {
    /* end of a token*/
    if (s[i] == c && s[i - 1] != c)
    { ++tokens_q; }
  }
  ++tokens_q;

  ret = (char **)calloc(tokens_q + 1, sizeof(char *));
  if (!ret)
  { return ret; }

  for (i = 1; s[i]; ++i)
  {

    if ((s[i + 1] == c || !s[i + 1]) && s[i] != c)
    {
      /* end of a token*/
      ret[current_token_i] = strndup(s + token_start_i, i - token_start_i + 1);
      if (!ret[current_token_i])
      {
        split_clean(ret);
        return NULL;
      }
      ++current_token_i;
    }
    else if (s[i] != c && s[i - 1] == c)
    {
      /* start of a token */
      token_start_i = i;
    }
  }

  /* Signal that the split array is ended (for iteration purposes) */
  ret[current_token_i + 1] = NULL;

  return ret;
}

DECL void
split_clean(char ** split)
{
  while (*split)
  {
    free(*split);
  }
  free(split);
}

#endif /* 0 */

DECL void
parse_command(char const * cmd)
{
  size_t i = 0;
  char* msgswp = NULL;
  /* size_t len = strlen(cmd); */
  /* TODO does not handle commands with leading space,
     use custom implemented to-spec isspace implementation */
  while (cmd[i] != '\0' &&
         cmd[i] != ' ')
  { ++i; }
  if (cmd[i] == '\0')
  {
    /* no arguments */
#ifndef NO_VULN_COMMANDS
    if (strcmp(cmd, "kill") == 0)
    { exit(1); }
#endif /* !NO_VULN_COMMANDS */
    if (strcmp(cmd, "remind") == 0)
    {
      msgswp = remind(current_username);
      ircmsg(creds.channel, "%s: %s", current_username, msgswp);
    }
    else if (strcmp(cmd, "help") == 0)
    { ircmsg(creds.channel, help_msg); }
    else if (strcmp(cmd, "magic") == 0)
    { ircmsg(creds.channel, "%s: " IRC_YELLOW "%d" IRC_STOP, current_username, (rand() % 100) + 1); }
    else if (strcmp(cmd, "dump") == 0)
    {
#ifndef NO_VULN_COMMANDS
      ircmsg(creds.channel, "%s: All projects:", current_username);
      msgswp = dump();
      ircmsg(creds.channel, msgswp);
#else
      ircmsg(creds.channel, "%s: dump disabled", current_username);
#endif /* !NO_VULN_COMMANDS */
    }
    else if (strcmp(cmd, "reroll") == 0)
    {
      /* ircmsg(creds.channel, "%s: Rerolling...", current_username); */
      purge_assignments(current_username);
      random_assign(current_username);
      ircmsg(creds.channel, "%s: %s", current_username, remind(current_username));
    }
  }
  else
  {
    /* some arguments */
    char const * const arg = cmd + i + 1;
    if (strncmp(cmd, "raw", i) == 0)
    {
#ifndef NO_VULN_COMMANDS
      /* ircmsg(creds.channel, "%s: Executing SQL `%s'.", current_username, arg); */
      msgswp = raw(arg);
      ircmsg(creds.channel, msgswp);
#else
      ircmsg(creds.channel, "%s: raw disabled", current_username);
#endif /* !NO_VULN_COMMANDS */
    }
    else if (strncmp(cmd, "repo", i) == 0)
    {
      /* ircmsg(creds.channel, "%s: Setting project repository...", current_username); */
      set_repo(creds.channel, arg);
      msgswp = remind(creds.channel);
      ircmsg(creds.channel, "%s: %s", current_username, msgswp);
    }
  }
  free(msgswp);
}

/* Parses the format username[:password]@server[:port]  */

#define GENCOPY(dst,src,l)                      \
  do                                            \
  {                                             \
    dst = malloc(sep + 1);                      \
    if (dst)                                    \
    {                                           \
      strncpy(dst,src,l);                       \
      dst[len] = '\0';                          \
    }                                           \
    else                                        \
    { return 1; }                               \
  } while (0)

DECL int
parse_url(char const * url)
{
  size_t len = strlen(url);
  size_t sep = 0, ls = 0, rs;
  while (++sep < len && url[sep] != '@');
  while (++ls  < len && url[ls] != ':')
  {
    if (ls >= sep)
    { ls = 0; break; }
  }
  rs = sep;
  while (++rs < len && url[rs] != ':');
  if (rs == len)
  { rs = 0; }
  GENCOPY(creds.username, url, ls ? ls : sep);
  if (ls)
  {
    GENCOPY(creds.password, url + ls + 1, sep - ls - 1);
  }
  if (rs)
  {
    GENCOPY(creds.server, url + sep + 1, rs - sep - 1);
    creds.port = atoi(url + rs + 1);
  }
  else
  {
    GENCOPY(creds.server, url + sep + 1, len - sep - 1);
  }
  return 0;
}

DECL void
creds_free(void)
{
  FULL_FREE(creds.username);
  FULL_FREE(creds.password);
  FULL_FREE(creds.channel);
  FULL_FREE(creds.server);
}