updater.c

/*-
 * Copyright (c) 2003-2006, Maxime Henrion <mux@FreeBSD.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

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

#include <assert.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include "config.h"
#include "diff.h"
#include "fattr.h"
#include "fixups.h"
#include "keyword.h"
#include "updater.h"
#include "misc.h"
#include "mux.h"
#include "proto.h"
#include "rcsfile.h"
#include "status.h"
#include "stream.h"

/* Internal error codes. */
#define	UPDATER_ERR_PROTO	(-1)	/* Protocol error. */
#define	UPDATER_ERR_MSG		(-2)	/* Error is in updater->errmsg. */
#define	UPDATER_ERR_READ	(-3)	/* Error reading from server. */
#define	UPDATER_ERR_DELETELIM	(-4)	/* File deletion limit exceeded. */

#define BUFSIZE 4096

/* Everything needed to update a file. */
struct file_update {
	struct statusrec srbuf;
	char *destpath;
	char *temppath;
	char *origpath;
	char *coname;		/* Points somewhere in destpath. */
	char *wantmd5;
	int isfixup;
	int free_wantmd5;	/* Indicates if we need to free the wantmd5
				   field in fup_reset(). */
	struct coll *coll;
	struct status *st;
	/* Those are only used for diff updating. */
	char *author;
	struct stream *orig;
	struct stream *to;
	int attic;
	int expand;
};

struct updater {
	struct config *config;
	struct stream *rd;
	char *errmsg;
	int deletecount;
};

static void	 fup_init(struct file_update *, struct coll *, struct status *);
static int	 fup_prepare(struct file_update *, char *, int);
static void	 fup_reset(struct file_update *);

static void	 updater_prunedirs(char *, char *);
static int	 updater_batch(struct updater *, int);
static int	 updater_docoll(struct updater *, struct file_update *, int);
static int	 updater_delete(struct updater *, struct file_update *);
static void	 updater_deletefile(const char *);
static int	 updater_checkout(struct updater *, struct file_update *);
static int	 updater_addfile(struct updater *, struct file_update *);
static int	 updater_addelta(struct rcsfile *rf, struct stream *rd,
		     char *revnum, char *diffbase, char *revdate, char *author);
static int	 updater_setattrs(struct updater *, struct file_update *,
		     char *, char *, char *, char *, char *, struct fattr *);
static int	 updater_setdirattrs(struct updater *, struct coll *,
		     struct file_update *, char *, char *);
static int	 updater_updatefile(struct updater *, struct file_update *fup,
		     const char *);
static int	 updater_updatenode(struct updater *, struct coll *,
		     struct file_update *, char *);
static int	 updater_diff(struct updater *, struct file_update *);
static int	 updater_diff_batch(struct updater *, struct file_update *);
static int	 updater_diff_apply(struct updater *, struct file_update *,
		     char *);
static void	 updater_openrcs(struct rcsfile **, struct file_update *,
		     const char *, const char *);
static int	 updater_rcsedit(struct updater *, struct file_update *, char *,
		     char *);
int		 updater_append_file(struct updater *, struct file_update *,
		     off_t);
static int	 updater_rsync(struct updater *, struct file_update *, size_t);
static int	 updater_read_checkout(struct stream *, struct stream *);

static void
fup_init(struct file_update *fup, struct coll *coll, struct status *st)
{

	memset(fup, 0, sizeof(*fup));
	fup->coll = coll;
	fup->st = st;
	fup->free_wantmd5 = 1;
}

static int
fup_prepare(struct file_update *fup, char *name, int attic)
{
	struct coll *coll;

	coll = fup->coll;
	fup->attic = 0;
	fup->origpath = NULL;
	fup->free_wantmd5 = 1;

	if (coll->co_options & CO_CHECKOUTMODE)
		fup->destpath = checkoutpath(coll->co_prefix, name);
	else {
		fup->destpath = cvspath(coll->co_prefix, name, attic);
		fup->origpath = atticpath(coll->co_prefix, name);
		/* If they're equal, we don't need special care. */
		if (fup->origpath != NULL &&
		    strcmp(fup->origpath, fup->destpath) == 0) {
			free(fup->origpath);
			fup->origpath = NULL;
		}
		fup->attic = attic;
	}
	if (fup->destpath == NULL)
		return (-1);
	fup->coname = fup->destpath + coll->co_prefixlen + 1;
	return (0);
}

/* Called after each file update to reinit the structure. */
static void
fup_reset(struct file_update *fup)
{
	struct statusrec *sr;

	sr = &fup->srbuf;

	if (fup->destpath != NULL) {
		free(fup->destpath);
		fup->destpath = NULL;
	}
	if (fup->temppath != NULL) {
		free(fup->temppath);
		fup->temppath = NULL;
	}
	if (fup->origpath != NULL) {
		free(fup->origpath);
		fup->origpath = NULL;
	}
	fup->coname = NULL;
	if (fup->author != NULL) {
		free(fup->author);
		fup->author = NULL;
	}
	fup->expand = 0;
	if (fup->wantmd5 != NULL) {
		if (fup->free_wantmd5)
			free(fup->wantmd5);
		fup->wantmd5 = NULL;
	}
	fup->free_wantmd5 = 1;
	fup->isfixup = 0;
	if (fup->orig != NULL) {
		stream_close(fup->orig);
		fup->orig = NULL;
	}
	if (fup->to != NULL) {
		stream_close(fup->to);
		fup->to = NULL;
	}
	if (sr->sr_file != NULL)
		free(sr->sr_file);
	if (sr->sr_tag != NULL)
		free(sr->sr_tag);
	if (sr->sr_date != NULL)
		free(sr->sr_date);
	if (sr->sr_revnum != NULL)
		free(sr->sr_revnum);
	if (sr->sr_revdate != NULL)
		free(sr->sr_revdate);
	fattr_free(sr->sr_clientattr);
	fattr_free(sr->sr_serverattr);
	memset(sr, 0, sizeof(*sr));
}

void *
updater(void *arg)
{
	struct thread_args *args;
	struct updater upbuf, *up;
	int error;

	args = arg;

	up = &upbuf;
	up->config = args->config;
	up->rd = args->rd;
	up->errmsg = NULL;
	up->deletecount = 0;

#ifdef UPDATER_DEBUG
	error = stream_log(up->rd, "updater.log");
	assert(!error);
#endif

	error = updater_batch(up, 0);

	/*
	 * Make sure to close the fixups even in case of an error,
	 * so that the detailer thread doesn't block indefinitely.
	 */
	fixups_close(up->config->fixups);
	if (!error)
		error = updater_batch(up, 1);
	switch (error) {
	case UPDATER_ERR_PROTO:
		xasprintf(&args->errmsg, "Updater failed: Protocol error");
		args->status = STATUS_FAILURE;
		break;
	case UPDATER_ERR_MSG:
		xasprintf(&args->errmsg, "Updater failed: %s", up->errmsg);
		free(up->errmsg);
		args->status = STATUS_FAILURE;
		break;
	case UPDATER_ERR_READ:
		if (stream_eof(up->rd)) {
			xasprintf(&args->errmsg, "Updater failed: "
			    "Premature EOF from server");
		} else {
			xasprintf(&args->errmsg, "Updater failed: "
			    "Network read failure: %s", strerror(errno));
		}
		args->status = STATUS_TRANSIENTFAILURE;
		break;
	case UPDATER_ERR_DELETELIM:
		xasprintf(&args->errmsg, "Updater failed: "
		    "File deletion limit exceeded");
		args->status = STATUS_FAILURE;
		break;
	default:
		assert(error == 0);
		args->status = STATUS_SUCCESS;
	};
	return (NULL);
}

