Florian Richter
14 years ago
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