/*
 * Copyright (c) 1990,1991 Regents of The University of Michigan.
 * All Rights Reserved.
 *
 * Permission to use, copy, modify, and distribute this software and
 * its documentation for any purpose and without fee is hereby granted,
 * provided that the above copyright notice appears in all copies and
 * that both that copyright notice and this permission notice appear
 * in supporting documentation, and that the name of The University
 * of Michigan not be used in advertising or publicity pertaining to
 * distribution of the software without specific, written prior
 * permission. This software is supplied as is without expressed or
 * implied warranties of any kind.
 *
 *	Research Systems Unix Group
 *	The University of Michigan
 *	c/o Mike Clark
 *	535 W. William Street
 *	Ann Arbor, Michigan
 *	+1-313-763-0525
 *	netatalk@itd.umich.edu
 */

/*
 * Our own memory maintenance for atp
*/

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif /* HAVE_CONFIG_H */

#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <unistd.h>

#include <sys/types.h>
#include <sys/time.h>
#include <sys/socket.h>

#include <netatalk/at.h>
#include <atalk/atp.h>
#include "atp_internals.h"

#ifdef EBUG
#include <stdio.h>
#endif /* EBUG */

#define			N_MORE_BUFS		10

static struct atpbuf 	*free_list = NULL;	/* free buffers */

#ifdef EBUG
static int		numbufs = 0;
#endif /* EBUG */

/* only call this when the free_list is empty...
 * N_MORE_BUFS must be >= one
*/
static int more_bufs(void)
{
    int			i;
    char		*mem;
    struct atpbuf	*bp;

    /* get the whole chunk in one malloc call
    */
    if ((mem = malloc(N_MORE_BUFS * sizeof(struct atpbuf))) == NULL) {
        errno = ENOBUFS;
        return -1;
    }

    /* now split into separate bufs
    */
    bp = free_list = (struct atpbuf *) mem;

    for (i = 1; i < N_MORE_BUFS; ++i) {
        bp->atpbuf_next = (struct atpbuf *)(mem += sizeof(struct atpbuf));
        bp = bp->atpbuf_next;
    }

    bp->atpbuf_next = NULL;
    return 0;
}


#ifdef EBUG
void atp_print_bufuse(ATP ah, char *s)
{
    struct atpbuf *bp;
    int i;
    int sentcount;
    int incount;
    int respcount;
    sentcount = 0;

    for (bp = ah->atph_sent; bp != NULL; bp = bp->atpbuf_next) {
        ++sentcount;

        for (i = 0; i < 8; ++i) {
            if (bp->atpbuf_info.atpbuf_xo.atpxo_packet[i] != NULL) {
                ++sentcount;
            }
        }
    }

    if (ah->atph_reqpkt != NULL) {
        ++sentcount;
    }

    incount = 0;

    for (bp = ah->atph_queue; bp != NULL; bp = bp->atpbuf_next, ++incount);

    respcount = 0;

    for (i = 0; i < 8; ++i) {
        if (ah->atph_resppkt[i] != NULL) {
            ++respcount;
        }
    }

    printf("<%d> %s: bufs total %d  sent %d  incoming %d  req %d  resp %d\n",
           getpid(), s, numbufs, sentcount, incount,
           (ah->atph_reqpkt != NULL) ? 1 : 0, respcount);
}
#endif /* EBUG */


struct atpbuf *atp_alloc_buf(void)
{
    struct atpbuf *bp;

    if (free_list == NULL && more_bufs()) {
        return NULL;
    }

    bp = free_list;
    free_list = free_list->atpbuf_next;
#ifdef EBUG
    ++numbufs;
#endif /* EBUG */
    return bp;
}


int atp_free_buf(struct atpbuf *bp)
{
    if (bp == NULL) {
        return -1;
    }

    bp->atpbuf_next = free_list;
    free_list = bp;
#ifdef EBUG
    --numbufs;
#endif /* EBUG */
    return 0;
}