static int
updater_batch(struct updater *up, int isfixups)
{
	struct stream *rd;
	struct coll *coll;
	struct status *st;
	struct file_update fup;
	char *line, *cmd, *errmsg, *collname, *release;
	int error;

	rd = up->rd;
	STAILQ_FOREACH(coll, &up->config->colls, co_next) {
		if (coll->co_options & CO_SKIP)
			continue;
		umask(coll->co_umask);
		line = stream_getln(rd, NULL);
		if (line == NULL)
			return (UPDATER_ERR_READ);
		cmd = proto_get_ascii(&line);
		collname = proto_get_ascii(&line);
		release = proto_get_ascii(&line);
		if (release == NULL || line != NULL)
			return (UPDATER_ERR_PROTO);
		if (strcmp(cmd, "COLL") != 0 ||
		    strcmp(collname, coll->co_name) != 0 ||
		    strcmp(release, coll->co_release) != 0)
			return (UPDATER_ERR_PROTO);

		if (!isfixups)
			lprintf(1, "Updating collection %s/%s\n", coll->co_name,
			    coll->co_release);

		if (coll->co_options & CO_COMPRESS)
			stream_filter_start(rd, STREAM_FILTER_ZLIB, NULL);

		st = status_open(coll, coll->co_scantime, &errmsg);
		if (st == NULL) {
			up->errmsg = errmsg;
			return (UPDATER_ERR_MSG);
		}
		fup_init(&fup, coll, st);
		error = updater_docoll(up, &fup, isfixups);
		status_close(st, &errmsg);
		fup_reset(&fup);
		if (errmsg != NULL) {
			/* Discard previous error. */
			if (up->errmsg != NULL)
				free(up->errmsg);
			up->errmsg = errmsg;
			return (UPDATER_ERR_MSG);
		}
		if (error)
			return (error);

		if (coll->co_options & CO_COMPRESS)
			stream_filter_stop(rd);
	}
	line = stream_getln(rd, NULL);
	if (line == NULL)
		return (UPDATER_ERR_READ);
	if (strcmp(line, ".") != 0)
		return (UPDATER_ERR_PROTO);
	return (0);
}

