NetBSD-5.0.2/dist/nvi/ipc/ip_send.c

/*	$NetBSD: ip_send.c,v 1.1.1.2 2008/05/18 14:31:24 aymeric Exp $ */

/*-
 * Copyright (c) 1996
 *	Keith Bostic.  All rights reserved.
 *
 * See the LICENSE file for redistribution information.
 */

#include "config.h"

#ifndef lint
static const char sccsid[] = "Id: ip_send.c,v 8.10 2001/06/25 15:19:25 skimo Exp (Berkeley) Date: 2001/06/25 15:19:25";
#endif /* not lint */

#include <sys/types.h>
#include <sys/queue.h>

#include <bitstring.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "../common/common.h"
#include "ip.h"

/*
 * vi_send --
 *	Construct and send an IP buffer.
 *
 * PUBLIC: int vi_send __P((int, char *, IP_BUF *));
 */
int
vi_send(int ofd, char *fmt, IP_BUF *ipbp)
{
	static char *bp;
	static size_t blen;
	size_t off;
	u_int32_t ilen;
	int nlen, n, nw;
	char *p;

	/*
	 * Have not created the channel to vi yet?  -- RAZ
	 *
	 * XXX
	 * How is that possible!?!?
	 *
	 * We'll soon find out.
	 */
	if (ofd == 0) {
		fprintf(stderr, "No channel\n");
		abort();
	}

	if (blen == 0 && (bp = malloc(blen = 512)) == NULL)
		return (1);

	p = bp;
	nlen = 0;
	*p++ = ipbp->code;
	nlen += IPO_CODE_LEN;

	if (fmt != NULL)
		for (; *fmt != '\0'; ++fmt)
			switch (*fmt) {
			case '1':				/* Value #1. */
				ilen = htonl(ipbp->val1);
				goto value;
			case '2':				/* Value #2. */
				ilen = htonl(ipbp->val2);
				goto value;
			case '3':				/* Value #3. */
				ilen = htonl(ipbp->val3);
value:				nlen += IPO_INT_LEN;
				if (nlen >= blen) {
					blen = blen * 2 + nlen;
					off = p - bp;
					if ((bp = realloc(bp, blen)) == NULL)
						return (1);
					p = bp + off;
				}
				memcpy(p, &ilen, IPO_INT_LEN);
				p += IPO_INT_LEN;
				break;
			case 'a':				/* String #1. */
			case 'b':				/* String #2. */
				ilen = *fmt == 'a' ? ipbp->len1 : ipbp->len2;
				nlen += IPO_INT_LEN + ilen;
				if (nlen >= blen) {
					blen = blen * 2 + nlen;
					off = p - bp;
					if ((bp = realloc(bp, blen)) == NULL)
						return (1);
					p = bp + off;
				}
				ilen = htonl(ilen);
				memcpy(p, &ilen, IPO_INT_LEN);
				p += IPO_INT_LEN;
				if (*fmt == 'a') {
					memcpy(p, ipbp->str1, ipbp->len1);
					p += ipbp->len1;
				} else {
					memcpy(p, ipbp->str2, ipbp->len2);
					p += ipbp->len2;
				}
				break;
			}
	for (n = p - bp, p = bp; n > 0; n -= nw, p += nw)
		if ((nw = write(ofd, p, n)) < 0)
			return (1);
	return (0);
}