/* OutputBuffer.java -- Copyright (C) 2003, 2006 Free Software Foundation, Inc. This file is a part of GNU Classpath. GNU Classpath is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. GNU Classpath 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 General Public License for more details. You should have received a copy of the GNU General Public License along with GNU Classpath; if not, write to the Free Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA Linking this library statically or dynamically with other modules is making a combined work based on this library. Thus, the terms and conditions of the GNU General Public License cover the whole combination. As a special exception, the copyright holders of this library give you permission to link this library with independent modules to produce an executable, regardless of the license terms of these independent modules, and to copy and distribute the resulting executable under terms of your choice, provided that you also meet, for each linked independent module, the terms and conditions of the license of that module. An independent module is a module which is not derived from or based on this library. If you modify this library, you may extend this exception to your version of the library, but you are not obligated to do so. If you do not wish to do so, delete this exception statement from your version. */ package gnu.javax.crypto.sasl; import gnu.java.security.Registry; import gnu.java.security.util.Util; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.math.BigInteger; /** * The implementation of an outgoing SASL buffer. *

* The data elements this class caters for are described in [1]. *

* References: *

    *
  1. * Secure Remote Password Authentication Mechanism;
    * draft-burdis-cat-srp-sasl-09,
    Keith Burdis and Raïf S. Naffah.
  2. *
*/ public class OutputBuffer { /** The internal output stream. */ private ByteArrayOutputStream out; public OutputBuffer() { super(); out = new ByteArrayOutputStream(); } /** * Encodes a SASL scalar quantity, count-octet long, to the * current buffer. * * @param count number of octets to encode b with. * @param b the scalar quantity. * @throws SaslEncodingException if an encoding size constraint is violated. * @throws IOException if any other I/O exception occurs during the operation. */ public void setScalar(int count, int b) throws IOException { if (count < 0 || count > 4) throw new SaslEncodingException("Invalid SASL scalar octet count: " + String.valueOf(count)); byte[] element = new byte[count]; for (int i = count; --i >= 0; b >>>= 8) element[i] = (byte) b; out.write(element); } /** * Encodes a SASL OS to the current buffer. * * @param b the OS element. * @throws SaslEncodingException if an encoding size constraint is violated. * @throws IOException if any other I/O exception occurs during the operation. */ public void setOS(byte[] b) throws IOException { final int length = b.length; if (length > Registry.SASL_ONE_BYTE_MAX_LIMIT) throw new SaslEncodingException("SASL octet-sequence too long"); out.write(length & 0xFF); out.write(b); } /** * Encodes a SASL EOS to the current buffer. * * @param b the EOS element. * @throws SaslEncodingException if an encoding size constraint is violated. * @throws IOException if any other I/O exception occurs during the operation. */ public void setEOS(byte[] b) throws IOException { final int length = b.length; if (length > Registry.SASL_TWO_BYTE_MAX_LIMIT) throw new SaslEncodingException("SASL extended octet-sequence too long"); byte[] lengthBytes = { (byte)(length >>> 8), (byte) length }; out.write(lengthBytes); out.write(b); } /** * Encodes a SASL MPI to the current buffer. * * @param val the MPI element. * @throws SaslEncodingException if an encoding size constraint is violated. * @throws IOException if any other I/O exception occurs during the operation. */ public void setMPI(BigInteger val) throws IOException { byte[] b = Util.trim(val); final int length = b.length; if (length > Registry.SASL_TWO_BYTE_MAX_LIMIT) throw new SaslEncodingException("SASL multi-precision integer too long"); byte[] lengthBytes = { (byte)(length >>> 8), (byte) length }; out.write(lengthBytes); out.write(b); } /** * Encodes a SASL Text to the current buffer. * * @param str the Text element. * @throws SaslEncodingException if an encoding size constraint is violated. * @throws SaslEncodingException if the UTF-8 encoding is not supported on * this platform. * @throws IOException if any other I/O exception occurs during the operation. */ public void setText(String str) throws IOException { byte[] b = str.getBytes("UTF8"); final int length = b.length; if (length > Registry.SASL_TWO_BYTE_MAX_LIMIT) throw new SaslEncodingException("SASL text too long"); byte[] lengthBytes = { (byte)(length >>> 8), (byte) length }; out.write(lengthBytes); out.write(b); } /** * Returns the encoded form of the current buffer including the 4-byte length * header. * * @throws SaslEncodingException if an encoding size constraint is violated. */ public byte[] encode() throws SaslEncodingException { byte[] buffer = wrap(); final int length = buffer.length; byte[] result = new byte[length + 4]; result[0] = (byte)(length >>> 24); result[1] = (byte)(length >>> 16); result[2] = (byte)(length >>> 8); result[3] = (byte) length; System.arraycopy(buffer, 0, result, 4, length); return result; } /** * Returns the encoded form of the current buffer excluding the 4-byte length * header. * * @throws SaslEncodingException if an encoding size constraint is violated. */ public byte[] wrap() throws SaslEncodingException { final int length = out.size(); if (length > Registry.SASL_BUFFER_MAX_LIMIT || length < 0) throw new SaslEncodingException("SASL buffer too long"); return out.toByteArray(); } }