mirror of
https://github.com/Brawl345/customizemii.git
synced 2024-11-15 04:45:09 +01:00
268 lines
11 KiB
C#
268 lines
11 KiB
C#
|
/* This file is part of TransmitMii
|
|||
|
|
* Copyright (C) 2009 Leathl
|
|||
|
|
*
|
|||
|
|
* TransmitMii 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 3 of the License, or
|
|||
|
|
* (at your option) any later version.
|
|||
|
|
*
|
|||
|
|
* TransmitMii 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
using System;
|
|||
|
|
using System.Net.Sockets;
|
|||
|
|
|
|||
|
|
namespace TransmitMii
|
|||
|
|
{
|
|||
|
|
partial class TransmitMii_Main
|
|||
|
|
{
|
|||
|
|
TcpClient theClient;
|
|||
|
|
NetworkStream theStream;
|
|||
|
|
System.Diagnostics.Stopwatch stopper = new System.Diagnostics.Stopwatch();
|
|||
|
|
private double transmittedLength;
|
|||
|
|
private int timeElapsed;
|
|||
|
|
private bool usedCompression;
|
|||
|
|
private double compressionRatio;
|
|||
|
|
|
|||
|
|
private bool Transmit_Compress(string fileName, byte[] fileData, bool JODI, bool compress)
|
|||
|
|
{
|
|||
|
|
stopper.Reset(); stopper.Start();
|
|||
|
|
|
|||
|
|
if (!Environment.OSVersion.ToString().Contains("Windows"))
|
|||
|
|
compress = false;
|
|||
|
|
|
|||
|
|
theClient = new TcpClient();
|
|||
|
|
|
|||
|
|
byte[] compFileData;
|
|||
|
|
int Blocksize = 4 * 1024;
|
|||
|
|
if (!JODI) Blocksize = 16 * 1024;
|
|||
|
|
byte[] buffer = new byte[4];
|
|||
|
|
string theIP = tbIP.Text;
|
|||
|
|
|
|||
|
|
StatusUpdate("Connecting...");
|
|||
|
|
try { theClient.Connect(theIP, 4299); }
|
|||
|
|
catch (Exception ex) { if (!Aborted) ErrorBox("Connection Failed:\n" + ex.Message); theClient.Close(); return false; }
|
|||
|
|
theStream = theClient.GetStream();
|
|||
|
|
|
|||
|
|
StatusUpdate("Connected... Sending Magic...");
|
|||
|
|
buffer[0] = (byte)'H';
|
|||
|
|
buffer[1] = (byte)'A';
|
|||
|
|
buffer[2] = (byte)'X';
|
|||
|
|
buffer[3] = (byte)'X';
|
|||
|
|
try { theStream.Write(buffer, 0, 4); }
|
|||
|
|
catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Magic:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
|
|||
|
|
StatusUpdate("Magic Sent... Sending Version Info...");
|
|||
|
|
if (JODI)
|
|||
|
|
{
|
|||
|
|
buffer[0] = 0;
|
|||
|
|
buffer[1] = 5;
|
|||
|
|
buffer[2] = (byte)(((fileName.Length + 2) >> 8) & 0xff);
|
|||
|
|
buffer[3] = (byte)((fileName.Length + 2) & 0xff);
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
buffer[0] = 0;
|
|||
|
|
buffer[1] = 1;
|
|||
|
|
buffer[2] = 0;
|
|||
|
|
buffer[3] = 0;
|
|||
|
|
}
|
|||
|
|
try { theStream.Write(buffer, 0, 4); }
|
|||
|
|
catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Version Info:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
|
|||
|
|
if (JODI && compress)
|
|||
|
|
{
|
|||
|
|
StatusUpdate("Version Info Sent... Compressing File...");
|
|||
|
|
try { compFileData = zlib.Compress(fileData); }
|
|||
|
|
catch
|
|||
|
|
{
|
|||
|
|
//Compression failed, let's continue without compression
|
|||
|
|
if (!Aborted) ErrorBox("a");
|
|||
|
|
compFileData = fileData;
|
|||
|
|
fileData = new byte[0];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
StatusUpdate("Compressed File... Sending Filesize...");
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
compFileData = fileData;
|
|||
|
|
fileData = new byte[0];
|
|||
|
|
|
|||
|
|
StatusUpdate("Version Info Sent... Sending Filesize...");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//First compressed filesize, then uncompressed filesize
|
|||
|
|
buffer[0] = (byte)((compFileData.Length >> 24) & 0xff);
|
|||
|
|
buffer[1] = (byte)((compFileData.Length >> 16) & 0xff);
|
|||
|
|
buffer[2] = (byte)((compFileData.Length >> 8) & 0xff);
|
|||
|
|
buffer[3] = (byte)(compFileData.Length & 0xff);
|
|||
|
|
try { theStream.Write(buffer, 0, 4); }
|
|||
|
|
catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Filesize:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
|
|||
|
|
if (JODI)
|
|||
|
|
{
|
|||
|
|
buffer[0] = (byte)((fileData.Length >> 24) & 0xff);
|
|||
|
|
buffer[1] = (byte)((fileData.Length >> 16) & 0xff);
|
|||
|
|
buffer[2] = (byte)((fileData.Length >> 8) & 0xff);
|
|||
|
|
buffer[3] = (byte)(fileData.Length & 0xff);
|
|||
|
|
try { theStream.Write(buffer, 0, 4); }
|
|||
|
|
catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Filesize:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
StatusUpdate("Filesize Sent... Sending File...");
|
|||
|
|
int off = 0;
|
|||
|
|
int cur = 0;
|
|||
|
|
int count = compFileData.Length / Blocksize;
|
|||
|
|
int left = compFileData.Length % Blocksize;
|
|||
|
|
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
do
|
|||
|
|
{
|
|||
|
|
StatusUpdate("Sending File: " + ((cur + 1) * 100 / count).ToString() + "%");
|
|||
|
|
theStream.Write(compFileData, off, Blocksize);
|
|||
|
|
off += Blocksize;
|
|||
|
|
cur++;
|
|||
|
|
} while (cur < count);
|
|||
|
|
|
|||
|
|
if (left > 0)
|
|||
|
|
{
|
|||
|
|
theStream.Write(compFileData, off, compFileData.Length - off);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
catch (Exception ex) { if (!Aborted) ErrorBox("Error sending File:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
|
|||
|
|
if (JODI)
|
|||
|
|
{
|
|||
|
|
StatusUpdate("File Sent... Sending Arguments...");
|
|||
|
|
byte[] theArgs = new byte[fileName.Length + 2];
|
|||
|
|
for (int i = 0; i < fileName.Length; i++) { theArgs[i] = (byte)fileName.ToCharArray()[i]; }
|
|||
|
|
try { theStream.Write(theArgs, 0, theArgs.Length); }
|
|||
|
|
catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Arguments:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
theStream.Close();
|
|||
|
|
theClient.Close();
|
|||
|
|
|
|||
|
|
StatusUpdate(string.Empty);
|
|||
|
|
|
|||
|
|
stopper.Stop();
|
|||
|
|
|
|||
|
|
this.timeElapsed = (int)stopper.ElapsedMilliseconds;
|
|||
|
|
this.usedCompression = compress;
|
|||
|
|
this.transmittedLength = Math.Round(compFileData.Length * 0.0009765625, 2);
|
|||
|
|
if (compress && fileData.Length != 0)
|
|||
|
|
this.compressionRatio = (compFileData.Length * 100) / fileData.Length;
|
|||
|
|
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//private bool Transmit(string fileName, byte[] fileData, bool JODI)
|
|||
|
|
//{
|
|||
|
|
// TcpClient theClient = new TcpClient();
|
|||
|
|
// NetworkStream theStream;
|
|||
|
|
|
|||
|
|
// int Blocksize = 4 * 1024;
|
|||
|
|
// if (!JODI) Blocksize = 16 * 1024;
|
|||
|
|
// byte[] buffer = new byte[4];
|
|||
|
|
// string theIP = tbIP.Text;
|
|||
|
|
|
|||
|
|
// StatusUpdate("Connecting...");
|
|||
|
|
// try { theClient.Connect(theIP, 4299); }
|
|||
|
|
// catch (Exception ex) { if (!Aborted) ErrorBox("Connection Failed:\n" + ex.Message); theClient.Close(); return false; }
|
|||
|
|
// theStream = theClient.GetStream();
|
|||
|
|
|
|||
|
|
// StatusUpdate("Connected... Sending Magic...");
|
|||
|
|
// buffer[0] = (byte)'H';
|
|||
|
|
// buffer[1] = (byte)'A';
|
|||
|
|
// buffer[2] = (byte)'X';
|
|||
|
|
// buffer[3] = (byte)'X';
|
|||
|
|
// try { theStream.Write(buffer, 0, 4); }
|
|||
|
|
// catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Magic:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
|
|||
|
|
// StatusUpdate("Magic Sent... Sending Version Info...");
|
|||
|
|
// if (JODI)
|
|||
|
|
// {
|
|||
|
|
// buffer[0] = 0;
|
|||
|
|
// buffer[1] = 5;
|
|||
|
|
// buffer[2] = (byte)(((fileName.Length + 2) >> 8) & 0xff);
|
|||
|
|
// buffer[3] = (byte)((fileName.Length + 2) & 0xff);
|
|||
|
|
// }
|
|||
|
|
// else
|
|||
|
|
// {
|
|||
|
|
// buffer[0] = 0;
|
|||
|
|
// buffer[1] = 1;
|
|||
|
|
// buffer[2] = 0;
|
|||
|
|
// buffer[3] = 0;
|
|||
|
|
// }
|
|||
|
|
// try { theStream.Write(buffer, 0, 4); }
|
|||
|
|
// catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Version Info:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
|
|||
|
|
// StatusUpdate("Version Info Sent... Sending Filesize...");
|
|||
|
|
// //First compressed filesize, then uncompressed filesize
|
|||
|
|
// buffer[0] = (byte)((fileData.Length >> 24) & 0xff);
|
|||
|
|
// buffer[1] = (byte)((fileData.Length >> 16) & 0xff);
|
|||
|
|
// buffer[2] = (byte)((fileData.Length >> 8) & 0xff);
|
|||
|
|
// buffer[3] = (byte)(fileData.Length & 0xff);
|
|||
|
|
// try { theStream.Write(buffer, 0, 4); }
|
|||
|
|
// catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Filesize:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
|
|||
|
|
// if (JODI)
|
|||
|
|
// {
|
|||
|
|
// buffer[0] = 0;
|
|||
|
|
// buffer[1] = 0;
|
|||
|
|
// buffer[2] = 0;
|
|||
|
|
// buffer[3] = 0;
|
|||
|
|
// try { theStream.Write(buffer, 0, 4); }
|
|||
|
|
// catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Filesize:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
// }
|
|||
|
|
|
|||
|
|
// StatusUpdate("Filesize Sent... Sending File...");
|
|||
|
|
// int off = 0;
|
|||
|
|
// int cur = 0;
|
|||
|
|
// int count = fileData.Length / Blocksize;
|
|||
|
|
// int left = fileData.Length % Blocksize;
|
|||
|
|
|
|||
|
|
// try
|
|||
|
|
// {
|
|||
|
|
// do
|
|||
|
|
// {
|
|||
|
|
// StatusUpdate("Sending File: " + ((cur + 1) * 100 / count).ToString() + "%");
|
|||
|
|
// theStream.Write(fileData, off, Blocksize);
|
|||
|
|
// off += Blocksize;
|
|||
|
|
// cur++;
|
|||
|
|
// } while (cur < count);
|
|||
|
|
|
|||
|
|
// if (left > 0)
|
|||
|
|
// {
|
|||
|
|
// theStream.Write(fileData, off, fileData.Length - off);
|
|||
|
|
// }
|
|||
|
|
// }
|
|||
|
|
// catch (Exception ex) { if (!Aborted) ErrorBox("Error sending File:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
|
|||
|
|
// if (JODI)
|
|||
|
|
// {
|
|||
|
|
// StatusUpdate("File Sent... Sending Arguments...");
|
|||
|
|
// byte[] theArgs = new byte[fileName.Length + 2];
|
|||
|
|
// for (int i = 0; i < fileName.Length; i++) { theArgs[i] = (byte)fileName.ToCharArray()[i]; }
|
|||
|
|
// try { theStream.Write(theArgs, 0, theArgs.Length); }
|
|||
|
|
// catch (Exception ex) { if (!Aborted) ErrorBox("Error sending Arguments:\n" + ex.Message); theStream.Close(); theClient.Close(); return false; }
|
|||
|
|
// }
|
|||
|
|
|
|||
|
|
// theStream.Close();
|
|||
|
|
// theClient.Close();
|
|||
|
|
|
|||
|
|
// StatusUpdate(string.Empty);
|
|||
|
|
|
|||
|
|
// return true;
|
|||
|
|
//}
|
|||
|
|
}
|
|||
|
|
}
|