This is an automated email from the ASF dual-hosted git repository. coheigea pushed a commit to branch master in repository https://gitbox.apache.org/repos/asf/ws-wss4j.git
The following commit(s) were added to refs/heads/master by this push: new b8e304b Missing file from last commit b8e304b is described below commit b8e304b2a9dfa2abf710f46e75dec4b8aa0889b5 Author: Colm O hEigeartaigh <cohei...@apache.org> AuthorDate: Wed May 13 07:55:52 2020 +0100 Missing file from last commit --- .../wss4j/stax/utils/LimitingInputStream.java | 68 ++++++++++++++++++++++ 1 file changed, 68 insertions(+) diff --git a/ws-security-stax/src/main/java/org/apache/wss4j/stax/utils/LimitingInputStream.java b/ws-security-stax/src/main/java/org/apache/wss4j/stax/utils/LimitingInputStream.java new file mode 100644 index 0000000..efa20dc --- /dev/null +++ b/ws-security-stax/src/main/java/org/apache/wss4j/stax/utils/LimitingInputStream.java @@ -0,0 +1,68 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF 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. + */ +package org.apache.wss4j.stax.utils; + +import org.apache.xml.security.utils.I18n; + +import java.io.FilterInputStream; +import java.io.IOException; +import java.io.InputStream; + +/** + */ +public final class LimitingInputStream extends FilterInputStream { + + private long limit; + private long count; + + public LimitingInputStream(InputStream in, long limit) { + super(in); + this.limit = limit; + } + + @Override + public int read() throws IOException { + int r = super.read(); + if (r >= 0) { + incrementCountAndTestLimit(r); + } + return r; + } + + @Override + public int read(byte[] b) throws IOException { + return read(b, 0, b.length); + } + + @Override + public int read(byte[] b, int off, int len) throws IOException { + int r = super.read(b, off, len); + if (r >= 0) { + incrementCountAndTestLimit(r); + } + return r; + } + + private void incrementCountAndTestLimit(long read) throws IOException { + this.count += read; + if (this.count > this.limit) { + throw new IOException(I18n.getExceptionMessage("secureProcessing.inputStreamLimitReached", new Object[]{this.limit})); + } + } +}