/*
**  server_type  - Dump the header from the web server. In most cases you'll 
**                 be able to tell the type of the web server. Some people 
**                 intentionally turn it off, for example: Yahoo.
**
**  Limitations and Comments:
**  an example of msock library.
**
**  usage: server_type http://hostname:port/
**
**    example:  server_type http://www.fccc.edu/
**              server_type http://www.fccc.edu:8080/
**
**  Development History:
**      who                  when           why
**      ma_muquit@fccc.edu   Jul-10-1999    first cut
*/

#include <msock.h>

/* proto */
static int parseURL(char *url,char *hostname,int *port,char *page);

int main (int argc,char **argv) 
{
    int
        rc,
        port,
        sock_fd;
    
    char
        szbuf[BUFSIZ],
        szhost[64],
        szpage[1024];

    if (argc < 2)
    {
        (void) fprintf(stderr,"usage: %s <URL>\n",argv[0]);
        (void) fprintf(stderr,"\nExample:\n");
        (void) fprintf(stderr," %s http://www.fccc.edu/\n",argv[0]);
        (void) fprintf(stderr," %s http://www.fccc.edu:8080/\n",
                       argv[0]);
        return(1);
    }
    rc=parseURL(argv[1],szhost,&port,szpage);
    if (rc != 0)
    {
        (void) fprintf(stderr,"Invalid URL: %s\n",argv[1]);
        return(1);
    }

    /* open a connection to the web server */
    sock_fd=ClientSocket(szhost,(u_short) port);
    if (sock_fd == -1)
    {
        (void) fprintf(stderr,"Failed to connect to %s\n",szhost);
        return (1);
    }

    /* we connected successfully, send HTTP  HEAD command*/
    (void) sprintf(szbuf,"HEAD / HTTP/1.0\r\n");
    (void) sockPuts(sock_fd,szbuf);
    (void) sockPuts(sock_fd,"User-Agent: hget1.1 by muquit\r\n");
     (void) sockPuts(sock_fd,"\r\n");


    /* cool! now read the result back from the server, and write it to stdout*/

    /* print HTTP Header came from server */
    while (sockGets(sock_fd,szbuf,sizeof(szbuf)-1))
    {
        (void) fprintf(stderr,"%s\n",szbuf);
        (void) fflush(stderr);
    }
    (void) fprintf(stderr,"\n");
    close(sock_fd);
    return(0);
}


/*
**  parseURL()
**  parse a URL of the form
**          http://host:port/thepage.html
**          http://host/thepage.html
**
**  Parameters:
**  char *url       the url to parse
**  char *hostname  the hostname (returns)
**  int  *port      the port (returns)
**  char *page      the page (returns)
**
**  Return Values:
**  0 on success
**  -1 on failure
**
**  Limitations and Comments:
**  - not much error checking. 
**  - hostname and page must have enough space preallocated by the caller.
**  - lots of possibilities of buffer overflow.
**
**  Development History:
**      who                  when           why
**      ma_muquit@fccc.edu   Jul-10-1999    first cut
*/

int parseURL(char *url,char *hostname,int *port,char *page)
{
    char 
        tmpbuf[32],
        *q,
        *r,
        *p,
        *ptmp;
    
    if (url == NULL || *url == '\0')
        return(-1);

    /* initialize */
    *port=80;
    *tmpbuf='\0';

    ptmp=strdup(url);
    if (!ptmp)
        return (-1);

    if (strlen(ptmp) < 7)
        return (-1);

    /* skip http:// */
    if ( (ptmp[0] == 'h' || ptmp[0] == 'H') &&
         (ptmp[1] == 't' || ptmp[1] == 'T') &&
         (ptmp[2] == 't' || ptmp[2] == 'T') &&
         (ptmp[3] == 'p' || ptmp[3] == 'P') &&
         (ptmp[4] == ':') &&
         (ptmp[5] == '/') &&
         (ptmp[6] == '/'))
    {
        p = ptmp+7;
    }
    else
    {
        return (-1);
    }

    q=hostname;
    r=tmpbuf;
    for (; (*p != '/') && (*p != '\0'); p++)
    {
        /* get host part out */
        if (*p != ':' && !isdigit(*p))
        {
            *q++ = *p;
        }

        /* get port out if any */
        if (isdigit(*p))
        {
            *r++ = *p;
        }
    }
    /* NULL terminate */
    *q='\0';
    *r='\0';
    if (*tmpbuf != '\0')
        *port=atoi(tmpbuf);

#ifdef 0
    /* the rest is the page */
    if (*p == '\0')
        return (-1);

    if (strcmp(p,"/") == 0)
    {
        (void) strcpy(page,"/index.html");
    }
    else
        (void) strcpy(page,p);
#endif

    return (0);
}