Revert "453829 removed code with yahoo copyright"
This reverts commit 65c7b621f6
.
The yahoo copyright messages were removed by the contributor in the bugzilla/pullrequest
This commit is contained in:
parent
bc22ba72dc
commit
4d7ea8127e
|
@ -0,0 +1,128 @@
|
||||||
|
//
|
||||||
|
// ========================================================================
|
||||||
|
// Copyright (c) 1995-2014 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.rewrite.handler;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.regex.Matcher;
|
||||||
|
|
||||||
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
import javax.servlet.http.HttpServletResponse;
|
||||||
|
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
/** Rule to add a header based on a Regex match
|
||||||
|
*/
|
||||||
|
public class HeaderRegexRule extends RegexRule
|
||||||
|
{
|
||||||
|
|
||||||
|
private String _name;
|
||||||
|
private String _value;
|
||||||
|
private boolean _add=false;
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
public HeaderRegexRule()
|
||||||
|
{
|
||||||
|
_handling = false;
|
||||||
|
_terminating = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
/**
|
||||||
|
* Sets the header name.
|
||||||
|
*
|
||||||
|
* @param name name of the header field
|
||||||
|
*/
|
||||||
|
public void setName(String name)
|
||||||
|
{
|
||||||
|
_name = name;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
/**
|
||||||
|
* Sets the header value. The value can be either a <code>String</code> or <code>int</code> value.
|
||||||
|
*
|
||||||
|
* @param value of the header field
|
||||||
|
*/
|
||||||
|
public void setValue(String value)
|
||||||
|
{
|
||||||
|
_value = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
/**
|
||||||
|
* Sets the Add flag.
|
||||||
|
* @param add If true, the header is added to the response, otherwise the header it is set on the response.
|
||||||
|
*/
|
||||||
|
public void setAdd(boolean add)
|
||||||
|
{
|
||||||
|
_add = add;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
@Override
|
||||||
|
protected String apply(String target, HttpServletRequest request, HttpServletResponse response, Matcher matcher)
|
||||||
|
throws IOException
|
||||||
|
{
|
||||||
|
// process header
|
||||||
|
if (_add)
|
||||||
|
response.addHeader(_name, _value);
|
||||||
|
else
|
||||||
|
response.setHeader(_name, _value);
|
||||||
|
return target;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
/**
|
||||||
|
* Returns the header name.
|
||||||
|
* @return the header name.
|
||||||
|
*/
|
||||||
|
public String getName()
|
||||||
|
{
|
||||||
|
return _name;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
/**
|
||||||
|
* Returns the header value.
|
||||||
|
* @return the header value.
|
||||||
|
*/
|
||||||
|
public String getValue()
|
||||||
|
{
|
||||||
|
return _value;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
/**
|
||||||
|
* Returns the add flag value.
|
||||||
|
*/
|
||||||
|
public boolean isAdd()
|
||||||
|
{
|
||||||
|
return _add;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ------------------------------------------------------------ */
|
||||||
|
/**
|
||||||
|
* Returns the header contents.
|
||||||
|
*/
|
||||||
|
@Override
|
||||||
|
public String toString()
|
||||||
|
{
|
||||||
|
return super.toString()+"["+_name+","+_value+"]";
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,131 @@
|
||||||
|
//
|
||||||
|
// ========================================================================
|
||||||
|
// Copyright (c) 1995-2014 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.rewrite.handler;
|
||||||
|
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.util.Iterator;
|
||||||
|
|
||||||
|
public class HeaderRegexRuleTest extends AbstractRuleTestCase
|
||||||
|
{
|
||||||
|
|
||||||
|
private HeaderRegexRule _rule;
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void init() throws Exception
|
||||||
|
{
|
||||||
|
start(false);
|
||||||
|
_rule = new HeaderRegexRule();
|
||||||
|
_rule.setRegex("\\*");
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testHeaderWithTextValues() throws IOException
|
||||||
|
{
|
||||||
|
// different keys
|
||||||
|
String headers[][] =
|
||||||
|
{
|
||||||
|
{ "hnum#1", "test1" },
|
||||||
|
{ "hnum#2", "2test2" },
|
||||||
|
{ "hnum#3", "test3" } };
|
||||||
|
assertHeaders(headers);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testHeaderWithNumberValues() throws IOException
|
||||||
|
{
|
||||||
|
String headers[][] =
|
||||||
|
{
|
||||||
|
{ "hello", "1" },
|
||||||
|
{ "hello", "-1" },
|
||||||
|
{ "hello", "100" },
|
||||||
|
{ "hello", "100" },
|
||||||
|
{ "hello", "100" },
|
||||||
|
{ "hello", "100" },
|
||||||
|
{ "hello", "100" },
|
||||||
|
{ "hello1", "200" } };
|
||||||
|
assertHeaders(headers);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testHeaderOverwriteValues() throws IOException
|
||||||
|
{
|
||||||
|
String headers[][] =
|
||||||
|
{
|
||||||
|
{ "size", "100" },
|
||||||
|
{ "size", "200" },
|
||||||
|
{ "size", "300" },
|
||||||
|
{ "size", "400" },
|
||||||
|
{ "size", "500" },
|
||||||
|
{ "title", "abc" },
|
||||||
|
{ "title", "bac" },
|
||||||
|
{ "title", "cba" },
|
||||||
|
{ "title1", "abba" },
|
||||||
|
{ "title1", "abba1" },
|
||||||
|
{ "title1", "abba" },
|
||||||
|
{ "title1", "abba1" } };
|
||||||
|
assertHeaders(headers);
|
||||||
|
Iterator<String> e = _response.getHeaders("size").iterator();
|
||||||
|
int count = 0;
|
||||||
|
while (e.hasNext())
|
||||||
|
{
|
||||||
|
e.next();
|
||||||
|
count++;
|
||||||
|
}
|
||||||
|
assertEquals(1,count);
|
||||||
|
assertEquals("500",_response.getHeader("size"));
|
||||||
|
assertEquals("cba",_response.getHeader("title"));
|
||||||
|
assertEquals("abba1",_response.getHeader("title1"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testMatch() throws Exception
|
||||||
|
{
|
||||||
|
_rule.setRegex("/my/dir/file/(.*)$");
|
||||||
|
_rule.setName("cache-control");
|
||||||
|
_rule.setValue("no-store");
|
||||||
|
_rule.matchAndApply("/my/dir/file/",_request,_response);
|
||||||
|
assertEquals("no-store",_response.getHeader("cache-control"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testNotMatch() throws Exception
|
||||||
|
{
|
||||||
|
reset();
|
||||||
|
_rule.setRegex("/my/dir/file/(.*)$");
|
||||||
|
_rule.setName("cache-control");
|
||||||
|
_rule.setValue("no-store");
|
||||||
|
_rule.matchAndApply("/my/dir/file_not_match/",_request,_response);
|
||||||
|
assertEquals(null,_response.getHeader("cache-control"));
|
||||||
|
}
|
||||||
|
|
||||||
|
private void assertHeaders(String headers[][]) throws IOException
|
||||||
|
{
|
||||||
|
for (String[] header : headers)
|
||||||
|
{
|
||||||
|
_rule.setName(header[0]);
|
||||||
|
_rule.setValue(header[1]);
|
||||||
|
_rule.apply(null,_request,_response,null);
|
||||||
|
assertEquals(header[1],_response.getHeader(header[0]));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue