/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 */

#include "mt.h"
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>

char *_strpbrk_escape(char *, char *);

/*
 * _strtok_escape()
 *   Like strtok_r, except we don't break on a token if it is escaped
 *   with the escape character (\).
 */
char *
_strtok_escape(char *string, char *sepset, char **lasts)
{
        char    *r;

        /* first or subsequent call */
        if (string == NULL)
                string = *lasts;

        if (string == 0)                /* return if no tokens remaining */
                return (NULL);

        if (*string == '\0')            /* return if no tokens remaining */
                return (NULL);

        /* move past token */
        if ((r = _strpbrk_escape(string, sepset)) == NULL)
                *lasts = 0;     /* indicate this is last token */
        else {
                *r = '\0';
                *lasts = r+1;
        }
        return (string);
}

/*
 * Return ptr to first occurrence of any non-escaped character from `brkset'
 * in the character string `string'; NULL if none exists.
 */
char *
_strpbrk_escape(char *string, char *brkset)
{
        const char *p;

        do {
                for (p = brkset; *p != '\0' && *p != *string; ++p)
                        ;
                if (p == string)
                        return ((char *)string);
                if (*p != '\0') {
                        if (*(string-1) != '\\')
                                return ((char *)string);
                }
        } while (*string++);

        return (NULL);
}


char   *
_escape(char *s, char *esc)
{
        int     nescs = 0;      /* number of escapes to place in s */
        int     i, j;
        int     len_s;
        char    *tmp;

        if (s == NULL || esc == NULL)
                return (NULL);

        len_s = strlen(s);
        for (i = 0; i < len_s; i++)
                if (strchr(esc, s[i]))
                        nescs++;
        if ((tmp = malloc(nescs + len_s + 1)) == NULL)
                return (NULL);
        for (i = 0, j = 0; i < len_s; i++) {
                if (strchr(esc, s[i])) {
                        tmp[j++] = '\\';
                }
                tmp[j++] = s[i];
        }
        tmp[len_s + nescs] = '\0';
        return (tmp);
}


char *
_unescape(char *s, char *esc)
{
        int     len_s;
        int     i, j;
        char    *tmp;

        if (s == NULL || esc == NULL)
                return (NULL);

        len_s = strlen(s);
        if ((tmp = malloc(len_s + 1)) == NULL)
                return (NULL);
        for (i = 0, j = 0; i < len_s; i++) {
                if (s[i] == '\\' && strchr(esc, s[i + 1]))
                        tmp[j++] = s[++i];
                else
                        tmp[j++] = s[i];
        }
        tmp[j] = '\0';
        return (tmp);
}

char *
_strdup_null(char *s)
{
        return (strdup(s ? s : ""));
}


/*
 * read a line into buffer from a mmap'ed file.
 * return length of line read.
 */
int
_readbufline(char *mapbuf,      /* input mmap buffer */
    int mapsize,                /* input size */
    char *buffer,               /* output storage */
    int buflen,                 /* output size */
    int *lastlen)               /* input read till here last time */
{
        int     linelen;

        for (;;) {
                linelen = 0;
                while (linelen < buflen - 1) {  /* "- 1" saves room for \n\0 */
                        if (*lastlen >= mapsize) {
                                if (linelen == 0 ||
                                        buffer[linelen - 1] == '\\') {
                                                return (-1);
                                        } else {
                                                buffer[linelen] = '\n';
                                                buffer[linelen + 1] = '\0';
                                                return (linelen);
                                        }
                        }
                        switch (mapbuf[*lastlen]) {
                        case '\n':
                                (*lastlen)++;
                                if (linelen > 0 &&
                                    buffer[linelen - 1] == '\\') {
                                        --linelen;      /* remove the '\\' */
                                } else {
                                        buffer[linelen] = '\n';
                                        buffer[linelen + 1] = '\0';
                                        return (linelen);
                                }
                                break;
                        default:
                                buffer[linelen] = mapbuf[*lastlen];
                                (*lastlen)++;
                                linelen++;
                        }
                }
                /* Buffer overflow -- eat rest of line and loop again */
                while (mapbuf[*lastlen] != '\n') {
                        if (mapbuf[*lastlen] == EOF) {
                                return (-1);
                        }
                        (*lastlen)++;
                };
        }
        /* NOTREACHED */
}