static int
updater_docoll(struct updater *up, struct file_update *fup, int isfixups)
{
	struct stream *rd;
	struct coll *coll;
	struct statusrec srbuf, *sr;
	struct fattr *rcsattr, *tmp;
	char *attr, *blocksize, *line, *msg;
	char *name, *tag, *date, *revdate;
	char *expand, *wantmd5, *revnum;
	char *rcsopt, *pos;
	time_t t;
	off_t position;
	int attic, cmd, error, needfixupmsg;

	cmd = -1;
	error = 0;
	rd = up->rd;
	coll = fup->coll;
	needfixupmsg = isfixups;
	while ((line = stream_getln(rd, NULL)) != NULL) {
		cmd = proto_get_char(&line);
		if (cmd == '.' && line == NULL)
			break;
		memset(&srbuf, 0, sizeof(srbuf));
		if (needfixupmsg) {
			lprintf(1, "Applying fixups for collection %s/%s\n",
			    coll->co_name, coll->co_release);
			needfixupmsg = 0;
		}
		switch (cmd) {
		case 'T':
			/* Update recorded information for checked-out file. */
			name = proto_get_ascii(&line);
			tag = proto_get_ascii(&line);
			date = proto_get_ascii(&line);
			revnum = proto_get_ascii(&line);
			revdate = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);

			rcsattr = fattr_decode(attr);
			if (rcsattr == NULL)
				return (UPDATER_ERR_PROTO);

			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			error = updater_setattrs(up, fup, name, tag, date,
			    revnum, revdate, rcsattr);
			fattr_free(rcsattr);
			if (error)
				return (error);
			break;
		case 'c':
			/* Checkout dead file. */
			name = proto_get_ascii(&line);
			tag = proto_get_ascii(&line);
			date = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);

			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			/* Theoritically, the file does not exist on the client.
			   Just to make sure, we'll delete it here, if it
			   exists. */
			if (access(fup->destpath, F_OK) == 0) {
				error = updater_delete(up, fup);
				if (error)
					return (error);
			}

			sr = &srbuf;
			sr->sr_type = SR_CHECKOUTDEAD;
			sr->sr_file = name;
			sr->sr_tag = tag;
			sr->sr_date = date;
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);

			error = status_put(fup->st, sr);
			fattr_free(sr->sr_serverattr);
			if (error) {
				up->errmsg = status_errmsg(fup->st);
				return (UPDATER_ERR_MSG);
			}
			break;
		case 'U':
			/* Update live checked-out file. */
			name = proto_get_ascii(&line);
			tag = proto_get_ascii(&line);
			date = proto_get_ascii(&line);
			proto_get_ascii(&line);	/* XXX - oldRevNum */
			proto_get_ascii(&line);	/* XXX - fromAttic */
			proto_get_ascii(&line);	/* XXX - logLines */
			expand = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			wantmd5 = proto_get_ascii(&line);
			if (wantmd5 == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);

			sr = &fup->srbuf;
			sr->sr_type = SR_CHECKOUTLIVE;
			sr->sr_file = xstrdup(name);
			sr->sr_date = xstrdup(date);
			sr->sr_tag = xstrdup(tag);
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);

			fup->expand = keyword_decode_expand(expand);
			if (fup->expand == -1)
				return (UPDATER_ERR_PROTO);
			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);

			fup->wantmd5 = xstrdup(wantmd5);
			fup->temppath = tempname(fup->destpath);
			error = updater_diff(up, fup);
			if (error)
				return (error);
			break;
		case 'u':
			/* Update dead checked-out file. */
			name = proto_get_ascii(&line);
			tag = proto_get_ascii(&line);
			date = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);

			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			error = updater_delete(up, fup);
			if (error)
				return (error);
			sr = &srbuf;
			sr->sr_type = SR_CHECKOUTDEAD;
			sr->sr_file = name;
			sr->sr_tag = tag;
			sr->sr_date = date;
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);
			error = status_put(fup->st, sr);
			fattr_free(sr->sr_serverattr);
			if (error) {
				up->errmsg = status_errmsg(fup->st);
				return (UPDATER_ERR_MSG);
			}
			break;
		case 'C':	/* Checkout file. */
		case 'Y':	/* Receive checkout-mode fixup. */
			name = proto_get_ascii(&line);
			tag = proto_get_ascii(&line);
			date = proto_get_ascii(&line);
			revnum = proto_get_ascii(&line);
			revdate = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);

			sr = &fup->srbuf;
			sr->sr_type = SR_CHECKOUTLIVE;
			sr->sr_file = xstrdup(name);
			sr->sr_tag = xstrdup(tag);
			sr->sr_date = xstrdup(date);
			sr->sr_revnum = xstrdup(revnum);
			sr->sr_revdate = xstrdup(revdate);
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);

			t = rcsdatetotime(revdate);
			if (t == -1)
				return (UPDATER_ERR_PROTO);

			sr->sr_clientattr = fattr_new(FT_FILE, t);
			tmp = fattr_forcheckout(sr->sr_serverattr,
			    coll->co_umask);
			fattr_override(sr->sr_clientattr, tmp, FA_MASK);
			fattr_free(tmp);
			fattr_mergedefault(sr->sr_clientattr);
			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			fup->temppath = tempname(fup->destpath);
			fup->isfixup = cmd == 'Y';
			error = updater_checkout(up, fup);
			if (error)
				return (error);
			break;
		case 'D':
			/* Delete file. */
			name = proto_get_ascii(&line);
			if (name == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			error = updater_delete(up, fup);
			if (error)
				return (error);
			error = status_delete(fup->st, name, 0);
			if (error) {
				up->errmsg = status_errmsg(fup->st);
				return (UPDATER_ERR_MSG);
			}
			break;
		case 'A':
		case 'a':
		case 'R':
			/* Add/replace regular file. */
			name = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			attic = cmd == 'a';
			error = fup_prepare(fup, name, attic);
			if (error)
				return (UPDATER_ERR_PROTO);

			fup->temppath = tempname(fup->destpath);
			sr = &fup->srbuf;
			sr->sr_type = attic ? SR_FILEDEAD : SR_FILELIVE;
			sr->sr_file = xstrdup(name);
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);
			if (attic)
				lprintf(1, " Create %s -> Attic\n", name);
			else
				lprintf(1, " Create %s\n", name);
			error = updater_addfile(up, fup);
			if (error)
				return (error);
			break;
		case 'r':
			/* Update regular file with rsync algorithm. */
			name = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			blocksize = proto_get_ascii(&line);
			wantmd5 = proto_get_ascii(&line);
			if (wantmd5 == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			fup->wantmd5 = xstrdup(wantmd5);
			fup->temppath = tempname(fup->destpath);
			sr = &fup->srbuf;
			sr->sr_file = xstrdup(name);
			sr->sr_serverattr = fattr_decode(attr);
			sr->sr_type = SR_FILELIVE;
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);
			error = updater_rsync(up, fup, strtol(blocksize, NULL,
			    10));
			if (error)
				return (error);
			break;
		case 'I':
			/* Create directory and add DirDown entry in status
			   file. */
			name = proto_get_ascii(&line);
			if (name == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			sr = &fup->srbuf;
			sr->sr_type = SR_DIRDOWN;
			sr->sr_file = xstrdup(name);
			sr->sr_serverattr = NULL;
			sr->sr_clientattr = fattr_new(FT_DIRECTORY, -1);
			fattr_mergedefault(sr->sr_clientattr);

			error = mkdirhier(fup->destpath, coll->co_umask);
			if (error)
				return (UPDATER_ERR_PROTO);
			if (access(fup->destpath, F_OK) != 0) {
				lprintf(1, " Mkdir %s\n", name);
				error = fattr_makenode(sr->sr_clientattr,
				    fup->destpath);
				if (error)
					return (UPDATER_ERR_PROTO);
			}
			error = status_put(fup->st, sr);
			if (error) {
				up->errmsg = status_errmsg(fup->st);
				return (UPDATER_ERR_MSG);
			}
			break;
		case 'i':
			/* Remove DirDown entry in status file. */
			name = proto_get_ascii(&line);
			if (name == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			error = status_delete(fup->st, name, 0);
			if (error) {
				up->errmsg = status_errmsg(fup->st);
				return (UPDATER_ERR_MSG);
			}
			break;
		case 'J':
			/* Set attributes of directory and update DirUp entry
			   in status file. */
			name = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			error = updater_setdirattrs(up, coll, fup, name, attr);
			if (error)
				return (error);
			break;
		case 'j':
			/* Remove directory and delete its DirUp entry in
			   status file. */
			name = proto_get_ascii(&line);
			if (name == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			error = fup_prepare(fup, name, 0);
			if (error)
				return (UPDATER_ERR_PROTO);
			lprintf(1, " Rmdir %s\n", name);
			updater_deletefile(fup->destpath);
			error = status_delete(fup->st, name, 0);
			if (error) {
				up->errmsg = status_errmsg(fup->st);
				return (UPDATER_ERR_MSG);
			}
			break;
		case 'L':
		case 'l':
			/* Update recorded information for cvs file. */
			name = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			attic = cmd == 'l';
			sr = &fup->srbuf;
			sr->sr_type = attic ? SR_FILEDEAD : SR_FILELIVE;
			sr->sr_file = xstrdup(name);
			sr->sr_serverattr = fattr_decode(attr);
			sr->sr_clientattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL ||
			    sr->sr_clientattr == NULL)
				return (UPDATER_ERR_PROTO);

			/* Save space. Described in detail in updatefile. */
			if (!(fattr_getmask(sr->sr_clientattr) & FA_LINKCOUNT)
			    || fattr_getlinkcount(sr->sr_clientattr) <= 1)
				fattr_maskout(sr->sr_clientattr,
				    FA_DEV | FA_INODE);
			fattr_maskout(sr->sr_clientattr, FA_FLAGS);
			error = status_put(fup->st, sr);
			if (error) {
				up->errmsg = status_errmsg(fup->st);
				return (UPDATER_ERR_MSG);
			}
			break;
		case 'N':
		case 'n':
			/* Create a node. */
			name = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			attic = cmd == 'n';
			error = fup_prepare(fup, name, attic);
			if (error)
				return (UPDATER_ERR_PROTO);
			sr = &fup->srbuf;
			sr->sr_type = (attic ? SR_FILEDEAD : SR_FILELIVE);
			sr->sr_file = xstrdup(name);
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);
			sr->sr_clientattr = fattr_new(FT_SYMLINK, -1);
			fattr_mergedefault(sr->sr_clientattr);
			fattr_maskout(sr->sr_clientattr, FA_FLAGS);
			error = updater_updatenode(up, coll, fup, name);
			if (error)
				return (error);
			break;
		case 'V':
		case 'v':
			/* Edit RCS file. */
			name = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			rcsopt = proto_get_ascii(&line);
			wantmd5 = proto_get_ascii(&line);
			if (wantmd5 == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			attic = cmd == 'v';
			error = fup_prepare(fup, name, attic);
			if (error)
				return (UPDATER_ERR_PROTO);
			fup->temppath = tempname(fup->destpath);
			fup->wantmd5 = xstrdup(wantmd5);
			sr = &fup->srbuf;
			sr->sr_type = attic ? SR_FILEDEAD : SR_FILELIVE;
			sr->sr_file = xstrdup(name);
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);

			error = updater_rcsedit(up, fup, name, rcsopt);
			if (error)
				return (error);
			break;
		case 'X':
		case 'x':
			/* Receive RCS file fixup. */
			name = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			if (attr == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			attic = cmd == 'x';
			error = fup_prepare(fup, name, attic);
			if (error)
				return (UPDATER_ERR_PROTO);

			fup->temppath = tempname(fup->destpath);
			fup->isfixup = 1;
			sr = &fup->srbuf;
			sr->sr_type = attic ? SR_FILEDEAD : SR_FILELIVE;
			sr->sr_file = xstrdup(name);
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);
			lprintf(1, " Fixup %s\n", name);
			error = updater_addfile(up, fup);
			if (error)
				return (error);
			break;
		case 'Z':
			/* Append to regular file. */
			name = proto_get_ascii(&line);
			attr = proto_get_ascii(&line);
			pos  = proto_get_ascii(&line);
			if (pos == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			error = fup_prepare(fup, name, 0);
			fup->temppath = tempname(fup->destpath);
			sr = &fup->srbuf;
			sr->sr_type = SR_FILELIVE;
			sr->sr_file = xstrdup(name);
			sr->sr_serverattr = fattr_decode(attr);
			if (sr->sr_serverattr == NULL)
				return (UPDATER_ERR_PROTO);
			position = strtol(pos, NULL, 10);
			lprintf(1, " Append to %s\n", name);
			error = updater_append_file(up, fup, position);
			if (error)
				return (error);
			break;
		case '!':
			/* Warning from server. */
			msg = proto_get_rest(&line);
			if (msg == NULL)
				return (UPDATER_ERR_PROTO);
			lprintf(-1, "Server warning: %s\n", msg);
			break;
		default:
			return (UPDATER_ERR_PROTO);
		}
		fup_reset(fup);
	}
	if (line == NULL && cmd != '.')
		return (UPDATER_ERR_READ);
	return (0);
}

/* Delete file. */
static int
updater_delete(struct updater *up, struct file_update *fup)
{
	struct config *config;
	struct coll *coll;

	config = up->config;
	coll = fup->coll;
	if (coll->co_options & CO_DELETE) {
		lprintf(1, " Delete %s\n", fup->coname);
		if (config->deletelim >= 0 &&
		    up->deletecount >= config->deletelim)
			return (UPDATER_ERR_DELETELIM);
		up->deletecount++;
		updater_deletefile(fup->destpath);
		if (coll->co_options & CO_CHECKOUTMODE)
			updater_prunedirs(coll->co_prefix, fup->destpath);
	} else {
		lprintf(1," NoDelete %s\n", fup->coname);
	}
	return (0);
}

