/// <summary>Write an XDR message to a TCP ChannelBuffer</summary>
public static ChannelBuffer WriteMessageTcp(XDR request, bool last)
{
Preconditions.CheckState(request.state == XDR.State.Writing);
ByteBuffer b = request.buf.Duplicate();
b.Flip();
byte[] fragmentHeader = XDR.RecordMark(b.Limit(), last);
ByteBuffer headerBuf = ByteBuffer.Wrap(fragmentHeader);
// TODO: Investigate whether making a copy of the buffer is necessary.
return(ChannelBuffers.CopiedBuffer(headerBuf, b));
}
protected internal virtual void SendError(ChannelHandlerContext ctx, string message
, HttpResponseStatus status)
{
HttpResponse response = new DefaultHttpResponse(HttpVersion.Http11, status);
response.SetHeader(HttpHeaders.Names.ContentType, "text/plain; charset=UTF-8");
// Put shuffle version into http header
response.SetHeader(ShuffleHeader.HttpHeaderName, ShuffleHeader.DefaultHttpHeaderName
);
response.SetHeader(ShuffleHeader.HttpHeaderVersion, ShuffleHeader.DefaultHttpHeaderVersion
);
response.SetContent(ChannelBuffers.CopiedBuffer(message, CharsetUtil.Utf8));
// Close the connection as soon as the error message is sent.
ctx.GetChannel().Write(response).AddListener(ChannelFutureListener.Close);
}
/// <exception cref="System.Exception"/>
public override void MessageReceived(ChannelHandlerContext ctx, MessageEvent e)
{
HttpRequest request = (HttpRequest)e.GetMessage();
if (request.GetMethod() == HttpMethod.Options)
{
// Mimic SPNEGO authentication
HttpResponse response = new DefaultHttpResponse(HttpVersion.Http11, HttpResponseStatus
.Ok);
response.AddHeader("Set-Cookie", "hadoop-auth=1234");
e.GetChannel().Write(response).AddListener(ChannelFutureListener.Close);
}
else
{
if (request.GetMethod() != HttpMethod.Get)
{
e.GetChannel().Close();
}
}
UnmodifiableIterator <KeyValuePair <string, TestDelegationTokenRemoteFetcher.Handler
> > iter = this.routes.GetEnumerator();
while (iter.HasNext())
{
KeyValuePair <string, TestDelegationTokenRemoteFetcher.Handler> entry = iter.Next(
);
if (request.GetUri().Contains(entry.Key))
{
TestDelegationTokenRemoteFetcher.Handler handler = entry.Value;
try
{
handler.Handle(e.GetChannel(), this.token, this.serviceUrl);
}
catch (Exception ee)
{
this._enclosing.assertionError = ee;
HttpResponse response = new DefaultHttpResponse(HttpVersion.Http11, HttpResponseStatus
.BadRequest);
response.SetContent(ChannelBuffers.CopiedBuffer(ee.Message, Encoding.Default));
e.GetChannel().Write(response).AddListener(ChannelFutureListener.Close);
}
return;
}
}
}
/// <summary>Write an XDR message to a UDP ChannelBuffer</summary>
public static ChannelBuffer WriteMessageUdp(XDR response)
{
Preconditions.CheckState(response.state == XDR.State.Reading);
// TODO: Investigate whether making a copy of the buffer is necessary.
return(ChannelBuffers.CopiedBuffer(response.buf));
}
