gntp-send.c

Go to the documentation of this file.

#ifdef _WIN32
#include <winsock2.h>
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>

#include "md5.h"
#include "tcp.h"
#include "growl.h"

static char* string_to_utf8_alloc(const char* str) {
#ifdef _WIN32
  unsigned int codepage;
  size_t in_len = strlen(str);
  wchar_t* wcsdata;
  char* mbsdata;
  size_t mbssize, wcssize;

  codepage = GetACP();
  wcssize = MultiByteToWideChar(codepage, 0, str, in_len,  NULL, 0);
  wcsdata = (wchar_t*) malloc((wcssize + 1) * sizeof(wchar_t));
  wcssize = MultiByteToWideChar(codepage, 0, str, in_len, wcsdata, wcssize + 1);
  wcsdata[wcssize] = 0;

  mbssize = WideCharToMultiByte(CP_UTF8, 0, (LPCWSTR) wcsdata, -1, NULL, 0, NULL, NULL);
  mbsdata = (char*) malloc((mbssize + 1));
  mbssize = WideCharToMultiByte(CP_UTF8, 0, (LPCWSTR) wcsdata, -1, mbsdata, mbssize, NULL, NULL);
  mbsdata[mbssize] = 0;
  free(wcsdata);
  return mbsdata;
#else
  char *ret = malloc (strlen (str) + 1);
  if(ret !=NULL) strcpy(ret,str);
  return ret;
#endif
}

int opterr = 1;
int optind = 1;
int optopt;
char *optarg;

int getopts(int argc, char** argv, char* opts) {
  static int sp = 1;
  register int c;
  register char *cp;

  if(sp == 1) {
    if(optind >= argc ||
        argv[optind][0] != '-' || argv[optind][1] == '\0')
      return(EOF);
    else if(strcmp(argv[optind], "--") == 0) {
      optind++;
      return(EOF);
    }
  }
  optopt = c = argv[optind][sp];
  if(c == ':' || (cp=strchr(opts, c)) == NULL) {
    if(argv[optind][++sp] == '\0') {
      optind++;
      sp = 1;
    }
    return('?');
  }
  if(*++cp == ':') {
    if(argv[optind][sp+1] != '\0')
      optarg = &argv[optind++][sp+1];
    else if(++optind >= argc) {
      sp = 1;
      return('?');
    } else
      optarg = argv[optind++];
    sp = 1;
  } else {
    if(argv[optind][++sp] == '\0') {
      sp = 1;
      optind++;
    }
    optarg = NULL;
  }
  return(c);
}

int main(int argc, char* argv[]) {
  int c;
  int rc;
  char* server = NULL;
  char* password = NULL;
  char* appname = "gntp-send";
  char* notify = "gntp-send notify";
  char* title = NULL;
  char* message = NULL;
  char* icon = NULL;
  char* url = NULL;
  int tcpsend = 1;

  opterr = 0;
  while ((c = getopts(argc, argv, "a:n:s:p:u") != -1)) {
    switch (optopt) {
    case 'a': appname = optarg; break;
    case 'n': notify = optarg; break;
    case 's': server = optarg; break;
    case 'p': password = optarg; break;
    case 'u': tcpsend = 0; break;
    case '?': break;
    default: argc = 0; break;
    }
    optarg = NULL;
  }

  if ((argc - optind) < 2 || (argc - optind) > 4) {
    fprintf(stderr, "%s: [-u] [-a APPNAME] [-n NOTIFY] [-s SERVER:PORT] [-p PASSWORD] title message [icon] [url]\n", argv[0]);
    exit(1);
  }

  title = string_to_utf8_alloc((const char*)argv[optind]);
  message = string_to_utf8_alloc((const char*)argv[optind + 1]);
  if ((argc - optind) >= 3) icon = string_to_utf8_alloc((const char*)argv[optind + 2]);
  if ((argc - optind) == 4) url = string_to_utf8_alloc((const char*)argv[optind + 3]);

  if (!server) server = "127.0.0.1";

  growl_init();
  if (tcpsend) {
    rc = growl(server,appname,notify,title,message,icon,password,url);
  } else {
    rc = growl_udp(server,appname,notify,title,message,icon,password,url);
  }
  growl_shutdown();

  if (title) free(title);
  if (message) free(message);
  if (icon) free(icon);
  if (url) free(url);

  return rc;
}