summaryrefslogtreecommitdiffstats
path: root/EssentialsUpdate/src/net/ess3/update/GetFile.java
blob: 9636e3bd13becac0ac0589026c2651848f5f1ac9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package net.ess3.update;

import java.io.*;
import java.math.BigInteger;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.logging.Level;
import org.bukkit.Bukkit;


public class GetFile
{
	private transient URLConnection connection;
	private transient MessageDigest digest;

	public GetFile(final String urlString) throws MalformedURLException, IOException
	{
		this(new URL(urlString));
	}

	public GetFile(final URL url) throws IOException
	{
		this.connection = url.openConnection();
		this.connection.setConnectTimeout(1000);
		this.connection.setReadTimeout(5000);
		this.connection.setUseCaches(false);
		this.connection.connect();
		final int respCode = ((HttpURLConnection)this.connection).getResponseCode();
		if (respCode >= 300 && respCode < 400 && this.connection.getHeaderField("Location") != null)
		{
			connection.getInputStream().close();
			final URL redirect = new URL(this.connection.getHeaderField("Location"));
			this.connection = redirect.openConnection();
			this.connection.setConnectTimeout(1000);
			this.connection.setReadTimeout(5000);
			this.connection.setUseCaches(false);
			this.connection.connect();
		}
	}

	public void saveTo(final File file) throws IOException
	{
		try
		{
			saveTo(file, null);
		}
		catch (NoSuchAlgorithmException ex)
		{
			throw new RuntimeException(ex);
		}
	}

	public void saveTo(final File file, final String key) throws IOException, NoSuchAlgorithmException
	{
		if (key != null)
		{
			digest = MessageDigest.getInstance("SHA256");
		}
		final byte[] buffer = new byte[1024 * 8];
		boolean brokenFile = false;
		final BufferedInputStream input = new BufferedInputStream(connection.getInputStream());
		try
		{
			final BufferedOutputStream output = new BufferedOutputStream(new FileOutputStream(file));
			try
			{
				int length;
				do
				{
					length = input.read(buffer);
					if (length >= 0)
					{
						if (key != null)
						{
							digest.update(buffer, 0, length);
						}
						output.write(buffer, 0, length);
					}
				}
				while (length >= 0);
				if (key != null)
				{
					final byte[] checksum = digest.digest();
					final String checksumString = new BigInteger(checksum).toString(36);
					if (!checksumString.equals(key))
					{
						brokenFile = true;
					}
				}
			}
			finally
			{
				output.close();
			}
			if (brokenFile && !file.delete())
			{
				Bukkit.getLogger().log(Level.SEVERE, "Could not delete file {0}", file.getPath());
			}
		}
		finally
		{
			input.close();
		}
		if (brokenFile)
		{
			throw new IOException("Checksum check failed.");
		}
	}
}