Skip to content

Instantly share code, notes, and snippets.

@gregw
Created December 1, 2016 09:06
Show Gist options
  • Save gregw/1c22cf9c1c0f3b7f9e14226c0dd73195 to your computer and use it in GitHub Desktop.
Save gregw/1c22cf9c1c0f3b7f9e14226c0dd73195 to your computer and use it in GitHub Desktop.
//
// ========================================================================
// Copyright (c) 1995-2016 Mort Bay Consulting Pty. Ltd.
// ------------------------------------------------------------------------
// All rights reserved. This program and the accompanying materials
// are made available under the terms of the Eclipse Public License v1.0
// and Apache License v2.0 which accompanies this distribution.
//
// The Eclipse Public License is available at
// http://www.eclipse.org/legal/epl-v10.html
//
// The Apache License v2.0 is available at
// http://www.opensource.org/licenses/apache2.0.php
//
// You may elect to redistribute this code under either of these licenses.
// ========================================================================
//
package org.eclipse.jetty.http2.client;
import java.net.InetSocketAddress;
import java.util.Random;
import java.util.concurrent.Phaser;
import java.util.concurrent.TimeUnit;
import org.eclipse.jetty.http.HttpFields;
import org.eclipse.jetty.http.HttpURI;
import org.eclipse.jetty.http.HttpVersion;
import org.eclipse.jetty.http.MetaData;
import org.eclipse.jetty.http2.api.Session;
import org.eclipse.jetty.http2.api.Stream;
import org.eclipse.jetty.http2.api.server.ServerSessionListener;
import org.eclipse.jetty.http2.frames.DataFrame;
import org.eclipse.jetty.http2.frames.HeadersFrame;
import org.eclipse.jetty.http2.frames.PushPromiseFrame;
import org.eclipse.jetty.http2.frames.ResetFrame;
import org.eclipse.jetty.util.BufferUtil;
import org.eclipse.jetty.util.Callback;
import org.eclipse.jetty.util.FuturePromise;
import org.eclipse.jetty.util.Jetty;
import org.eclipse.jetty.util.Promise;
import org.eclipse.jetty.util.TypeUtil;
import org.eclipse.jetty.util.ssl.SslContextFactory;
public class Client
{
public static void main(String[] args) throws Exception
{
HTTP2Client client = new HTTP2Client();
SslContextFactory sslContextFactory = new SslContextFactory(true);
client.addBean(sslContextFactory);
client.start();
Random random = new Random();
String host = "api.facebook.com";
int port = 443;
FuturePromise<Session> sessionPromise = new FuturePromise<>();
client.connect(sslContextFactory, new InetSocketAddress(host, port), new ServerSessionListener.Adapter(), sessionPromise);
Session session = sessionPromise.get(5, TimeUnit.SECONDS);
try
{
while (true)
{
System.err.println("==========================================");
byte[] data = new byte[1+random.nextInt(512)];
random.nextBytes(data);
String r = TypeUtil.toHexString(data);
HttpFields requestFields = new HttpFields();
requestFields.put("notice", "This is not a DOS. Debugging a HTTP/2 difference with FB");
requestFields.put("User-Agent", client.getClass().getName() + "/" + Jetty.VERSION);
requestFields.put("random", r);
MetaData.Request metaData = new MetaData.Request("GET",
new HttpURI("https://" + host + ":" + port + "/restserver.php?random="+r),
HttpVersion.HTTP_2, requestFields);
HeadersFrame headersFrame = new HeadersFrame(metaData, null, true);
final Phaser phaser = new Phaser(2);
session.newStream(headersFrame, new Promise.Adapter<>(), new Stream.Listener.Adapter()
{
@Override
public void onHeaders(Stream stream, HeadersFrame frame)
{
System.err.println(frame);
System.err.println(((MetaData.Response)frame.getMetaData()).getStatus());
System.err.println(frame.getMetaData().getFields());
if (frame.isEndStream())
phaser.arrive();
}
@Override
public void onData(Stream stream, DataFrame frame, Callback callback)
{
System.err.println(frame);
System.err.println(BufferUtil.toDetailString(frame.getData()));
callback.succeeded();
if (frame.isEndStream())
phaser.arrive();
}
@Override
public Stream.Listener onPush(Stream stream, PushPromiseFrame frame)
{
System.err.println(frame);
phaser.register();
return this;
}
@Override
public void onReset(Stream stream, ResetFrame frame)
{
System.err.println("onReset "+stream);
System.err.println(frame);
System.err.println(frame.getError());
super.onReset(stream,frame);
}
@Override
public void onTimeout(Stream stream, Throwable x)
{
System.err.println("onTimeout "+stream);
super.onTimeout(stream,x);
}
@Override
public boolean onIdleTimeout(Stream stream, Throwable x)
{
System.err.println("onIdleTimeout "+stream);
return super.onIdleTimeout(stream,x);
}
});
phaser.awaitAdvanceInterruptibly(phaser.arrive(), 5, TimeUnit.SECONDS);
}
}
finally
{
client.stop();
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment