forked from HBM/cppstream
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Matthias Loy
committed
Apr 12, 2018
1 parent
447a45c
commit 84697e5
Showing
2 changed files
with
119 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,56 @@ | ||
// Copyright 2014 Hottinger Baldwin Messtechnik | ||
// Distributed under MIT license | ||
// See file LICENSE provided | ||
|
||
|
||
#include <sys/socket.h> | ||
#include <sys/uio.h> | ||
|
||
#include <cstring> | ||
|
||
#include "bufferedreader.h" | ||
|
||
namespace hbm { | ||
BufferedReader::BufferedReader() | ||
: m_fillLevel(0) | ||
, m_alreadyRead(0) | ||
{ | ||
} | ||
|
||
ssize_t BufferedReader::recv(int sockfd, void *buf, size_t desiredLen) | ||
{ | ||
// check whether there is something left | ||
size_t bytesLeft = m_fillLevel - m_alreadyRead; | ||
|
||
if(bytesLeft>=desiredLen) { | ||
// there is more than or as much as desired | ||
memcpy(buf, m_buffer+m_alreadyRead, desiredLen); | ||
m_alreadyRead += desiredLen; | ||
return static_cast < ssize_t > (desiredLen); | ||
} else if(bytesLeft>0) { | ||
// return the rest which is less than desired (a short read) | ||
memcpy(buf, m_buffer+m_alreadyRead, bytesLeft); | ||
m_alreadyRead = m_fillLevel; | ||
return static_cast < ssize_t > (bytesLeft); | ||
} | ||
|
||
// try to read as much as possible into the provided buffer.In addition we fill our internal buffer if there is already more to read. | ||
// readv saves us from reading into the internal buffer first and copying into the provided buffer afterwards. | ||
struct iovec iov[2]; | ||
iov[0].iov_base = buf; | ||
iov[0].iov_len = desiredLen; | ||
iov[1].iov_base = m_buffer; | ||
iov[1].iov_len = sizeof(m_buffer); | ||
|
||
ssize_t retVal = ::readv(sockfd, iov, 2); | ||
m_alreadyRead = 0; | ||
|
||
if (retVal>static_cast < ssize_t > (desiredLen)) { | ||
// readv returns the total number of bytes read | ||
m_fillLevel = retVal-desiredLen; | ||
return static_cast < ssize_t > (desiredLen); | ||
} | ||
m_fillLevel = 0; | ||
return retVal; | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,63 @@ | ||
// Copyright 2014 Hottinger Baldwin Messtechnik | ||
// Distributed under MIT license | ||
// See file LICENSE provided | ||
|
||
#include <fstream> | ||
#include <stdexcept> | ||
#include <iostream> | ||
|
||
#include <WinSock2.h> | ||
#undef max | ||
#undef min | ||
|
||
#include <cstring> | ||
|
||
#include "bufferedreader.h" | ||
|
||
namespace hbm { | ||
BufferedReader::BufferedReader() | ||
: m_fillLevel(0) | ||
, m_alreadyRead(0) | ||
{ | ||
} | ||
|
||
ssize_t BufferedReader::recv(int sockfd, void *buf, size_t desiredLen) | ||
{ | ||
// check whether there is something left | ||
size_t bytesLeft = m_fillLevel - m_alreadyRead; | ||
if(bytesLeft>=desiredLen) { | ||
// there is more than or as much as desired | ||
memcpy(buf, m_buffer+m_alreadyRead, desiredLen); | ||
m_alreadyRead += desiredLen; | ||
return static_cast < ssize_t > (desiredLen); | ||
} else if(bytesLeft>0) { | ||
// return the rest which is less than desired (a short read) | ||
memcpy(buf, m_buffer+m_alreadyRead, bytesLeft); | ||
m_alreadyRead = m_fillLevel; | ||
return static_cast < ssize_t > (bytesLeft); | ||
} | ||
|
||
WSABUF buffers[2]; | ||
DWORD Flags = 0; | ||
DWORD numberOfBytesRecvd; | ||
buffers[0].buf = reinterpret_cast < CHAR* > (buf); | ||
buffers[0].len = desiredLen; | ||
buffers[1].buf = reinterpret_cast < CHAR* > (m_buffer); | ||
buffers[1].len = sizeof(m_buffer); | ||
|
||
int retVal = WSARecv(sockfd, buffers, 2, &numberOfBytesRecvd, &Flags, NULL, NULL); | ||
m_alreadyRead = 0; | ||
if (retVal < 0) { | ||
m_fillLevel = 0; | ||
return retVal; | ||
} | ||
|
||
if (numberOfBytesRecvd>static_cast < DWORD > (desiredLen)) { | ||
// WSARecv returns the total number of bytes read | ||
m_fillLevel = numberOfBytesRecvd - desiredLen; | ||
return static_cast < ssize_t > (desiredLen); | ||
} | ||
m_fillLevel = 0; | ||
return numberOfBytesRecvd; | ||
} | ||
} |