89 lines
3.0 KiB
Java
89 lines
3.0 KiB
Java
/*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* The OpenSearch Contributors require contributions made to
|
|
* this file be licensed under the Apache-2.0 license or a
|
|
* compatible open source license.
|
|
*/
|
|
|
|
/*
|
|
* Licensed to Elasticsearch under one or more contributor
|
|
* license agreements. See the NOTICE file distributed with
|
|
* this work for additional information regarding copyright
|
|
* ownership. Elasticsearch licenses this file to you under
|
|
* the Apache License, Version 2.0 (the "License"); you may
|
|
* not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing,
|
|
* software distributed under the License is distributed on an
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
* KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations
|
|
* under the License.
|
|
*/
|
|
/*
|
|
* Modifications Copyright OpenSearch Contributors. See
|
|
* GitHub history for details.
|
|
*/
|
|
|
|
package fixture.s3;
|
|
|
|
import com.sun.net.httpserver.HttpExchange;
|
|
import com.sun.net.httpserver.HttpHandler;
|
|
import com.sun.net.httpserver.HttpServer;
|
|
import org.opensearch.rest.RestStatus;
|
|
|
|
import java.io.IOException;
|
|
import java.net.InetAddress;
|
|
import java.net.InetSocketAddress;
|
|
import java.util.Objects;
|
|
|
|
public class S3HttpFixture {
|
|
|
|
private final HttpServer server;
|
|
|
|
S3HttpFixture(final String[] args) throws Exception {
|
|
this.server = HttpServer.create(new InetSocketAddress(InetAddress.getByName(args[0]), Integer.parseInt(args[1])), 0);
|
|
this.server.createContext("/", Objects.requireNonNull(createHandler(args)));
|
|
}
|
|
|
|
final void start() throws Exception {
|
|
try {
|
|
server.start();
|
|
// wait to be killed
|
|
Thread.sleep(Long.MAX_VALUE);
|
|
} finally {
|
|
server.stop(0);
|
|
}
|
|
}
|
|
|
|
protected HttpHandler createHandler(final String[] args) {
|
|
final String bucket = Objects.requireNonNull(args[2]);
|
|
final String basePath = args[3];
|
|
final String accessKey = Objects.requireNonNull(args[4]);
|
|
|
|
return new S3HttpHandler(bucket, basePath) {
|
|
@Override
|
|
public void handle(final HttpExchange exchange) throws IOException {
|
|
final String authorization = exchange.getRequestHeaders().getFirst("Authorization");
|
|
if (authorization == null || authorization.contains(accessKey) == false) {
|
|
sendError(exchange, RestStatus.FORBIDDEN, "AccessDenied", "Bad access key");
|
|
return;
|
|
}
|
|
super.handle(exchange);
|
|
}
|
|
};
|
|
}
|
|
|
|
public static void main(final String[] args) throws Exception {
|
|
if (args == null || args.length < 5) {
|
|
throw new IllegalArgumentException("S3HttpFixture expects 5 arguments [address, port, bucket, base path, access key]");
|
|
}
|
|
final S3HttpFixture fixture = new S3HttpFixture(args);
|
|
fixture.start();
|
|
}
|
|
}
|