Imanol-Mikel Barba Sabariego / PAD · Files

package pad.prac2;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

import java.io.OutputStream;

import java.net.InetSocketAddress;
import java.net.Socket;
import java.net.SocketAddress;

import java.nio.charset.Charset;

public class MySocket extends Socket
{
	private BufferedReader input;

	private OutputStream output;

	public MySocket()
	{
		super();

	}

	public void connect(String host, int port)
	{
		SocketAddress addr = new InetSocketAddress(host,port);

		try
		{

			super.connect(addr);
			initializeStreams();

		}
		catch(IOException ioExc)
		{

			System.out.println("TCP: Error occured while connecting to remote host");

		}
	}

	public void initializeStreams()

	{
		try
		{

			output = super.getOutputStream();

			input = new BufferedReader(new InputStreamReader(super.getInputStream()));

		}
		catch(IOException ioExc)
		{

			System.out.println("TCP: Error initializing socket");

		}
	}

	public void sendMsg(String msg) throws IOException

	{
		String length = new Integer(msg.length()).toString();

		this.write(length + '\0' + msg);

	}

	public String recvMsg() throws IOException

	{
		String len,str;
		int length;
		len = this.readLine();

		if(len == "")

		{
			return len;
		}
		length = Integer.parseInt(len);
		str = this.read(length);
		return str;
	}

	public void write(String str) throws IOException

	{

		output.write(str.getBytes("UTF-8"));

	}

	public String read(int bytes)
	{
		char[] buffer = new char[bytes];
		try
		{
			input.read(buffer, 0, bytes);

			byte[] bytesRead = new String(buffer).getBytes("UTF-8"); //Cosas del Unicode, por culpa del **** Java
			return new String(bytesRead,"UTF-8");

		}
		catch(IOException ioExc)
		{
			System.out.println("TCP: Error retrieving data from remote endpoint");

			return null;
		}
	}

	public String readLine() throws IOException

	{

		String line = "";
		char[] c = new char[1];
		while(true)

		{

			input.read(c,0,1);
			if(c[0] == '\0')

			{

				break;

			}

			line += c[0];
		}
		return line;
	}

	public void close()
	{
		try
		{
			super.shutdownInput();
			super.shutdownOutput();
			super.close();

		}
		catch(IOException ioExc)
		{

			System.out.println("TCP: Error while closing socket");

		}
	}
}