C++ (Cpp) ioBuf Examples

Example #1

File: JemallocNodumpAllocatorTest.cpp Project: GYGit/folly

TEST(JemallocNodumpAllocatorTest, IOBuf) {
  folly::JemallocNodumpAllocator jna;

#ifdef FOLLY_JEMALLOC_NODUMP_ALLOCATOR_SUPPORTED
  if (folly::usingJEMalloc()) {
    EXPECT_NE(0, jna.getArenaIndex());
  }
#endif // FOLLY_JEMALLOC_NODUMP_ALLOCATOR_SUPPORTED

  const size_t size{1024};
  void* ptr = jna.allocate(size);
  EXPECT_NE(nullptr, ptr);
  folly::IOBuf ioBuf(folly::IOBuf::TAKE_OWNERSHIP, ptr, size);
  EXPECT_EQ(size, ioBuf.capacity());
  EXPECT_EQ(ptr, ioBuf.data());
  uint8_t* data = ioBuf.writableData();
  EXPECT_EQ(ptr, data);
  for (auto i = 0u; i < ioBuf.capacity(); ++i) {
    data[i] = 'A';
  }
  uint8_t* p = static_cast<uint8_t*> (ptr);
  for (auto i = 0u; i < size; ++i) {
    EXPECT_EQ('A', p[i]);
  }
}

Example #2

File: miniftp.cpp Project: Ambrevar/fr_public

sBool sMiniFTPClient::GetFile(const sChar *filename,sFile *writeTo)
{
  sSize currentPos = 0;
  if(State == CS_NOTARGET)
    return sFALSE;

  for(;;)
  {
    if(State != CS_CONNECTED && !TryReconnect(RetryCount))
      break;

    sU8 sizeBuf[8];
    if(SendCommand(sMFC_GET,filename,currentPos)
      && ReadAll(sizeBuf,8))
    {
      sSize totalSize;
      sUnalignedLittleEndianLoad64(sizeBuf,(sU64&) totalSize);

      if(!Progress(filename,currentPos,totalSize,ProgressUser))
        return MaybeNextTime();

      sFixedArray<sU8> ioBuf(FileIOBufferSize);
      while(currentPos < totalSize)
      {
        sDInt size = (sDInt) sMin<sS64>(totalSize-currentPos,ioBuf.GetSize());
        sDInt read;

        if(!Socket.Read(&ioBuf[0],size,read) || !read)
        {
          MaybeNextTime();
          break;
        }

        if(!writeTo->Write(&ioBuf[0],read))
          return sFALSE;

        currentPos += read;
        if(!Progress(filename,currentPos,totalSize,ProgressUser))
          return MaybeNextTime();
      }

      if(currentPos == totalSize)
        return sTRUE;
    }
    else if(Error == sMFE_OK)
      MaybeNextTime();
    else
      break;
  }

  return sFALSE;
}

Example #3

File: miniftp.cpp Project: Ambrevar/fr_public

sBool sMiniFTPClient::PutFile(const sChar *filename,sFile *readFrom)
{
  sSize currentPos = 0;
  sSize totalSize = readFrom->GetSize();
  if(State == CS_NOTARGET)
    return sFALSE;

  for(;;)
  {
    if(State != CS_CONNECTED && !TryReconnect(RetryCount))
      break;

    if(!ContinueUploads)
      currentPos = 0;

    if(SendCommand(sMFC_PUT,filename,currentPos))
    {
      sU8 sizeBuf[8];
      sUnalignedLittleEndianStore64(sizeBuf,totalSize);
      if(!WriteAll(sizeBuf,8))
        continue;

      if(!Progress(filename,currentPos,totalSize,ProgressUser))
        return MaybeNextTime();

      sFixedArray<sU8> ioBuf(FileIOBufferSize);

      while(currentPos < totalSize)
      {
        sInt size = (sInt) sMin<sS64>(totalSize-currentPos,ioBuf.GetSize());
        if(!readFrom->Read(&ioBuf[0],size))
          return MaybeNextTime();

        if(!WriteAll(&ioBuf[0],size))
          break;

        currentPos += size;
        if(!Progress(filename,currentPos,totalSize,ProgressUser))
          return sFALSE;
      }

      if(currentPos == totalSize)
        return sTRUE;
    }
    else if(Error == sMFE_OK)
      MaybeNextTime();
    else
      break;
  }

  return sFALSE;
}

Example #4

File: miniftp.cpp Project: Ambrevar/fr_public

sBool sMiniFTPClient::ListFiles(const sChar *basepath,sArray<sChar> &listing)
{
  listing.Clear();
  if(State == CS_NOTARGET)
    return sFALSE;

  for(;;)
  {
    if(State != CS_CONNECTED && !TryReconnect(RetryCount))
      break;
  
    sU8 sizeBuf[8];
    if(SendCommand(sMFC_LIST,basepath,0) && ReadAll(sizeBuf,8))
    {
      sSize totalSize;
      sUnalignedLittleEndianLoad64(sizeBuf,(sU64&) totalSize);
      if((totalSize & 1)              // we expect 16bit chars
        || totalSize > 16*1024*1024)  // 16MB limit for directory listings (for now)
        return sFALSE;

      if(!Progress(basepath,0,totalSize,ProgressUser))
        return MaybeNextTime();

      sFixedArray<sU8> ioBuf((sInt) totalSize);
      sSize currentPos = 0;
      while(currentPos < totalSize)
      {
        sDInt size = (sDInt) (totalSize-currentPos);
        sDInt read;

        if(!Socket.Read(&ioBuf[(sInt) currentPos],size,read) || !read)
        {
          MaybeNextTime();
          break;
        }

        currentPos += read;
        if(!Progress(basepath,currentPos,totalSize,ProgressUser))
          return MaybeNextTime();
      }

      if(currentPos == totalSize)
      {
        listing.HintSize(sU32(totalSize/2));
        listing.AddMany(sU32(totalSize/2));

        for(sInt i=0;i<totalSize/2;i++)
        {
          sU16 v;
          sUnalignedLittleEndianLoad16(&ioBuf[i*2],v);
          listing[i] = v;
        }

        return sTRUE;
      }
    }
    else if(Error == sMFE_OK)
      MaybeNextTime();
    else
      break;
  }

  return sFALSE;
}