set-service.c 1.93 KB
/* GNU Mailutils -- a suite of utilities for electronic mail
   Copyright (C) 2010, 2011 Free Software Foundation, Inc.

   This library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 3 of the License, or (at your option) any later version.

   This library 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
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General
   Public License along with this library.  If not, see 
   <http://www.gnu.org/licenses/>. */

#ifdef HAVE_CONFIG_H
# include <config.h>
#endif

#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <netdb.h>
#include <arpa/inet.h>
#ifdef HAVE_STRINGS_H
# include <strings.h>
#endif

#include <mailutils/types.h>
#include <mailutils/errno.h>
#include <mailutils/sys/url.h>
#include <mailutils/url.h>

int
mu_url_set_service (mu_url_t url, const char *str)
{
  unsigned port;
  char *copy;
  
  if (!url)
    return EINVAL;
  if (str)
    {
      unsigned long n;
      char *p;
      
      n = strtoul (str, &p, 10);
      if (*p)
	{
	  /* FIXME: 1. This assumes MU_URL_PARSE_PORTSRV. */
	  /* FIXME: 2. Another proto? */
	  struct servent *sp = getservbyname (str, "tcp");
	  if (!sp)
	    return MU_ERR_TCP_NO_PORT; /*FIXME: Error code?*/
	  port = ntohs (sp->s_port);
	}
      else if (n > USHRT_MAX)
	return ERANGE;

      copy = strdup (str);
      if (!copy)
	return ENOMEM;
      url->flags |= MU_URL_PORT;
    }
  else
    {
      copy = NULL;
      port = 0;
      url->flags &= ~MU_URL_PORT;
    }
  url->_get_port = NULL;
  url->_get_portstr = NULL;
  free (url->portstr);
  url->port = port;
  url->portstr = copy;
  mu_url_invalidate (url);
  return 0;
}