xref: /haiku/src/add-ons/kernel/file_systems/nfs4/Request.cpp (revision be4c74ead31246bd9b332f83a9e298969385fc4e)

/*
 * Copyright 2012 Haiku, Inc. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Paweł Dziepak, pdziepak@quarnos.org
 */


#include "Inode.h"
#include "Request.h"


status_t
Request::Send(Cookie* cookie)
{
	switch (fServer->ID().fProtocol) {
		case ProtocolUDP:	return _SendUDP(cookie);
		case ProtocolTCP:	return _SendTCP(cookie);
	}

	return B_BAD_VALUE;
}


status_t
Request::_SendUDP(Cookie* cookie)
{
	RPC::Reply *rpl = NULL;
	RPC::Request *rpc;

	status_t result = fServer->SendCallAsync(fBuilder.Request(), &rpl, &rpc);
	if (result != B_OK)
		return result;

	if (cookie != NULL)
		cookie->RegisterRequest(rpc);

	result = fServer->WaitCall(rpc);
	if (result != B_OK) {
		int attempts = 1;
		while (result != B_OK && attempts++ < kRetryLimit) {
			result = fServer->ResendCallAsync(fBuilder.Request(), rpc);
			if (result != B_OK) {
				if (cookie != NULL)
					cookie->UnregisterRequest(rpc);
				return result;
			}

			result = fServer->WaitCall(rpc);
		}

		if (result != B_OK) {
			if (cookie != NULL)
				cookie->UnregisterRequest(rpc);
			fServer->CancelCall(rpc);
			delete rpc;
			return result;
		}
	}

	if (cookie != NULL)
		cookie->UnregisterRequest(rpc);

	if (rpc->fError != B_OK) {
		delete rpl;
		result = rpc->fError;
		delete rpc;
		return result;
	} else {
		fReply.SetTo(rpl);
		delete rpc;
		return B_OK;
	}
}


status_t
Request::_SendTCP(Cookie* cookie)
{
	RPC::Reply *rpl = NULL;
	RPC::Request *rpc;

	status_t result;
	int attempts = 0;
	do {
		result = fServer->SendCallAsync(fBuilder.Request(), &rpl, &rpc);
		if (result == B_NO_MEMORY)
			return result;
		else if (result != B_OK) {
			fServer->Repair();
			continue;
		}

		if (cookie != NULL)
			cookie->RegisterRequest(rpc);

		result = fServer->WaitCall(rpc);
		if (result != B_OK) {
			if (cookie != NULL)
				cookie->UnregisterRequest(rpc);

			fServer->CancelCall(rpc);
			delete rpc;

			fServer->Repair();
		}
	} while (result != B_OK && attempts++ < kRetryLimit);

	if (cookie != NULL)
		cookie->UnregisterRequest(rpc);

	if (rpc->fError != B_OK) {
		delete rpl;
		result = rpc->fError;
		delete rpc;
		return result;
	} else {
		fReply.SetTo(rpl);
		delete rpc;
		return B_OK;
	};
}


void
Request::Reset()
{
	fBuilder.Reset();
	fReply.Reset();
}