summaryrefslogtreecommitdiff
path: root/src/Sockets/tests/copy.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/Sockets/tests/copy.cpp')
-rw-r--r--src/Sockets/tests/copy.cpp264
1 files changed, 0 insertions, 264 deletions
diff --git a/src/Sockets/tests/copy.cpp b/src/Sockets/tests/copy.cpp
deleted file mode 100644
index 2134ac2..0000000
--- a/src/Sockets/tests/copy.cpp
+++ /dev/null
@@ -1,264 +0,0 @@
-#include <TcpSocket.h>
-#include <SocketHandler.h>
-#include <ListenSocket.h>
-#include <Utility.h>
-
-#define BUFSZ 7024
-
-#ifdef _WIN32
-#define MIN(a,b) (a<b?a:b)
-
-#endif
-
-static int quit = 0;
-static size_t lim = 120000;
-
-
-class CopySocket : public TcpSocket
-{
-public:
- CopySocket(ISocketHandler& h) : TcpSocket(h), m_fil(NULL), m_out(NULL), m_b_server(false), m_sz(0) {
- SetTransferLimit(60000);
- }
- CopySocket(ISocketHandler& h, size_t sz) : TcpSocket(h), m_fil(NULL), m_out(NULL), m_b_server(false), m_sz(sz) {
- SetTransferLimit(60000);
- quit++;
- }
- CopySocket(ISocketHandler& h, const std::string& filename) : TcpSocket(h), m_filename(filename), m_fil(NULL), m_out(NULL), m_b_server(false), m_sz(0) {
- SetTransferLimit(60000);
- quit++;
- }
- ~CopySocket() {
- quit--;
- }
-
- void OnConnect() {
- Utility::GetTime(&m_start);
- if (!m_filename.size())
- {
- Send("\n");
- char buf[BUFSZ];
- int n = MIN(m_sz, BUFSZ);
- while (n > 0 && GetOutputLength() < lim)
- {
- SendBuf(buf, n);
- m_sz -= n;
- n = MIN(m_sz, BUFSZ);
- }
- if (!n)
- {
- SetCloseAndDelete();
- }
- else
- {
- SendBuf(buf, n);
- m_sz -= n;
- }
- return;
- }
- size_t x = 0;
- for (size_t i = 0; i < m_filename.size(); i++)
- if (m_filename[i] == '/')
- x = i + 1;
- Send(m_filename.substr(x) + "\n");
- m_fil = fopen(m_filename.c_str(), "rb");
- if (m_fil)
- {
- char buf[BUFSZ];
- int n = fread(buf, 1, BUFSZ, m_fil);
- while (n > 0 && GetOutputLength() < lim)
- {
- SendBuf(buf, n);
- n = fread(buf, 1, BUFSZ, m_fil);
- }
- if (!n)
- {
- SetCloseAndDelete();
- fclose(m_fil);
- m_fil = NULL;
- }
- else
- {
- SendBuf(buf, n);
- }
- }
- }
-
- void OnWriteComplete() {
- }
-
- void OnTransferLimit() {
- if (!m_filename.size())
- {
- char buf[BUFSZ];
- int n = MIN(m_sz, BUFSZ);
- while (n > 0 && GetOutputLength() < lim)
- {
- SendBuf(buf, n);
- m_sz -= n;
- n = MIN(m_sz, BUFSZ);
- }
- if (!n)
- {
- SetCloseAndDelete();
- }
- else
- {
- SendBuf(buf, n);
- m_sz -= n;
- }
- return;
- }
- if (m_fil)
- {
- char buf[BUFSZ];
- int n = fread(buf, 1, BUFSZ, m_fil);
- while (n > 0 && GetOutputLength() < lim)
- {
- SendBuf(buf, n);
- n = fread(buf, 1, BUFSZ, m_fil);
- }
- if (!n)
- {
- SetCloseAndDelete();
- fclose(m_fil);
- m_fil = NULL;
- }
- else
- {
- SendBuf(buf, n);
- }
- }
- }
-
- void OnAccept() {
- Utility::GetTime(&m_start);
- m_b_server = true;
- SetLineProtocol();
- DisableInputBuffer();
- }
-
- void OnLine(const std::string& line) {
- if (line.size())
- m_out = fopen(line.c_str(), "wb");
- SetLineProtocol(false);
- DisableInputBuffer();
- }
-
- void OnDelete() {
- struct timeval stop;
- Utility::GetTime(&stop);
- stop.tv_sec -= m_start.tv_sec;
- stop.tv_usec -= m_start.tv_usec;
- if (stop.tv_usec < 0)
- {
- stop.tv_usec += 1000000;
- stop.tv_sec -= 1;
- }
- double t = stop.tv_usec;
- t /= 1000000;
- t += stop.tv_sec;
- printf("OnDelete: %s\n", m_b_server ? "SERVER" : "CLIENT");
- printf(" Time: %ld.%06ld (%f)\n", stop.tv_sec, stop.tv_usec, t);
- double r = GetBytesReceived();
- printf(" bytes in: %lld (%f Mbytes/sec)\n", GetBytesReceived(), r / t / 1000000);
- double s = GetBytesSent();
- printf(" bytes out: %lld (%f Mbytes/sec)\n", GetBytesSent(), s / t / 1000000);
- printf("\n");
- if (m_out)
- fclose(m_out);
- }
-
- void OnRawData(const char *buf, size_t len) {
- if (m_out)
- fwrite(buf, 1, len, m_out);
- }
-
-private:
- std::string m_filename;
- FILE *m_fil;
- FILE *m_out;
- bool m_b_server;
- struct timeval m_start;
- size_t m_sz;
-};
-
-
-int main(int argc, char *argv[])
-{
- std::string host = "127.0.0.1";
- int port = 12344;
- std::list<std::string> filenames;
-
- for (int i = 1; i < argc; i++)
- {
- if (!strcmp(argv[i], "-host") && i < argc - 1)
- host = argv[++i];
- else
- if (!strcmp(argv[i], "-port") && i < argc - 1)
- port = atoi(argv[++i]);
- else
- if (!strcmp(argv[i], "-h"))
- {
- fprintf(stderr, "Usage: %s [-host <host>] [-port <port>] [<file to send>]\n", *argv);
- fprintf(stderr, " Will run as host only if <file to send> isn't specified.\n");
- fprintf(stderr, " host default: 127.0.0.1\n");
- fprintf(stderr, " port default: 12344\n");
- return 0;
- }
- else
- {
- filenames.push_back( argv[i] );
- }
- }
-
- try
- {
- SocketHandler h;
- ListenSocket<CopySocket> l(h);
- if (filenames.empty())
- {
- if (l.Bind( port ) != 0)
- {
- fprintf(stderr, "Bind() port %d failed - exiting\n", port);
- return -1;
- }
- h.Add(&l);
- }
- for (std::list<std::string>::iterator it = filenames.begin(); it != filenames.end(); it++)
- {
- std::string filename = *it;
- size_t sz = atol(filename.c_str());
- if (sz)
- {
- CopySocket *sock = new CopySocket(h, sz);
- sock -> SetDeleteByHandler();
- sock -> Open(host, port);
- h.Add(sock);
- }
- else
- {
- CopySocket *sock = new CopySocket(h, filename);
- sock -> SetDeleteByHandler();
- sock -> Open(host, port);
- h.Add(sock);
- }
- }
- if (!filenames.size())
- {
- fprintf(stderr, "Starting as server only, listening on port %d\n", port);
- quit++;
- }
- while (quit > 0)
- {
- h.Select(5, 0);
- }
- return 0;
- }
- catch (const Exception& e)
- {
- printf("%s\n", e.ToString().c_str());
- }
-}
-
-