static void
updater_deletefile(const char *path)
{
	int error;

	error = fattr_delete(path);
	if (error && errno != ENOENT) {
		lprintf(-1, "Cannot delete \"%s\": %s\n",
		    path, strerror(errno));
	}
}

static int
updater_setattrs(struct updater *up, struct file_update *fup, char *name,
    char *tag, char *date, char *revnum, char *revdate, struct fattr *rcsattr)
{
	struct statusrec sr;
	struct status *st;
	struct coll *coll;
	struct fattr *fileattr, *fa;
	char *path;
	int error, rv;

	coll = fup->coll;
	st = fup->st;
	path = fup->destpath;

	fileattr = fattr_frompath(path, FATTR_NOFOLLOW);
	if (fileattr == NULL) {
		/* The file has vanished. */
		error = status_delete(st, name, 0);
		if (error) {
			up->errmsg = status_errmsg(st);
			return (UPDATER_ERR_MSG);
		}
		return (0);
	}
	fa = fattr_forcheckout(rcsattr, coll->co_umask);
	fattr_override(fileattr, fa, FA_MASK);
	fattr_free(fa);

	rv = fattr_install(fileattr, path, NULL);
	if (rv == -1) {
		lprintf(1, " SetAttrs %s\n", fup->coname);
		fattr_free(fileattr);
		xasprintf(&up->errmsg, "Cannot set attributes for \"%s\": %s",
		    path, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	if (rv == 1) {
		lprintf(1, " SetAttrs %s\n", fup->coname);
		fattr_free(fileattr);
		fileattr = fattr_frompath(path, FATTR_NOFOLLOW);
		if (fileattr == NULL) {
			/* We're being very unlucky. */
			error = status_delete(st, name, 0);
			if (error) {
				up->errmsg = status_errmsg(st);
				return (UPDATER_ERR_MSG);
			}
			return (0);
		}
	}

	fattr_maskout(fileattr, FA_COIGNORE);

	sr.sr_type = SR_CHECKOUTLIVE;
	sr.sr_file = name;
	sr.sr_tag = tag;
	sr.sr_date = date;
	sr.sr_revnum = revnum;
	sr.sr_revdate = revdate;
	sr.sr_clientattr = fileattr;
	sr.sr_serverattr = rcsattr;

	error = status_put(st, &sr);
	fattr_free(fileattr);
	if (error) {
		up->errmsg = status_errmsg(st);
		return (UPDATER_ERR_MSG);
	}
	return (0);
}

static void
updater_needfixup(struct updater *up, struct file_update *fup, const char *msg)
{
	struct statusrec *sr;
	struct coll *coll;

	coll = fup->coll;
	sr = &fup->srbuf;
	lprintf(-1, "%s: %s -- ", fup->coname, msg);
	if (fup->isfixup) {
		lprintf(-1, "file not updated\n");
	} else {
		lprintf(-1, "will transfer entire file\n");
		fixups_put(up->config->fixups, fup->coll, sr->sr_file);
	}
	if (coll->co_options & CO_KEEPBADFILES)
		lprintf(-1, "Bad version saved in %s\n", fup->temppath);
	else
		updater_deletefile(fup->temppath);
}

static int
updater_updatefile(struct updater *up, struct file_update *fup, const char *md5)
{
	struct coll *coll;
	struct status *st;
	struct statusrec *sr;
	struct fattr *fileattr;
	int error, rv;

	coll = fup->coll;
	sr = &fup->srbuf;
	st = fup->st;

	if (strcmp(fup->wantmd5, md5) != 0) {
		updater_needfixup(up, fup, "Checksum mismatch");
		return (0);
	}

	fattr_umask(sr->sr_clientattr, coll->co_umask);
	rv = fattr_install(sr->sr_clientattr, fup->destpath, fup->temppath);
	if (rv == -1) {
		xasprintf(&up->errmsg, "Cannot install \"%s\" to \"%s\": %s",
		    fup->temppath, fup->destpath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}

	/* XXX Executes */
	/*
	 * We weren't necessarily able to set all the file attributes to the
	 * desired values, and any executes may have altered the attributes.
	 * To make sure we record the actual attribute values, we fetch
	 * them from the file.
	 *
	 * However, we preserve the link count as received from the
	 * server.  This is important for preserving hard links in mirror
	 * mode.
	 */
	fileattr = fattr_frompath(fup->destpath, FATTR_NOFOLLOW);
	if (fileattr == NULL) {
		xasprintf(&up->errmsg, "Cannot stat \"%s\": %s", fup->destpath,
		    strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	fattr_override(fileattr, sr->sr_clientattr, FA_LINKCOUNT);
	fattr_free(sr->sr_clientattr);
	sr->sr_clientattr = fileattr;

	/*
	 * To save space, don't write out the device and inode unless
	 * the link count is greater than 1.  These attributes are used
	 * only for detecting hard links.  If the link count is 1 then we
	 * know there aren't any hard links.
	 */
	if (!(fattr_getmask(sr->sr_clientattr) & FA_LINKCOUNT) ||
	    fattr_getlinkcount(sr->sr_clientattr) <= 1)
		fattr_maskout(sr->sr_clientattr, FA_DEV | FA_INODE);

	if (coll->co_options & CO_CHECKOUTMODE)
		fattr_maskout(sr->sr_clientattr, FA_COIGNORE);

	error = status_put(st, sr);
	if (error) {
		up->errmsg = status_errmsg(st);
		return (UPDATER_ERR_MSG);
	}
	return (0);
}

/*
 * Update attributes of a directory.
 */
static int
updater_setdirattrs(struct updater *up, struct coll *coll,
    struct file_update *fup, char *name, char *attr)
{
	struct statusrec *sr;
	struct fattr *fa;
	int error, rv;

	sr = &fup->srbuf;
	sr->sr_type = SR_DIRUP;
	sr->sr_file = xstrdup(name);
	sr->sr_clientattr = fattr_decode(attr);
	sr->sr_serverattr = fattr_decode(attr);
	if (sr->sr_clientattr == NULL || sr->sr_serverattr == NULL)
		return (UPDATER_ERR_PROTO);
	fattr_mergedefault(sr->sr_clientattr);
	fattr_umask(sr->sr_clientattr, coll->co_umask);
	rv = fattr_install(sr->sr_clientattr, fup->destpath, NULL);
	lprintf(1, " SetAttrs %s\n", name);
	if (rv == -1) {
		xasprintf(&up->errmsg, "Cannot install \"%s\" to \"%s\": %s",
		    fup->temppath, fup->destpath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	/*
	 * Now, make sure they were set and record what was set in the status
	 * file.
	 */
	fa = fattr_frompath(fup->destpath, FATTR_NOFOLLOW);
	if (fa == NULL) {
		xasprintf(&up->errmsg, "Cannot open \%s\": %s", fup->destpath,
		    strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	fattr_free(sr->sr_clientattr);
	fattr_maskout(fa, FA_FLAGS);
	sr->sr_clientattr = fa;
	error = status_put(fup->st, sr);
	if (error) {
		up->errmsg = status_errmsg(fup->st);
		return (UPDATER_ERR_MSG);
	}

	return (0);
}

static int
updater_diff(struct updater *up, struct file_update *fup)
{
	char md5[MD5_DIGEST_SIZE];
	struct coll *coll;
	struct statusrec *sr;
	struct fattr *fa, *tmp;
	char *line, *author, *path, *revnum, *revdate;
	int error, cmd;

	coll = fup->coll;
	sr = &fup->srbuf;
	path = fup->destpath;

	cmd = -1;
	lprintf(1, " Edit %s\n", fup->coname);
	while ((line = stream_getln(up->rd, NULL)) != NULL) {
		cmd = proto_get_char(&line);
		if (cmd == '.' && line == NULL)
			break;
		if (cmd != 'D')
			return (UPDATER_ERR_PROTO);
		revnum = proto_get_ascii(&line);
		proto_get_ascii(&line); /* XXX - diffbase */
		revdate = proto_get_ascii(&line);
		author = proto_get_ascii(&line);
		if (author == NULL || line != NULL)
			return (UPDATER_ERR_PROTO);
		if (sr->sr_revnum != NULL)
			free(sr->sr_revnum);
		if (sr->sr_revdate != NULL)
			free(sr->sr_revdate);
		if (fup->author != NULL)
			free(fup->author);
		sr->sr_revnum = xstrdup(revnum);
		sr->sr_revdate = xstrdup(revdate);
		fup->author = xstrdup(author);
		if (fup->orig == NULL) {
			/* First patch, the "origin" file is the one we have. */
			fup->orig = stream_open_file(path, O_RDONLY);
			if (fup->orig == NULL) {
				xasprintf(&up->errmsg, "%s: Cannot open: %s",
				    path, strerror(errno));
				return (UPDATER_ERR_MSG);
			}
		} else {
			/* Subsequent patches. */
			stream_close(fup->orig);
			fup->orig = fup->to;
			stream_rewind(fup->orig);
			unlink(fup->temppath);
			free(fup->temppath);
			fup->temppath = tempname(path);
		}
		fup->to = stream_open_file(fup->temppath,
		    O_RDWR | O_CREAT | O_TRUNC, 0600);
		if (fup->to == NULL) {
			xasprintf(&up->errmsg, "%s: Cannot open: %s",
			    fup->temppath, strerror(errno));
			return (UPDATER_ERR_MSG);
		}
		lprintf(2, "  Add delta %s %s %s\n", sr->sr_revnum,
		    sr->sr_revdate, fup->author);
		error = updater_diff_batch(up, fup);
		if (error)
			return (error);
	}
	if (line == NULL && cmd != '.')
		return (UPDATER_ERR_READ);

	fa = fattr_frompath(path, FATTR_FOLLOW);
	tmp = fattr_forcheckout(sr->sr_serverattr, coll->co_umask);
	fattr_override(fa, tmp, FA_MASK);
	fattr_free(tmp);
	fattr_maskout(fa, FA_MODTIME);
	sr->sr_clientattr = fa;

	if (MD5_File(fup->temppath, md5, NULL) == -1) {
		xasprintf(&up->errmsg,
		    "Cannot calculate checksum for \"%s\": %s",
		    path, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	error = updater_updatefile(up, fup, md5);
	return (error);
}

/*
 * Edit a file and add delta.
 */
static int
updater_diff_batch(struct updater *up, struct file_update *fup)
{
	struct stream *rd;
	char *line, *state, *tok;
	int cmd, error;

	cmd = -1;
	state = NULL;
	rd = up->rd;
	while ((line = stream_getln(rd, NULL)) != NULL) {
		cmd = proto_get_char(&line);
		if (cmd == -1)
			return (UPDATER_ERR_PROTO);
		if (cmd == '.' && line == NULL)
			break;
		switch (cmd) {
		case 'L':
			line = stream_getln(rd, NULL);
			/* XXX - We're just eating the log for now. */
			while (line != NULL && strcmp(line, ".") != 0 &&
			    strcmp(line, ".+") != 0)
				line = stream_getln(rd, NULL);
			if (line == NULL) {
				error = UPDATER_ERR_READ;
				goto bad;
			}
			break;
		case 'S':
			tok = proto_get_ascii(&line);
			if (tok == NULL || line != NULL) {
				error = UPDATER_ERR_PROTO;
				goto bad;
			}
			if (state != NULL)
				free(state);
			state = xstrdup(tok);
			break;
		case 'T':
			error = updater_diff_apply(up, fup, state);
			if (error)
				goto bad;
			break;
		default:
			error = UPDATER_ERR_PROTO;
			goto bad;
		}
	}
	if (line == NULL && cmd != '.') {
		error = UPDATER_ERR_READ;
		goto bad;
	}
	if (state != NULL)
		free(state);
	return (0);
bad:
	if (state != NULL)
		free(state);
	return (error);
}

int
updater_diff_apply(struct updater *up, struct file_update *fup, char *state)
{
	struct diffinfo dibuf, *di;
	struct coll *coll;
	struct statusrec *sr;
	int error;

	coll = fup->coll;
	sr = &fup->srbuf;
	di = &dibuf;

	di->di_rcsfile = sr->sr_file;
	di->di_cvsroot = coll->co_cvsroot;
	di->di_revnum = sr->sr_revnum;
	di->di_revdate = sr->sr_revdate;
	di->di_author = fup->author;
	di->di_tag = sr->sr_tag;
	di->di_state = state;
	di->di_expand = fup->expand;

	error = diff_apply(up->rd, fup->orig, fup->to, coll->co_keyword, di, 1);
	if (error) {
		/* XXX Bad error message */
		xasprintf(&up->errmsg, "Bad diff from server");
		return (UPDATER_ERR_MSG);
	}
	return (0);
}

/* Update or create a node. */
static int
updater_updatenode(struct updater *up, struct coll *coll,
    struct file_update *fup, char *name)
{
	struct fattr *fileattr;
	struct status *st;
	struct statusrec *sr;
	int error, rv;

	sr = &fup->srbuf;
	st = fup->st;

	if (fattr_type(sr->sr_serverattr) == FT_SYMLINK) {
		lprintf(1, " Symlink %s -> %s\n", name,
		    fattr_getlinktarget(sr->sr_serverattr));
	} else {
		lprintf(1, " Mknod %s\n", name);
	}

	/* Create directory. */
	error = mkdirhier(fup->destpath, coll->co_umask);
	if (error)
		return (UPDATER_ERR_PROTO);

	/* If it does not exist, create it. */
	if (access(fup->destpath, F_OK) != 0)
		fattr_makenode(sr->sr_serverattr, fup->destpath);

	/*
	 * Coming from attic? I don't think this is a problem since we have
	 * determined attic before we call this function (Look at UpdateNode in
	 * cvsup).
	 */
	fattr_umask(sr->sr_serverattr, coll->co_umask);
	rv = fattr_install(sr->sr_serverattr, fup->destpath, fup->temppath);
	if (rv == -1) {
		xasprintf(&up->errmsg, "Cannot update attributes on "
	    "\"%s\": %s", fup->destpath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	/*
	 * XXX: Executes not implemented. Have not encountered much use for it
	 * yet.
	 */
	/*
	 * We weren't necessarily able to set all the file attributes to the
	 * desired values, and any executes may have altered the attributes.
	 * To make sure we record the actual attribute values, we fetch
	 * them from the file.
	 *
	 * However, we preserve the link count as received from the
	 * server.  This is important for preserving hard links in mirror
	 * mode.
	 */
	fileattr = fattr_frompath(fup->destpath, FATTR_NOFOLLOW);
	if (fileattr == NULL) {
		xasprintf(&up->errmsg, "Cannot stat \"%s\": %s", fup->destpath,
		    strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	fattr_override(fileattr, sr->sr_clientattr, FA_LINKCOUNT);
	fattr_free(sr->sr_clientattr);
	sr->sr_clientattr = fileattr;

	/*
	 * To save space, don't write out the device and inode unless
	 * the link count is greater than 1.  These attributes are used
	 * only for detecting hard links.  If the link count is 1 then we
	 * know there aren't any hard links.
	 */
	if (!(fattr_getmask(sr->sr_clientattr) & FA_LINKCOUNT) ||
	    fattr_getlinkcount(sr->sr_clientattr) <= 1)
		fattr_maskout(sr->sr_clientattr, FA_DEV | FA_INODE);

	/* If it is a symlink, write only out it's path. */
	if (fattr_type(sr->sr_serverattr) == FT_SYMLINK) {
		fattr_maskout(sr->sr_clientattr, ~(FA_FILETYPE |
		    FA_LINKTARGET));
	}
	fattr_maskout(sr->sr_clientattr, FA_FLAGS);
	error = status_put(st, sr);
	if (error) {
		up->errmsg = status_errmsg(st);
		return (UPDATER_ERR_MSG);
	}
	return (0);
}

/*
 * Fetches a new file in CVS mode.
 */
static int
updater_addfile(struct updater *up, struct file_update *fup)
{
	struct coll *coll;
	struct stream *to;
	struct statusrec *sr;
	char buf[BUFSIZE];
	char md5[MD5_DIGEST_SIZE];
	ssize_t nread;
	off_t fsize, remains;
	char *line, *path;
	int cmd, error;

	coll = fup->coll;
	path = fup->destpath;
	sr = &fup->srbuf;
	fsize = fattr_filesize(sr->sr_serverattr);

	error = mkdirhier(path, coll->co_umask);
	if (error)
		return (UPDATER_ERR_PROTO);
	to = stream_open_file(fup->temppath, O_WRONLY | O_CREAT | O_TRUNC, 0755);
	if (to == NULL) {
		xasprintf(&up->errmsg, "%s: Cannot create: %s",
		    fup->temppath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	stream_filter_start(to, STREAM_FILTER_MD5, md5);
	remains = fsize;
	do {
		nread = stream_read(up->rd, buf, (BUFSIZE > remains ?
		    remains : BUFSIZE));
		if (nread == -1) {
			stream_close(to);
			return (UPDATER_ERR_PROTO);
		}
		remains -= nread;
		if (stream_write(to, buf, nread) == -1) {
			stream_close(to);
			goto bad;
		}
	} while (remains > 0);
	stream_close(to);
	line = stream_getln(up->rd, NULL);
	if (line == NULL)
		return (UPDATER_ERR_PROTO);
	/* Check for EOF. */
	if (!(*line == '.' || (strncmp(line, ".<", 2) != 0)))
		return (UPDATER_ERR_PROTO);
	line = stream_getln(up->rd, NULL);
	if (line == NULL)
		return (UPDATER_ERR_PROTO);

	cmd = proto_get_char(&line);
	fup->wantmd5 = proto_get_ascii(&line);
	fup->free_wantmd5 = 0;
	if (fup->wantmd5 == NULL || line != NULL || cmd != '5')
		return (UPDATER_ERR_PROTO);

	sr->sr_clientattr = fattr_frompath(fup->temppath, FATTR_NOFOLLOW);
	if (sr->sr_clientattr == NULL)
		return (UPDATER_ERR_PROTO);
	fattr_override(sr->sr_clientattr, sr->sr_serverattr,
	    FA_MODTIME | FA_MASK);
	error = updater_updatefile(up, fup, md5);
	return (error);
bad:
	xasprintf(&up->errmsg, "%s: Cannot write: %s", fup->temppath,
	    strerror(errno));
	return (UPDATER_ERR_MSG);
}

static int
updater_checkout(struct updater *up, struct file_update *fup)
{
	char md5[MD5_DIGEST_SIZE];
	struct statusrec *sr;
	struct coll *coll;
	struct stream *to;
	ssize_t nbytes;
	size_t size;
	char *path, *line;
	int cmd, error, first;

	coll = fup->coll;
	sr = &fup->srbuf;
	path = fup->destpath;

	if (fup->isfixup)
		lprintf(1, " Fixup %s\n", fup->coname);
	else
		lprintf(1, " Checkout %s\n", fup->coname);
	error = mkdirhier(path, coll->co_umask);
	if (error) {
		xasprintf(&up->errmsg,
		    "Cannot create directories leading to \"%s\": %s",
		    path, strerror(errno));
		return (UPDATER_ERR_MSG);
	}

	to = stream_open_file(fup->temppath,
	    O_WRONLY | O_CREAT | O_TRUNC, 0600);
	if (to == NULL) {
		xasprintf(&up->errmsg, "%s: Cannot create: %s",
		    fup->temppath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	stream_filter_start(to, STREAM_FILTER_MD5, md5);
	line = stream_getln(up->rd, &size);
	first = 1;
	while (line != NULL) {
		if (line[size - 1] == '\n')
			size--;
	       	if ((size == 1 && *line == '.') ||
		    (size == 2 && memcmp(line, ".+", 2) == 0))
			break;
		if (size >= 2 && memcmp(line, "..", 2) == 0) {
			size--;
			line++;
		}
		if (!first) {
			nbytes = stream_write(to, "\n", 1);
			if (nbytes == -1)
				goto bad;
		}
		nbytes = stream_write(to, line, size);
		if (nbytes == -1)
			goto bad;
		line = stream_getln(up->rd, &size);
		first = 0;
	}
	if (line == NULL) {
		stream_close(to);
		return (UPDATER_ERR_READ);
	}
	if (size == 1 && *line == '.') {
		nbytes = stream_write(to, "\n", 1);
		if (nbytes == -1)
			goto bad;
	}
	stream_close(to);
	/* Get the checksum line. */
	line = stream_getln(up->rd, NULL);
	if (line == NULL)
		return (UPDATER_ERR_READ);
	cmd = proto_get_char(&line);
	fup->wantmd5 = proto_get_ascii(&line);
	fup->free_wantmd5 = 0;
	if (fup->wantmd5 == NULL || line != NULL || cmd != '5')
		return (UPDATER_ERR_PROTO);
	error = updater_updatefile(up, fup, md5);
	if (error)
		return (error);
	return (0);
bad:
	xasprintf(&up->errmsg, "%s: Cannot write: %s", fup->temppath,
	    strerror(errno));
	return (UPDATER_ERR_MSG);
}

/*
 * Remove all empty directories below file.
 * This function will trash the path passed to it.
 */
static void
updater_prunedirs(char *base, char *file)
{
	char *cp;
	int error;

	while ((cp = strrchr(file, '/')) != NULL) {
		*cp = '\0';
		if (strcmp(base, file) == 0)
			return;
		error = rmdir(file);
		if (error)
			return;
	}
}

/* Open the RCS file for edition if it hasn't been already. */
static void
updater_openrcs(struct rcsfile **rfp, struct file_update *fup, const char *path,
    const char *name)
{
	struct coll *coll;

	if (*rfp != NULL)
		return;

	coll = fup->coll;
	lprintf(1, " Edit %s", fup->coname);
	if (fup->attic)
		lprintf(1, " -> Attic");
	lprintf(1, "\n");
	*rfp = rcsfile_frompath(path, name, coll->co_cvsroot, coll->co_tag, 0);
}

/*
 * Edit an RCS file.
 *
 * XXX RCS options are not supported.
 */
static int
updater_rcsedit(struct updater *up, struct file_update *fup, char *name,
    char __unused *rcsopt)
{
	struct coll *coll;
	struct stream *dest;
	struct statusrec *sr;
	struct status *st;
	struct rcsfile *rf;
	struct fattr *oldfattr;
	char md5[MD5_DIGEST_SIZE];
	char *branch, *expandtxt, *line, *path, *revnum, *tag, *temppath;
	char *diffbase, *revdate, *author;
	int cmd, error, expand;

	coll = fup->coll;
	sr = &fup->srbuf;
	st = fup->st;
	temppath = fup->temppath;
	path = fup->origpath != NULL ? fup->origpath : fup->destpath;

	/* If the path is new, we must create the Attic dir if needed. */
	if (fup->origpath != NULL) {
		error = mkdirhier(fup->destpath, coll->co_umask);
		if (error) {
			xasprintf(&up->errmsg, "Unable to create Attic dir for "
			    "%s", fup->origpath);
			return (UPDATER_ERR_MSG);
		}
	}
	/*
	 * XXX: we could avoid parsing overhead if we're reading ahead before we
	 * parse the file.
	 */
	oldfattr = fattr_frompath(path, FATTR_NOFOLLOW);
	if (oldfattr == NULL) {
		xasprintf(&up->errmsg, "%s: Cannot get attributes: %s", path,
		    strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	fattr_merge(sr->sr_serverattr, oldfattr);

	cmd = -1;
	rf = NULL;
	while ((line = stream_getln(up->rd, NULL)) != NULL) {
		cmd = proto_get_char(&line);
		if (cmd == '.' && line == NULL)
			break;
		switch (cmd) {
		case 'B':
			branch = proto_get_ascii(&line);
			if (branch == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			updater_openrcs(&rf, fup, path, name);
			if (rf == NULL)
				break;
			lprintf(2, "  Set default branch to %s\n", branch);
			branch = xstrdup(branch);
			rcsfile_setval(rf, RCSFILE_BRANCH, branch, 0);
			break;
		case 'b':
			updater_openrcs(&rf, fup, path, name);
			if (rf == NULL)
				break;
			lprintf(2, "  Clear default branch\n");
			rcsfile_setval(rf, RCSFILE_BRANCH, NULL, 0);
			break;
		case 'D':
			revnum = proto_get_ascii(&line);
			diffbase = proto_get_ascii(&line);
			revdate = proto_get_ascii(&line);
			author = proto_get_ascii(&line);
			if (author == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			updater_openrcs(&rf, fup, path, name);
			if (rf == NULL)
				break;
			lprintf(2, "  Add delta %s %s %s\n", revnum, revdate,
			    author);
			error = updater_addelta(rf, up->rd, revnum, diffbase,
			    revdate, author);
			if (error)
				return (error);
			break;
		case 'd':
			revnum = proto_get_ascii(&line);
			if (revnum == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			updater_openrcs(&rf, fup, path, name);
			if (rf == NULL)
				break;
			lprintf(2, "  Delete delta %s\n", revnum);
			rcsfile_deleterev(rf, revnum);
			break;
		case 'E':
			expandtxt = proto_get_ascii(&line);
			if (expandtxt == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			expand = keyword_decode_expand(expandtxt);
			updater_openrcs(&rf, fup, path, name);
			if (rf == NULL)
				break;
			if (expand == EXPAND_DEFAULT)
				lprintf(2, "  Set keyword expansion to "
				    "default\n");
			else
				lprintf(2, "  Set keyword expansion to %s\n",
				    expandtxt);
			rcsfile_setexpand(rf, expand);
			break;
		case 'T':
			tag = proto_get_ascii(&line);
			revnum = proto_get_ascii(&line);
			if (revnum == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			updater_openrcs(&rf, fup, path, name);
			if (rf == NULL)
				break;
			lprintf(2, "  Add tag %s -> %s\n", tag, revnum);
			rcsfile_addtag(rf, tag, revnum);
			break;
		case 't':
			tag = proto_get_ascii(&line);
			revnum = proto_get_ascii(&line);
			if (revnum == NULL || line != NULL)
				return (UPDATER_ERR_PROTO);
			updater_openrcs(&rf, fup, path, name);
			if (rf == NULL)
				break;
			lprintf(2, "  Delete tag %s -> %s\n", tag, revnum);
			rcsfile_deletetag(rf, tag, revnum);
			break;
		default:
			return (UPDATER_ERR_PROTO);
		}

		if (rf == NULL) {
			fattr_free(oldfattr);
			/* We failed to open the RCS file for edition. */
			if (!(coll->co_options & CO_EXACTRCS)) {
				xasprintf(&up->errmsg, "%s: Invalid RCS file",
				    fup->destpath);
				return (UPDATER_ERR_MSG);
			}

			updater_needfixup(up, fup, "Invalid RCS file");
			/* We requested a fixup but we must still read the
			   remaining update commands for this RCS file to stay
			   in sync. */
			while ((line = stream_getln(up->rd, NULL)) != NULL) {
				if (strcmp(line, ".") == 0)
					break;
			}
			if (line == NULL)
				return (UPDATER_ERR_READ);
			return (0);
		}
	}
	if (line == NULL && cmd != '.') {
		fattr_free(oldfattr);
		return (UPDATER_ERR_READ);
	}

	if (rf == NULL) {
		fattr_maskout(oldfattr, ~FA_MODTIME);
		if (fattr_equal(oldfattr, sr->sr_serverattr))
		 	lprintf(1, " SetAttrs %s", fup->coname);
		else
			lprintf(1, " Touch %s", fup->coname);
		/* Install new attributes. */
		fattr_umask(sr->sr_serverattr, coll->co_umask);
		fattr_install(sr->sr_serverattr, fup->destpath, NULL);
		if (fup->attic)
			lprintf(1, " -> Attic");
		lprintf(1, "\n");
		fattr_free(oldfattr);
		goto finish;
	}

	fattr_free(oldfattr);
	/* Write and rename temp file. */
	dest = stream_open_file(fup->temppath,
	    O_RDWR | O_CREAT | O_TRUNC, 0600);
	if (dest == NULL) {
		xasprintf(&up->errmsg, "Error opening file %s for writing: %s",
		    fup->temppath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	stream_filter_start(dest, STREAM_FILTER_MD5RCS, md5);
	error = rcsfile_write(rf, dest);
	stream_close(dest);
	rcsfile_free(rf);
	if (error) {
		xasprintf(&up->errmsg, "%s: Cannot write: %s", fup->temppath,
		    strerror(errno));
		return (UPDATER_ERR_MSG);
	}

finish:
	sr->sr_clientattr = fattr_frompath(path, FATTR_NOFOLLOW);
	if (sr->sr_clientattr == NULL) {
		xasprintf(&up->errmsg, "%s: Cannot get attributes: %s",
		    fup->destpath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	fattr_override(sr->sr_clientattr, sr->sr_serverattr,
	    FA_MODTIME | FA_MASK);
	if (rf != NULL) {
		error = updater_updatefile(up, fup, md5);
		if (error)
			return (error);
	} else {
		/* Record its attributes since we touched it. */
		if (!(fattr_getmask(sr->sr_clientattr) & FA_LINKCOUNT) ||
		    fattr_getlinkcount(sr->sr_clientattr) <= 1)
		fattr_maskout(sr->sr_clientattr, FA_DEV | FA_INODE);
		error = status_put(st, sr);
		if (error) {
			up->errmsg = status_errmsg(st);
			return (UPDATER_ERR_MSG);
		}
	}

	/* In this case, we need to remove the old file afterwards. */
	/* XXX: Can we be sure that a file not edited is moved? I don't think
	 * this is a problem, since if a file is moved, it should be edited to
	 * show if it's dead or not.
	 */
	if (fup->origpath != NULL)
		updater_deletefile(fup->origpath);
	return (0);
}

/*
 * Add a delta to a RCS file.
 */
static int
updater_addelta(struct rcsfile *rf, struct stream *rd, char *revnum,
    char *diffbase, char *revdate, char *author)
{
	struct delta *d;
	char *line, *logline, *state, *textline;
	size_t size;
	int cmd;

	cmd = -1;
	size = 0;
	/* First add the delta so we have it. */
	d = rcsfile_addelta(rf, revnum, revdate, author, diffbase);
	if (d == NULL) {
		lprintf(-1, "Error adding delta %s\n", revnum);
		return (UPDATER_ERR_READ);
	}
	while ((line = stream_getln(rd, NULL)) != NULL) {
		cmd = proto_get_char(&line);
		if (cmd == -1)
			return (UPDATER_ERR_PROTO);
		if (cmd == '.' && line == NULL)
			break;
		switch (cmd) {
		case 'L':
			/* Do the same as in 'C' command. */
			logline = stream_getln(rd, &size);
			while (logline != NULL) {
				if (size == 2 && *logline == '.')
					break;
				if (size == 3 &&
				    memcmp(logline, ".+", 2) == 0) {
					rcsdelta_truncatelog(d, -1);
					break;
				}
				if (size >= 3 &&
				    memcmp(logline, "..", 2) == 0) {
					size--;
					logline++;
				}
				if (rcsdelta_appendlog(d, logline, size)
				    < 0)
					return (-1);
				logline = stream_getln(rd, &size);
			}
			break;
		case 'N':
		case 'n':
			/* XXX: Not supported. */
			break;
		case 'S':
			state = proto_get_ascii(&line);
			if (state == NULL)
				return (UPDATER_ERR_PROTO);
			rcsdelta_setstate(d, state);
			break;
		case 'T':
			/* Do the same as in 'C' command. */
			textline = stream_getln(rd, &size);
			while (textline != NULL) {
				if (size == 2 && *textline == '.')
					break;
				if (size == 3 &&
				    memcmp(textline, ".+", 2) == 0) {
					/* Truncate newline. */
					rcsdelta_truncatetext(d, -1);
					break;
				}
				if (size >= 3 &&
				    memcmp(textline, "..", 2) == 0) {
					size--;
					textline++;
				}
				if (rcsdelta_appendtext(d, textline, size) < 0)
					return (-1);
				textline = stream_getln(rd, &size);
			}
			break;
		}
	}
	if (line == NULL && cmd != '.')
		return (UPDATER_ERR_READ);
	return (0);
}

int
updater_append_file(struct updater *up, struct file_update *fup, off_t pos)
{
	struct fattr *fa;
	struct stream *to;
	struct statusrec *sr;
	ssize_t nread;
	off_t bytes;
	char buf[BUFSIZE], md5[MD5_DIGEST_SIZE];
	char *line;
	int cmd, error, fd;

	sr = &fup->srbuf;
	fa = sr->sr_serverattr;
	to = stream_open_file(fup->temppath, O_WRONLY | O_CREAT | O_TRUNC,
	    0755);
	if (to == NULL) {
		xasprintf(&up->errmsg, "%s: Cannot open: %s", fup->temppath,
		    strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	fd = open(fup->destpath, O_RDONLY);
	if (fd < 0) {
		xasprintf(&up->errmsg, "%s: Cannot open: %s", fup->destpath,
		    strerror(errno));
		return (UPDATER_ERR_MSG);
	}

	stream_filter_start(to, STREAM_FILTER_MD5, md5);
	/* First write the existing content. */
	while ((nread = read(fd, buf, BUFSIZE)) > 0) {
		if (stream_write(to, buf, nread) == -1)
			goto bad;
	}
	if (nread == -1) {
		xasprintf(&up->errmsg, "%s: Error reading: %s", fup->destpath,
		    strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	close(fd);

	bytes = fattr_filesize(fa) - pos;
	/* Append the new data. */
	do {
		nread = stream_read(up->rd, buf,
		    (BUFSIZE > bytes) ? bytes : BUFSIZE);
		if (nread == -1)
			return (UPDATER_ERR_PROTO);
		bytes -= nread;
		if (stream_write(to, buf, nread) == -1)
			goto bad;
	} while (bytes > 0);
	stream_close(to);

	line = stream_getln(up->rd, NULL);
	if (line == NULL)
		return (UPDATER_ERR_PROTO);
	/* Check for EOF. */
	if (!(*line == '.' || (strncmp(line, ".<", 2) != 0)))
		return (UPDATER_ERR_PROTO);
	line = stream_getln(up->rd, NULL);
	if (line == NULL)
		return (UPDATER_ERR_PROTO);

	cmd = proto_get_char(&line);
	fup->wantmd5 = proto_get_ascii(&line);
	fup->free_wantmd5 = 0;
	if (fup->wantmd5 == NULL || line != NULL || cmd != '5')
		return (UPDATER_ERR_PROTO);

	sr->sr_clientattr = fattr_frompath(fup->destpath, FATTR_NOFOLLOW);
	if (sr->sr_clientattr == NULL)
		return (UPDATER_ERR_PROTO);
	fattr_override(sr->sr_clientattr, sr->sr_serverattr,
	    FA_MODTIME | FA_MASK);
	error = updater_updatefile(up, fup, md5);
	return (error);
bad:
	xasprintf(&up->errmsg, "%s: Cannot write: %s", fup->temppath,
	    strerror(errno));
	return (UPDATER_ERR_MSG);
}

/*
 * Read file data from stream of checkout commands, and write it to the
 * destination.
 */
static int
updater_read_checkout(struct stream *src, struct stream *dest)
{
	ssize_t nbytes;
	size_t size;
	char *line;
	int first;

	first = 1;
	line = stream_getln(src, &size);
	while (line != NULL) {
		if (line[size - 1] == '\n')
			size--;
		if ((size == 1 && *line == '.') ||
		    (size == 2 && strncmp(line, ".+", 2) == 0))
			break;
		if (size >= 2 && strncmp(line, "..", 2) == 0) {
			size--;
			line++;
		}
		if (!first) {
			nbytes = stream_write(dest, "\n", 1);
			if (nbytes == -1)
				return (UPDATER_ERR_MSG);
		}
		nbytes = stream_write(dest, line, size);
		if (nbytes == -1)
			return (UPDATER_ERR_MSG);
		line = stream_getln(src, &size);
		first = 0;
	}
	if (line == NULL)
		return (UPDATER_ERR_READ);
	if (size == 1 && *line == '.') {
		nbytes = stream_write(dest, "\n", 1);
		if (nbytes == -1)
			return (UPDATER_ERR_MSG);
	}
	return (0);
}

/* Update file using the rsync protocol. */
static int
updater_rsync(struct updater *up, struct file_update *fup, size_t blocksize)
{
	struct statusrec *sr;
	struct stream *to;
	char md5[MD5_DIGEST_SIZE];
	ssize_t nbytes;
	size_t blocknum, blockstart, blockcount;
	char *buf, *line;
	int error, orig;

	sr = &fup->srbuf;

	lprintf(1, " Rsync %s\n", fup->coname);
	/* First open all files that we are going to work on. */
	to = stream_open_file(fup->temppath, O_WRONLY | O_CREAT | O_TRUNC,
	    0600);
	if (to == NULL) {
		xasprintf(&up->errmsg, "%s: Cannot create: %s",
		    fup->temppath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	orig = open(fup->destpath, O_RDONLY);
	if (orig < 0) {
		xasprintf(&up->errmsg, "%s: Cannot open: %s",
		    fup->destpath, strerror(errno));
		return (UPDATER_ERR_MSG);
	}
	stream_filter_start(to, STREAM_FILTER_MD5, md5);

	error = updater_read_checkout(up->rd, to);
	if (error) {
		xasprintf(&up->errmsg, "%s: Cannot write: %s", fup->temppath,
		    strerror(errno));
		return (error);
	}

	/* Buffer must contain blocksize bytes. */
	buf = xmalloc(blocksize);
	/* Done with the initial text, read and write chunks. */
	line = stream_getln(up->rd, NULL);
	while (line != NULL) {
		if (strcmp(line, ".") == 0)
			break;
		error = UPDATER_ERR_PROTO;
		if (proto_get_sizet(&line, &blockstart, 10) != 0)
			goto bad;
		if (proto_get_sizet(&line, &blockcount, 10) != 0)
			goto bad;
		/* Read blocks from original file. */
		lseek(orig, (blocksize * blockstart), SEEK_SET);
		error = UPDATER_ERR_MSG;
		for (blocknum = 0; blocknum < blockcount; blocknum++) {
			nbytes = read(orig, buf, blocksize);
			if (nbytes < 0) {
				xasprintf(&up->errmsg, "%s: Cannot read: %s",
				    fup->destpath, strerror(errno));
				goto bad;
			}
			nbytes = stream_write(to, buf, nbytes);
			if (nbytes == -1) {
				xasprintf(&up->errmsg, "%s: Cannot write: %s",
				    fup->temppath, strerror(errno));
				goto bad;
			}
		}
		/* Get the remaining text from the server. */
		error = updater_read_checkout(up->rd, to);
		if (error) {
			xasprintf(&up->errmsg, "%s: Cannot write: %s",
			    fup->temppath, strerror(errno));
			goto bad;
		}
		line = stream_getln(up->rd, NULL);
	}
	stream_close(to);
	close(orig);

	sr->sr_clientattr = fattr_frompath(fup->destpath, FATTR_NOFOLLOW);
	if (sr->sr_clientattr == NULL)
		return (UPDATER_ERR_PROTO);
	fattr_override(sr->sr_clientattr, sr->sr_serverattr,
	    FA_MODTIME | FA_MASK);

	error = updater_updatefile(up, fup, md5);
bad:
	free(buf);
	return (error);
}