14fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy/**************************************************************** 24fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * Licensed to the Apache Software Foundation (ASF) under one * 34fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * or more contributor license agreements. See the NOTICE file * 44fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * distributed with this work for additional information * 54fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * regarding copyright ownership. The ASF licenses this file * 64fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * to you under the Apache License, Version 2.0 (the * 74fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * "License"); you may not use this file except in compliance * 84fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * with the License. You may obtain a copy of the License at * 94fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * * 104fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * http://www.apache.org/licenses/LICENSE-2.0 * 114fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * * 124fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * Unless required by applicable law or agreed to in writing, * 134fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * software distributed under the License is distributed on an * 144fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * 154fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * KIND, either express or implied. See the License for the * 164fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * specific language governing permissions and limitations * 174fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * under the License. * 184fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy ****************************************************************/ 194fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 204fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedypackage org.apache.james.mime4j; 214fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 224fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedyimport java.io.IOException; 234fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedyimport java.io.InputStream; 244fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 254fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy/** 264fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * <code>InputStream</code> used by the parser to wrap the original user 274fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * supplied stream. This stream keeps track of the current line number and 284fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * can also be truncated. When truncated the stream will appear to have 294fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * reached end of file. This is used by the parser's 304fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * {@link org.apache.james.mime4j.MimeStreamParser#stop()} method. 314fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * 324fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * 334fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * @version $Id: RootInputStream.java,v 1.2 2004/10/02 12:41:10 ntherning Exp $ 344fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy */ 354fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedyclass RootInputStream extends InputStream { 364fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy private InputStream is = null; 374fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy private int lineNumber = 1; 384fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy private int prev = -1; 394fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy private boolean truncated = false; 404fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 414fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy /** 424fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * Creates a new <code>RootInputStream</code>. 434fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * 444fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * @param in the stream to read from. 454fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy */ 464fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy public RootInputStream(InputStream is) { 474fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy this.is = is; 484fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 494fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 504fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy /** 514fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * Gets the current line number starting at 1 524fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * (the number of <code>\r\n</code> read so far plus 1). 534fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * 544fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * @return the current line number. 554fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy */ 564fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy public int getLineNumber() { 574fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy return lineNumber; 584fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 594fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 604fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy /** 614fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * Truncates this <code>InputStream</code>. After this call any 624fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * call to {@link #read()}, {@link #read(byte[]) or 634fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * {@link #read(byte[], int, int)} will return 644fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * -1 as if end-of-file had been reached. 654fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy */ 664fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy public void truncate() { 674fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy this.truncated = true; 684fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 694fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 704fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy /** 714fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * @see java.io.InputStream#read() 724fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy */ 734fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy public int read() throws IOException { 744fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy if (truncated) { 754fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy return -1; 764fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 774fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 784fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy int b = is.read(); 794fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy if (prev == '\r' && b == '\n') { 804fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy lineNumber++; 814fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 824fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy prev = b; 834fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy return b; 844fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 854fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 864fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy /** 874fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * 884fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * @see java.io.InputStream#read(byte[], int, int) 894fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy */ 904fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy public int read(byte[] b, int off, int len) throws IOException { 914fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy if (truncated) { 924fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy return -1; 934fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 944fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 954fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy int n = is.read(b, off, len); 964fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy for (int i = off; i < off + n; i++) { 974fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy if (prev == '\r' && b[i] == '\n') { 984fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy lineNumber++; 994fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 1004fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy prev = b[i]; 1014fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 1024fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy return n; 1034fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 1044fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy 1054fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy /** 1064fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy * @see java.io.InputStream#read(byte[]) 1074fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy */ 1084fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy public int read(byte[] b) throws IOException { 1094fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy return read(b, 0, b.length); 1104fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy } 1114fa0a3295bcacbdcd6a9e7709cf17aa5adb90356Scott Kennedy} 112