parent
2d2df64870
commit
ed3bcfaf71
@ -0,0 +1,87 @@
|
|||||||
|
package net.yacy.http;
|
||||||
|
|
||||||
|
import java.io.ByteArrayOutputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
|
import javax.servlet.ServletException;
|
||||||
|
import javax.servlet.ServletOutputStream;
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
import javax.servlet.http.HttpServletResponseWrapper;
|
||||||
|
|
||||||
|
import org.eclipse.jetty.server.Handler;
|
||||||
|
import org.eclipse.jetty.server.HandlerContainer;
|
||||||
|
import org.eclipse.jetty.server.Request;
|
||||||
|
|
||||||
|
import org.eclipse.jetty.server.handler.HandlerWrapper;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* jetty http handler:
|
||||||
|
* Handles Server-side Includes, used for trickling display of search results
|
||||||
|
*/
|
||||||
|
public abstract class ContentModHandler extends HandlerWrapper implements Handler, HandlerContainer {
|
||||||
|
|
||||||
|
public ContentModHandler() {
|
||||||
|
super();
|
||||||
|
}
|
||||||
|
|
||||||
|
public ContentModHandler(Handler h) {
|
||||||
|
super();
|
||||||
|
this.setHandler(h);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void handle(String target, Request baseRequest, HttpServletRequest request,
|
||||||
|
HttpServletResponse response) throws IOException, ServletException {
|
||||||
|
// wrap response
|
||||||
|
ContentModResponseWrapper wrapped_response = new ContentModResponseWrapper(response);
|
||||||
|
super.handle(target, baseRequest, request, wrapped_response);
|
||||||
|
wrapped_response.commit(this, request, response);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected abstract void doContentMod(byte[] in, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException ;
|
||||||
|
|
||||||
|
private class ContentModResponseWrapper extends HttpServletResponseWrapper {
|
||||||
|
|
||||||
|
private HttpServletResponse wrappedResponse;
|
||||||
|
private ByteArrayServletOutputStream wrappedOutputStream = new ByteArrayServletOutputStream();
|
||||||
|
|
||||||
|
public ContentModResponseWrapper(HttpServletResponse response) {
|
||||||
|
super(response);
|
||||||
|
wrappedResponse = response;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public ServletOutputStream getOutputStream() throws IOException {
|
||||||
|
return wrappedOutputStream;
|
||||||
|
}
|
||||||
|
|
||||||
|
public void commit(ContentModHandler cmh, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
|
||||||
|
cmh.doContentMod(wrappedOutputStream.getBuffer(), request, response);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private class ByteArrayServletOutputStream extends ServletOutputStream {
|
||||||
|
|
||||||
|
private ByteArrayOutputStream buffer = new ByteArrayOutputStream();
|
||||||
|
|
||||||
|
public ByteArrayServletOutputStream() {
|
||||||
|
super();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void write(int b) throws IOException {
|
||||||
|
buffer.write(b);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void write(byte[] b) throws IOException {
|
||||||
|
buffer.write(b);
|
||||||
|
}
|
||||||
|
|
||||||
|
public byte[] getBuffer() {
|
||||||
|
return buffer.toByteArray();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
@ -0,0 +1,81 @@
|
|||||||
|
/**
|
||||||
|
* SSIHandler
|
||||||
|
* Copyright 2011 by Michael Peter Christen; mc@yacy.net, Frankfurt a. M., Germany
|
||||||
|
* First released 13.04.2011 at http://yacy.net
|
||||||
|
*
|
||||||
|
* $LastChangedDate$
|
||||||
|
* $LastChangedRevision$
|
||||||
|
* $LastChangedBy$
|
||||||
|
*
|
||||||
|
* This library is free software; you can redistribute it and/or
|
||||||
|
* modify it under the terms of the GNU Lesser General Public
|
||||||
|
* License as published by the Free Software Foundation; either
|
||||||
|
* version 2.1 of the License, or (at your option) any later version.
|
||||||
|
*
|
||||||
|
* This library is distributed in the hope that it will be useful,
|
||||||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||||
|
* Lesser General Public License for more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU Lesser General Public License
|
||||||
|
* along with this program in the file lgpl21.txt
|
||||||
|
* If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package net.yacy.http;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.OutputStream;
|
||||||
|
|
||||||
|
import javax.servlet.RequestDispatcher;
|
||||||
|
import javax.servlet.ServletException;
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
|
||||||
|
import org.eclipse.jetty.server.Handler;
|
||||||
|
import org.eclipse.jetty.server.HandlerContainer;
|
||||||
|
|
||||||
|
import net.yacy.kelondro.util.ByteBuffer;
|
||||||
|
|
||||||
|
public class SSIHandler extends ContentModHandler implements Handler, HandlerContainer {
|
||||||
|
|
||||||
|
public SSIHandler(Handler h) {
|
||||||
|
super(h);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected void doContentMod(final byte[] in, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
|
||||||
|
ByteBuffer buffer = new ByteBuffer(in);
|
||||||
|
OutputStream out = response.getOutputStream();
|
||||||
|
int off = 0; // starting offset
|
||||||
|
int p = buffer.indexOf("<!--#".getBytes(), off);
|
||||||
|
int q;
|
||||||
|
while (p >= 0) {
|
||||||
|
q = buffer.indexOf("-->".getBytes(), p + 10);
|
||||||
|
out.write(in, off, p - off);
|
||||||
|
out.flush();
|
||||||
|
parseSSI(buffer, p, request, response);
|
||||||
|
off = q + 3;
|
||||||
|
p = buffer.indexOf("<!--#".getBytes(), off);
|
||||||
|
}
|
||||||
|
out.write(in, off, in.length - off);
|
||||||
|
out.flush();
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void parseSSI(final ByteBuffer in, final int off, HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
|
||||||
|
if (in.startsWith("<!--#include virtual=\"".getBytes(), off)) {
|
||||||
|
final int q = in.indexOf("\"".getBytes(), off + 22);
|
||||||
|
if (q > 0) {
|
||||||
|
final String path = in.toString(off + 22, q - off - 22);
|
||||||
|
try {
|
||||||
|
RequestDispatcher dispatcher = request.getRequestDispatcher("/"+path);
|
||||||
|
dispatcher.include(request, response);
|
||||||
|
} catch (Exception e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
throw new ServletException();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in new issue