// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- // vim: ts=8 sw=2 smarttab /* * Ceph - scalable distributed file system * * Copyright (C) 2013 Cloudwatt * * Author: Loic Dachary * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU Library Public License as published by * the Free Software Foundation; either version 2, or (at your option) * any later version. * * This program 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 Library Public License for more details. * */ #include "include/memory.h" #include #include #include #include "include/buffer.h" #include "include/utime.h" #include "include/coredumpctl.h" #include "include/encoding.h" #include "common/environment.h" #include "common/Clock.h" #include "common/safe_io.h" #include "gtest/gtest.h" #include "stdlib.h" #include "fcntl.h" #include "sys/stat.h" #include "include/crc32c.h" #include "common/sctp_crc32.h" #define MAX_TEST 1000000 #define FILENAME "bufferlist" static char cmd[128]; TEST(Buffer, constructors) { bool ceph_buffer_track = get_env_bool("CEPH_BUFFER_TRACK"); unsigned len = 17; uint64_t history_alloc_bytes = 0; uint64_t history_alloc_num = 0; // // buffer::create // if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } { bufferptr ptr(buffer::create(len)); history_alloc_bytes += len; history_alloc_num++; EXPECT_EQ(len, ptr.length()); if (ceph_buffer_track) { EXPECT_EQ(len, (unsigned)buffer::get_total_alloc()); EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } } // // buffer::claim_char // if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } { char* str = new char[len]; ::memset(str, 'X', len); bufferptr ptr(buffer::claim_char(len, str)); if (ceph_buffer_track) { EXPECT_EQ(len, (unsigned)buffer::get_total_alloc()); EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } EXPECT_EQ(len, ptr.length()); EXPECT_EQ(str, ptr.c_str()); bufferptr clone = ptr.clone(); history_alloc_bytes += len; history_alloc_num++; EXPECT_EQ(0, ::memcmp(clone.c_str(), ptr.c_str(), len)); } // // buffer::create_static // if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } { char* str = new char[len]; bufferptr ptr(buffer::create_static(len, str)); if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } EXPECT_EQ(len, ptr.length()); EXPECT_EQ(str, ptr.c_str()); delete [] str; } // // buffer::create_malloc // if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } { bufferptr ptr(buffer::create_malloc(len)); history_alloc_bytes += len; history_alloc_num++; if (ceph_buffer_track) { EXPECT_EQ(len, (unsigned)buffer::get_total_alloc()); EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } EXPECT_EQ(len, ptr.length()); // this doesn't throw on my x86_64 wheezy box --sage //EXPECT_THROW(buffer::create_malloc((unsigned)ULLONG_MAX), buffer::bad_alloc); } // // buffer::claim_malloc // if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } { char* str = (char*)malloc(len); ::memset(str, 'X', len); bufferptr ptr(buffer::claim_malloc(len, str)); if (ceph_buffer_track) { EXPECT_EQ(len, (unsigned)buffer::get_total_alloc()); EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } EXPECT_EQ(len, ptr.length()); EXPECT_EQ(str, ptr.c_str()); bufferptr clone = ptr.clone(); history_alloc_bytes += len; history_alloc_num++; EXPECT_EQ(0, ::memcmp(clone.c_str(), ptr.c_str(), len)); } // // buffer::copy // if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } { const std::string expected(len, 'X'); bufferptr ptr(buffer::copy(expected.c_str(), expected.size())); history_alloc_bytes += len; history_alloc_num++; if (ceph_buffer_track) { EXPECT_EQ(len, (unsigned)buffer::get_total_alloc()); EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } EXPECT_NE(expected.c_str(), ptr.c_str()); EXPECT_EQ(0, ::memcmp(expected.c_str(), ptr.c_str(), len)); } // // buffer::create_page_aligned // if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } { bufferptr ptr(buffer::create_page_aligned(len)); history_alloc_bytes += len; history_alloc_num++; ::memset(ptr.c_str(), 'X', len); if (ceph_buffer_track) { EXPECT_EQ(len, (unsigned)buffer::get_total_alloc()); EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } // doesn't throw on my x86_64 wheezy box --sage //EXPECT_THROW(buffer::create_page_aligned((unsigned)ULLONG_MAX), buffer::bad_alloc); #ifndef DARWIN ASSERT_TRUE(ptr.is_page_aligned()); #endif // DARWIN bufferptr clone = ptr.clone(); history_alloc_bytes += len; history_alloc_num++; EXPECT_EQ(0, ::memcmp(clone.c_str(), ptr.c_str(), len)); if (ceph_buffer_track) { EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } } #ifdef CEPH_HAVE_SPLICE if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } { // no fd EXPECT_THROW(buffer::create_zero_copy(len, -1, NULL), buffer::error_code); history_alloc_bytes += len; history_alloc_num++; unsigned zc_len = 4; ::unlink(FILENAME); snprintf(cmd, sizeof(cmd), "echo ABC > %s", FILENAME); EXPECT_EQ(0, ::system(cmd)); int fd = ::open(FILENAME, O_RDONLY); bufferptr ptr(buffer::create_zero_copy(zc_len, fd, NULL)); history_alloc_bytes += zc_len; history_alloc_num++; EXPECT_EQ(zc_len, ptr.length()); if (ceph_buffer_track) { EXPECT_EQ(zc_len, (unsigned)buffer::get_total_alloc()); EXPECT_EQ(history_alloc_bytes, buffer::get_history_alloc_bytes()); EXPECT_EQ(history_alloc_num, buffer::get_history_alloc_num()); } ::close(fd); ::unlink(FILENAME); } #endif if (ceph_buffer_track) { EXPECT_EQ(0, buffer::get_total_alloc()); } } void bench_buffer_alloc(int size, int num) { utime_t start = ceph_clock_now(); for (int i=0; i %s", FILENAME); EXPECT_EQ(0, ::system(cmd)); fd = ::open(FILENAME, O_RDONLY); assert(fd >= 0); } void TearDown() override { ::close(fd); ::unlink(FILENAME); } int fd; unsigned len; }; TEST_F(TestRawPipe, create_zero_copy) { bufferptr ptr(buffer::create_zero_copy(len, fd, NULL)); EXPECT_EQ(len, ptr.length()); if (get_env_bool("CEPH_BUFFER_TRACK")) { EXPECT_EQ(len, (unsigned)buffer::get_total_alloc()); } } TEST_F(TestRawPipe, c_str_no_fd) { EXPECT_THROW(bufferptr ptr(buffer::create_zero_copy(len, -1, NULL)), buffer::error_code); } TEST_F(TestRawPipe, c_str_basic) { bufferptr ptr = bufferptr(buffer::create_zero_copy(len, fd, NULL)); EXPECT_EQ(0, memcmp(ptr.c_str(), "ABC\n", len)); EXPECT_EQ(len, ptr.length()); } TEST_F(TestRawPipe, c_str_twice) { // make sure we're creating a copy of the data and not consuming it bufferptr ptr = bufferptr(buffer::create_zero_copy(len, fd, NULL)); EXPECT_EQ(len, ptr.length()); EXPECT_EQ(0, memcmp(ptr.c_str(), "ABC\n", len)); EXPECT_EQ(0, memcmp(ptr.c_str(), "ABC\n", len)); } TEST_F(TestRawPipe, c_str_basic_offset) { loff_t offset = len - 1; ::lseek(fd, offset, SEEK_SET); bufferptr ptr = bufferptr(buffer::create_zero_copy(len - offset, fd, NULL)); EXPECT_EQ(len - offset, ptr.length()); EXPECT_EQ(0, memcmp(ptr.c_str(), "\n", len - offset)); } TEST_F(TestRawPipe, c_str_dest_short) { ::lseek(fd, 1, SEEK_SET); bufferptr ptr = bufferptr(buffer::create_zero_copy(2, fd, NULL)); EXPECT_EQ(2u, ptr.length()); EXPECT_EQ(0, memcmp(ptr.c_str(), "BC", 2)); } TEST_F(TestRawPipe, c_str_source_short) { ::lseek(fd, 1, SEEK_SET); bufferptr ptr = bufferptr(buffer::create_zero_copy(len, fd, NULL)); EXPECT_EQ(len - 1, ptr.length()); EXPECT_EQ(0, memcmp(ptr.c_str(), "BC\n", len - 1)); } TEST_F(TestRawPipe, c_str_explicit_zero_offset) { int64_t offset = 0; ::lseek(fd, 1, SEEK_SET); bufferptr ptr = bufferptr(buffer::create_zero_copy(len, fd, &offset)); EXPECT_EQ(len, offset); EXPECT_EQ(len, ptr.length()); EXPECT_EQ(0, memcmp(ptr.c_str(), "ABC\n", len)); } TEST_F(TestRawPipe, c_str_explicit_positive_offset) { int64_t offset = 1; bufferptr ptr = bufferptr(buffer::create_zero_copy(len - offset, fd, &offset)); EXPECT_EQ(len, offset); EXPECT_EQ(len - 1, ptr.length()); EXPECT_EQ(0, memcmp(ptr.c_str(), "BC\n", len - 1)); } TEST_F(TestRawPipe, c_str_explicit_positive_empty_result) { int64_t offset = len; bufferptr ptr = bufferptr(buffer::create_zero_copy(len - offset, fd, &offset)); EXPECT_EQ(len, offset); EXPECT_EQ(0u, ptr.length()); } TEST_F(TestRawPipe, c_str_source_short_explicit_offset) { int64_t offset = 1; bufferptr ptr = bufferptr(buffer::create_zero_copy(len, fd, &offset)); EXPECT_EQ(len, offset); EXPECT_EQ(len - 1, ptr.length()); EXPECT_EQ(0, memcmp(ptr.c_str(), "BC\n", len - 1)); } TEST_F(TestRawPipe, c_str_dest_short_explicit_offset) { int64_t offset = 1; bufferptr ptr = bufferptr(buffer::create_zero_copy(2, fd, &offset)); EXPECT_EQ(3, offset); EXPECT_EQ(2u, ptr.length()); EXPECT_EQ(0, memcmp(ptr.c_str(), "BC", 2)); } TEST_F(TestRawPipe, buffer_list_read_fd_zero_copy) { bufferlist bl; EXPECT_EQ(-EBADF, bl.read_fd_zero_copy(-1, len)); bl = bufferlist(); EXPECT_EQ(0, bl.read_fd_zero_copy(fd, len)); EXPECT_EQ(len, bl.length()); EXPECT_EQ(0u, bl.front().unused_tail_length()); EXPECT_EQ(1u, bl.get_num_buffers()); EXPECT_EQ(len, bl.front().raw_length()); EXPECT_EQ(0, memcmp(bl.c_str(), "ABC\n", len)); EXPECT_TRUE(bl.can_zero_copy()); } TEST_F(TestRawPipe, buffer_list_write_fd_zero_copy) { ::unlink(FILENAME); bufferlist bl; EXPECT_EQ(0, bl.read_fd_zero_copy(fd, len)); EXPECT_TRUE(bl.can_zero_copy()); int out_fd = ::open(FILENAME, O_RDWR|O_CREAT|O_TRUNC, 0600); EXPECT_EQ(0, bl.write_fd_zero_copy(out_fd)); struct stat st; memset(&st, 0, sizeof(st)); EXPECT_EQ(0, ::stat(FILENAME, &st)); EXPECT_EQ(len, st.st_size); char buf[len + 1]; EXPECT_EQ((int)len, safe_read(out_fd, buf, len + 1)); EXPECT_EQ(0, memcmp(buf, "ABC\n", len)); ::close(out_fd); ::unlink(FILENAME); } #endif // CEPH_HAVE_SPLICE // // +-----------+ +-----+ // | | | | // | offset +----------------+ | // | | | | // | length +---- | | // | | \------- | | // +-----------+ \---+ | // | ptr | +-----+ // +-----------+ | raw | // +-----+ // TEST(BufferPtr, constructors) { unsigned len = 17; // // ptr::ptr() // { buffer::ptr ptr; EXPECT_FALSE(ptr.have_raw()); EXPECT_EQ((unsigned)0, ptr.offset()); EXPECT_EQ((unsigned)0, ptr.length()); } // // ptr::ptr(raw *r) // { bufferptr ptr(buffer::create(len)); EXPECT_TRUE(ptr.have_raw()); EXPECT_EQ((unsigned)0, ptr.offset()); EXPECT_EQ(len, ptr.length()); EXPECT_EQ(ptr.raw_length(), ptr.length()); EXPECT_EQ(1, ptr.raw_nref()); } // // ptr::ptr(unsigned l) // { bufferptr ptr(len); EXPECT_TRUE(ptr.have_raw()); EXPECT_EQ((unsigned)0, ptr.offset()); EXPECT_EQ(len, ptr.length()); EXPECT_EQ(1, ptr.raw_nref()); } // // ptr(const char *d, unsigned l) // { const std::string str(len, 'X'); bufferptr ptr(str.c_str(), len); EXPECT_TRUE(ptr.have_raw()); EXPECT_EQ((unsigned)0, ptr.offset()); EXPECT_EQ(len, ptr.length()); EXPECT_EQ(1, ptr.raw_nref()); EXPECT_EQ(0, ::memcmp(str.c_str(), ptr.c_str(), len)); } // // ptr(const ptr& p) // { const std::string str(len, 'X'); bufferptr original(str.c_str(), len); bufferptr ptr(original); EXPECT_TRUE(ptr.have_raw()); EXPECT_EQ(original.get_raw(), ptr.get_raw()); EXPECT_EQ(2, ptr.raw_nref()); EXPECT_EQ(0, ::memcmp(original.c_str(), ptr.c_str(), len)); } // // ptr(const ptr& p, unsigned o, unsigned l) // { const std::string str(len, 'X'); bufferptr original(str.c_str(), len); bufferptr ptr(original, 0, 0); EXPECT_TRUE(ptr.have_raw()); EXPECT_EQ(original.get_raw(), ptr.get_raw()); EXPECT_EQ(2, ptr.raw_nref()); EXPECT_EQ(0, ::memcmp(original.c_str(), ptr.c_str(), len)); PrCtl unset_dumpable; EXPECT_DEATH(bufferptr(original, 0, original.length() + 1), ""); EXPECT_DEATH(bufferptr(bufferptr(), 0, 0), ""); } // // ptr(ptr&& p) // { const std::string str(len, 'X'); bufferptr original(str.c_str(), len); bufferptr ptr(std::move(original)); EXPECT_TRUE(ptr.have_raw()); EXPECT_FALSE(original.have_raw()); EXPECT_EQ(0, ::memcmp(str.c_str(), ptr.c_str(), len)); EXPECT_EQ(1, ptr.raw_nref()); } } TEST(BufferPtr, operator_assign) { // // ptr& operator= (const ptr& p) // bufferptr ptr(10); ptr.copy_in(0, 3, "ABC"); char dest[1]; { bufferptr copy = ptr; copy.copy_out(1, 1, dest); ASSERT_EQ('B', dest[0]); } // // ptr& operator= (ptr&& p) // bufferptr move = std::move(ptr); { move.copy_out(1, 1, dest); ASSERT_EQ('B', dest[0]); } EXPECT_FALSE(ptr.have_raw()); } TEST(BufferPtr, assignment) { unsigned len = 17; // // override a bufferptr set with the same raw // { bufferptr original(len); bufferptr same_raw(original.get_raw()); unsigned offset = 5; unsigned length = len - offset; original.set_offset(offset); original.set_length(length); same_raw = original; ASSERT_EQ(2, original.raw_nref()); ASSERT_EQ(same_raw.get_raw(), original.get_raw()); ASSERT_EQ(same_raw.offset(), original.offset()); ASSERT_EQ(same_raw.length(), original.length()); } // // self assignment is a noop // { bufferptr original(len); original = original; ASSERT_EQ(1, original.raw_nref()); ASSERT_EQ((unsigned)0, original.offset()); ASSERT_EQ(len, original.length()); } // // a copy points to the same raw // { bufferptr original(len); unsigned offset = 5; unsigned length = len - offset; original.set_offset(offset); original.set_length(length); bufferptr ptr; ptr = original; ASSERT_EQ(2, original.raw_nref()); ASSERT_EQ(ptr.get_raw(), original.get_raw()); ASSERT_EQ(original.offset(), ptr.offset()); ASSERT_EQ(original.length(), ptr.length()); } } TEST(BufferPtr, clone) { unsigned len = 17; bufferptr ptr(len); ::memset(ptr.c_str(), 'X', len); bufferptr clone = ptr.clone(); EXPECT_EQ(0, ::memcmp(clone.c_str(), ptr.c_str(), len)); } TEST(BufferPtr, swap) { unsigned len = 17; bufferptr ptr1(len); ::memset(ptr1.c_str(), 'X', len); unsigned ptr1_offset = 4; ptr1.set_offset(ptr1_offset); unsigned ptr1_length = 3; ptr1.set_length(ptr1_length); bufferptr ptr2(len); ::memset(ptr2.c_str(), 'Y', len); unsigned ptr2_offset = 5; ptr2.set_offset(ptr2_offset); unsigned ptr2_length = 7; ptr2.set_length(ptr2_length); ptr1.swap(ptr2); EXPECT_EQ(ptr2_length, ptr1.length()); EXPECT_EQ(ptr2_offset, ptr1.offset()); EXPECT_EQ('Y', ptr1[0]); EXPECT_EQ(ptr1_length, ptr2.length()); EXPECT_EQ(ptr1_offset, ptr2.offset()); EXPECT_EQ('X', ptr2[0]); } TEST(BufferPtr, release) { unsigned len = 17; bufferptr ptr1(len); { bufferptr ptr2(ptr1); EXPECT_EQ(2, ptr1.raw_nref()); } EXPECT_EQ(1, ptr1.raw_nref()); } TEST(BufferPtr, have_raw) { { bufferptr ptr; EXPECT_FALSE(ptr.have_raw()); } { bufferptr ptr(1); EXPECT_TRUE(ptr.have_raw()); } } TEST(BufferPtr, at_buffer_head) { bufferptr ptr(2); EXPECT_TRUE(ptr.at_buffer_head()); ptr.set_offset(1); EXPECT_FALSE(ptr.at_buffer_head()); } TEST(BufferPtr, at_buffer_tail) { bufferptr ptr(2); EXPECT_TRUE(ptr.at_buffer_tail()); ptr.set_length(1); EXPECT_FALSE(ptr.at_buffer_tail()); } TEST(BufferPtr, is_n_page_sized) { { bufferptr ptr(CEPH_PAGE_SIZE); EXPECT_TRUE(ptr.is_n_page_sized()); } { bufferptr ptr(1); EXPECT_FALSE(ptr.is_n_page_sized()); } } TEST(BufferPtr, is_partial) { bufferptr a; EXPECT_FALSE(a.is_partial()); bufferptr b(10); EXPECT_FALSE(b.is_partial()); bufferptr c(b, 1, 9); EXPECT_TRUE(c.is_partial()); bufferptr d(b, 0, 9); EXPECT_TRUE(d.is_partial()); } TEST(BufferPtr, accessors) { unsigned len = 17; bufferptr ptr(len); ptr.c_str()[0] = 'X'; ptr[1] = 'Y'; const bufferptr const_ptr(ptr); EXPECT_NE((void*)NULL, (void*)ptr.get_raw()); EXPECT_EQ('X', ptr.c_str()[0]); { bufferptr ptr; PrCtl unset_dumpable; EXPECT_DEATH(ptr.c_str(), ""); EXPECT_DEATH(ptr[0], ""); } EXPECT_EQ('X', const_ptr.c_str()[0]); { const bufferptr const_ptr; PrCtl unset_dumpable; EXPECT_DEATH(const_ptr.c_str(), ""); EXPECT_DEATH(const_ptr[0], ""); } EXPECT_EQ(len, const_ptr.length()); EXPECT_EQ((unsigned)0, const_ptr.offset()); EXPECT_EQ((unsigned)0, const_ptr.start()); EXPECT_EQ(len, const_ptr.end()); EXPECT_EQ(len, const_ptr.end()); { bufferptr ptr(len); unsigned unused = 1; ptr.set_length(ptr.length() - unused); EXPECT_EQ(unused, ptr.unused_tail_length()); } { bufferptr ptr; EXPECT_EQ((unsigned)0, ptr.unused_tail_length()); } { PrCtl unset_dumpable; EXPECT_DEATH(ptr[len], ""); EXPECT_DEATH(const_ptr[len], ""); } { const bufferptr const_ptr; PrCtl unset_dumpable; EXPECT_DEATH(const_ptr.raw_c_str(), ""); EXPECT_DEATH(const_ptr.raw_length(), ""); EXPECT_DEATH(const_ptr.raw_nref(), ""); } EXPECT_NE((const char *)NULL, const_ptr.raw_c_str()); EXPECT_EQ(len, const_ptr.raw_length()); EXPECT_EQ(2, const_ptr.raw_nref()); { bufferptr ptr(len); unsigned wasted = 1; ptr.set_length(ptr.length() - wasted * 2); ptr.set_offset(wasted); EXPECT_EQ(wasted * 2, ptr.wasted()); } } TEST(BufferPtr, cmp) { bufferptr empty; bufferptr a("A", 1); bufferptr ab("AB", 2); bufferptr af("AF", 2); bufferptr acc("ACC", 3); EXPECT_GE(-1, empty.cmp(a)); EXPECT_LE(1, a.cmp(empty)); EXPECT_GE(-1, a.cmp(ab)); EXPECT_LE(1, ab.cmp(a)); EXPECT_EQ(0, ab.cmp(ab)); EXPECT_GE(-1, ab.cmp(af)); EXPECT_LE(1, af.cmp(ab)); EXPECT_GE(-1, acc.cmp(af)); EXPECT_LE(1, af.cmp(acc)); } TEST(BufferPtr, is_zero) { char str[2] = { '\0', 'X' }; { const bufferptr ptr(buffer::create_static(2, str)); EXPECT_FALSE(ptr.is_zero()); } { const bufferptr ptr(buffer::create_static(1, str)); EXPECT_TRUE(ptr.is_zero()); } } TEST(BufferPtr, copy_out) { { const bufferptr ptr; PrCtl unset_dumpable; EXPECT_DEATH(ptr.copy_out((unsigned)0, (unsigned)0, NULL), ""); } { char in[] = "ABC"; const bufferptr ptr(buffer::create_static(strlen(in), in)); EXPECT_THROW(ptr.copy_out((unsigned)0, strlen(in) + 1, NULL), buffer::end_of_buffer); EXPECT_THROW(ptr.copy_out(strlen(in) + 1, (unsigned)0, NULL), buffer::end_of_buffer); char out[1] = { 'X' }; ptr.copy_out((unsigned)1, (unsigned)1, out); EXPECT_EQ('B', out[0]); } } TEST(BufferPtr, copy_out_bench) { for (int s=1; s<=8; s*=2) { utime_t start = ceph_clock_now(); int buflen = 1048576; int count = 1000; uint64_t v; for (int i=0; i-------> >--------------------> | | // | buffer | +-----+ | | | | // +----------+ ptr | | | | // | _len | list +-----+ | | | | // +----------+ +------+ ,--->+ >-----> | | // | _buffers >----> >----- +-----+ | +-----+ | // +----------+ +----^-+ \ ptr | raw | // | last_p | / `-->+-----+ | +-----+ | // +--------+-+ / + >-----> | | // | ,- ,--->+-----+ | | | | // | / ,--- | | | | // | / ,--- | | | | // +-v--+-^--+--^+-------+ | | | | // | bl | ls | p | p_off >--------------->| | | // +----+----+-----+-----+ | +-----+ | // | | off >------------->| raw | // +---------------+-----+ | | // iterator +---------+ // TEST(BufferListIterator, constructors) { // // iterator() // { buffer::list::iterator i; EXPECT_EQ((unsigned)0, i.get_off()); } // // iterator(list *l, unsigned o=0) // { bufferlist bl; bl.append("ABC", 3); { bufferlist::iterator i(&bl); EXPECT_EQ((unsigned)0, i.get_off()); EXPECT_EQ('A', *i); } { bufferlist::iterator i(&bl, 1); EXPECT_EQ('B', *i); EXPECT_EQ((unsigned)2, i.get_remaining()); } } // // iterator(list *l, unsigned o, std::list::iterator ip, unsigned po) // not tested because of http://tracker.ceph.com/issues/4101 // // iterator(const iterator& other) // { bufferlist bl; bl.append("ABC", 3); bufferlist::iterator i(&bl, 1); bufferlist::iterator j(i); EXPECT_EQ(*i, *j); ++j; EXPECT_NE(*i, *j); EXPECT_EQ('B', *i); EXPECT_EQ('C', *j); bl.c_str()[1] = 'X'; j.advance(-1); EXPECT_EQ('X', *j); } // // const_iterator(const iterator& other) // { bufferlist bl; bl.append("ABC", 3); bufferlist::iterator i(&bl); bufferlist::const_iterator ci(i); EXPECT_EQ(0u, ci.get_off()); EXPECT_EQ('A', *ci); } } TEST(BufferListIterator, empty_create_append_copy) { bufferlist bl, bl2, bl3, out; bl2.append("bar"); bl.swap(bl2); bl2.append("xxx"); bl.append(bl2); bl.rebuild(); bl.copy(0, 6, out); ASSERT_TRUE(out.contents_equal(bl)); } TEST(BufferListIterator, operator_assign) { bufferlist bl; bl.append("ABC", 3); bufferlist::iterator i(&bl, 1); i = i; EXPECT_EQ('B', *i); bufferlist::iterator j; j = i; EXPECT_EQ('B', *j); } TEST(BufferListIterator, get_off) { bufferlist bl; bl.append("ABC", 3); bufferlist::iterator i(&bl, 1); EXPECT_EQ((unsigned)1, i.get_off()); } TEST(BufferListIterator, get_remaining) { bufferlist bl; bl.append("ABC", 3); bufferlist::iterator i(&bl, 1); EXPECT_EQ((unsigned)2, i.get_remaining()); } TEST(BufferListIterator, end) { bufferlist bl; { bufferlist::iterator i(&bl); EXPECT_TRUE(i.end()); } bl.append("ABC", 3); { bufferlist::iterator i(&bl); EXPECT_FALSE(i.end()); } } TEST(BufferListIterator, advance) { bufferlist bl; const std::string one("ABC"); bl.append(bufferptr(one.c_str(), one.size())); const std::string two("DEF"); bl.append(bufferptr(two.c_str(), two.size())); { bufferlist::iterator i(&bl); EXPECT_THROW(i.advance(200), buffer::end_of_buffer); } { bufferlist::iterator i(&bl); EXPECT_THROW(i.advance(-1), buffer::end_of_buffer); } { bufferlist::iterator i(&bl); EXPECT_EQ('A', *i); i.advance(1); EXPECT_EQ('B', *i); i.advance(3); EXPECT_EQ('E', *i); i.advance(-3); EXPECT_EQ('B', *i); i.advance(-1); EXPECT_EQ('A', *i); } } TEST(BufferListIterator, get_ptr_and_advance) { bufferptr a("one", 3); bufferptr b("two", 3); bufferptr c("three", 5); bufferlist bl; bl.append(a); bl.append(b); bl.append(c); const char *ptr; bufferlist::iterator p = bl.begin(); ASSERT_EQ(3u, p.get_ptr_and_advance(11, &ptr)); ASSERT_EQ(bl.length() - 3u, p.get_remaining()); ASSERT_EQ(0, memcmp(ptr, "one", 3)); ASSERT_EQ(2u, p.get_ptr_and_advance(2, &ptr)); ASSERT_EQ(0, memcmp(ptr, "tw", 2)); ASSERT_EQ(1u, p.get_ptr_and_advance(4, &ptr)); ASSERT_EQ(0, memcmp(ptr, "o", 1)); ASSERT_EQ(5u, p.get_ptr_and_advance(5, &ptr)); ASSERT_EQ(0, memcmp(ptr, "three", 5)); ASSERT_EQ(0u, p.get_remaining()); } TEST(BufferListIterator, iterator_crc32c) { bufferlist bl1; bufferlist bl2; bufferlist bl3; string s1(100, 'a'); string s2(50, 'b'); string s3(7, 'c'); string s; bl1.append(s1); bl1.append(s2); bl1.append(s3); s = s1 + s2 + s3; bl2.append(s); bufferlist::iterator it = bl2.begin(); ASSERT_EQ(bl1.crc32c(0), it.crc32c(it.get_remaining(), 0)); ASSERT_EQ(0u, it.get_remaining()); it = bl1.begin(); ASSERT_EQ(bl2.crc32c(0), it.crc32c(it.get_remaining(), 0)); bl3.append(s.substr(98, 55)); it = bl1.begin(); it.advance(98); ASSERT_EQ(bl3.crc32c(0), it.crc32c(55, 0)); ASSERT_EQ(4u, it.get_remaining()); bl3.clear(); bl3.append(s.substr(98 + 55)); it = bl1.begin(); it.advance(98 + 55); ASSERT_EQ(bl3.crc32c(0), it.crc32c(10, 0)); ASSERT_EQ(0u, it.get_remaining()); } TEST(BufferListIterator, seek) { bufferlist bl; bl.append("ABC", 3); bufferlist::iterator i(&bl, 1); EXPECT_EQ('B', *i); i.seek(2); EXPECT_EQ('C', *i); } TEST(BufferListIterator, operator_star) { bufferlist bl; { bufferlist::iterator i(&bl); EXPECT_THROW(*i, buffer::end_of_buffer); } bl.append("ABC", 3); { bufferlist::iterator i(&bl); EXPECT_EQ('A', *i); EXPECT_THROW(i.advance(200), buffer::end_of_buffer); EXPECT_THROW(*i, buffer::end_of_buffer); } } TEST(BufferListIterator, operator_equal) { bufferlist bl; bl.append("ABC", 3); { bufferlist::iterator i(&bl); bufferlist::iterator j(&bl); EXPECT_EQ(i, j); } { bufferlist::const_iterator ci = bl.begin(); bufferlist::iterator i = bl.begin(); EXPECT_EQ(i, ci); EXPECT_EQ(ci, i); } } TEST(BufferListIterator, operator_nequal) { bufferlist bl; bl.append("ABC", 3); { bufferlist::iterator i(&bl); bufferlist::iterator j(&bl); EXPECT_NE(++i, j); } { bufferlist::const_iterator ci = bl.begin(); bufferlist::const_iterator cj = bl.begin(); ++ci; EXPECT_NE(ci, cj); bufferlist::iterator i = bl.begin(); EXPECT_NE(i, ci); EXPECT_NE(ci, i); } { // tests begin(), end(), operator++() also string s("ABC"); int i = 0; for (auto c : bl) { EXPECT_EQ(s[i++], c); } } } TEST(BufferListIterator, operator_plus_plus) { bufferlist bl; { bufferlist::iterator i(&bl); EXPECT_THROW(++i, buffer::end_of_buffer); } bl.append("ABC", 3); { bufferlist::iterator i(&bl); ++i; EXPECT_EQ('B', *i); } } TEST(BufferListIterator, get_current_ptr) { bufferlist bl; { bufferlist::iterator i(&bl); EXPECT_THROW(++i, buffer::end_of_buffer); } bl.append("ABC", 3); { bufferlist::iterator i(&bl, 1); const buffer::ptr ptr = i.get_current_ptr(); EXPECT_EQ('B', ptr[0]); EXPECT_EQ((unsigned)1, ptr.offset()); EXPECT_EQ((unsigned)2, ptr.length()); } } TEST(BufferListIterator, copy) { bufferlist bl; const char *expected = "ABC"; bl.append(expected, 3); // // void copy(unsigned len, char *dest); // { char* copy = (char*)malloc(3); ::memset(copy, 'X', 3); bufferlist::iterator i(&bl); // // demonstrates that it seeks back to offset if p == ls->end() // EXPECT_THROW(i.advance(200), buffer::end_of_buffer); i.copy(2, copy); EXPECT_EQ(0, ::memcmp(copy, expected, 2)); EXPECT_EQ('X', copy[2]); i.seek(0); i.copy(3, copy); EXPECT_EQ(0, ::memcmp(copy, expected, 3)); } // // void buffer::list::iterator::copy_deep(unsigned len, ptr &dest) // { bufferptr ptr; bufferlist::iterator i(&bl); i.copy_deep(2, ptr); EXPECT_EQ((unsigned)2, ptr.length()); EXPECT_EQ('A', ptr[0]); EXPECT_EQ('B', ptr[1]); } // // void buffer::list::iterator::copy_shallow(unsigned len, ptr &dest) // { bufferptr ptr; bufferlist::iterator i(&bl); i.copy_shallow(2, ptr); EXPECT_EQ((unsigned)2, ptr.length()); EXPECT_EQ('A', ptr[0]); EXPECT_EQ('B', ptr[1]); } // // void buffer::list::iterator::copy(unsigned len, list &dest) // { bufferlist copy; bufferlist::iterator i(&bl); // // demonstrates that it seeks back to offset if p == ls->end() // EXPECT_THROW(i.advance(200), buffer::end_of_buffer); i.copy(2, copy); EXPECT_EQ(0, ::memcmp(copy.c_str(), expected, 2)); i.seek(0); i.copy(3, copy); EXPECT_EQ('A', copy[0]); EXPECT_EQ('B', copy[1]); EXPECT_EQ('A', copy[2]); EXPECT_EQ('B', copy[3]); EXPECT_EQ('C', copy[4]); EXPECT_EQ((unsigned)(2 + 3), copy.length()); } // // void buffer::list::iterator::copy_all(list &dest) // { bufferlist copy; bufferlist::iterator i(&bl); // // demonstrates that it seeks back to offset if p == ls->end() // EXPECT_THROW(i.advance(200), buffer::end_of_buffer); i.copy_all(copy); EXPECT_EQ('A', copy[0]); EXPECT_EQ('B', copy[1]); EXPECT_EQ('C', copy[2]); EXPECT_EQ((unsigned)3, copy.length()); } // // void copy(unsigned len, std::string &dest) // { std::string copy; bufferlist::iterator i(&bl); // // demonstrates that it seeks back to offset if p == ls->end() // EXPECT_THROW(i.advance(200), buffer::end_of_buffer); i.copy(2, copy); EXPECT_EQ(0, ::memcmp(copy.c_str(), expected, 2)); i.seek(0); i.copy(3, copy); EXPECT_EQ('A', copy[0]); EXPECT_EQ('B', copy[1]); EXPECT_EQ('A', copy[2]); EXPECT_EQ('B', copy[3]); EXPECT_EQ('C', copy[4]); EXPECT_EQ((unsigned)(2 + 3), copy.length()); } } TEST(BufferListIterator, copy_in) { bufferlist bl; const char *existing = "XXX"; bl.append(existing, 3); // // void buffer::list::iterator::copy_in(unsigned len, const char *src) // { bufferlist::iterator i(&bl); // // demonstrates that it seeks back to offset if p == ls->end() // EXPECT_THROW(i.advance(200), buffer::end_of_buffer); const char *expected = "ABC"; i.copy_in(3, expected); EXPECT_EQ(0, ::memcmp(bl.c_str(), expected, 3)); EXPECT_EQ('A', bl[0]); EXPECT_EQ('B', bl[1]); EXPECT_EQ('C', bl[2]); EXPECT_EQ((unsigned)3, bl.length()); } // // void buffer::list::iterator::copy_in(unsigned len, const list& otherl) // { bufferlist::iterator i(&bl); // // demonstrates that it seeks back to offset if p == ls->end() // EXPECT_THROW(i.advance(200), buffer::end_of_buffer); bufferlist expected; expected.append("ABC", 3); i.copy_in(3, expected); EXPECT_EQ(0, ::memcmp(bl.c_str(), expected.c_str(), 3)); EXPECT_EQ('A', bl[0]); EXPECT_EQ('B', bl[1]); EXPECT_EQ('C', bl[2]); EXPECT_EQ((unsigned)3, bl.length()); } } // iterator& buffer::list::const_iterator::operator++() TEST(BufferListConstIterator, operator_plus_plus) { bufferlist bl; { bufferlist::const_iterator i(&bl); EXPECT_THROW(++i, buffer::end_of_buffer); } bl.append("ABC", 3); { const bufferlist const_bl(bl); bufferlist::const_iterator i(const_bl.begin()); ++i; EXPECT_EQ('B', *i); } } TEST(BufferList, constructors) { // // list() // { bufferlist bl; ASSERT_EQ((unsigned)0, bl.length()); } // // list(unsigned prealloc) // { bufferlist bl(1); ASSERT_EQ((unsigned)0, bl.length()); bl.append('A'); ASSERT_EQ('A', bl[0]); } // // list(const list& other) // { bufferlist bl(1); bl.append('A'); ASSERT_EQ('A', bl[0]); bufferlist copy(bl); ASSERT_EQ('A', copy[0]); } // // list(list&& other) // { bufferlist bl(1); bl.append('A'); bufferlist copy = std::move(bl); ASSERT_EQ(0U, bl.length()); ASSERT_EQ(1U, copy.length()); ASSERT_EQ('A', copy[0]); } } void bench_bufferlist_alloc(int size, int num, int per) { utime_t start = ceph_clock_now(); for (int i=0; i %s ; chmod 0 %s", FILENAME, FILENAME); EXPECT_EQ(0, ::system(cmd)); if (getuid() != 0) { EXPECT_EQ(-EACCES, bl.read_file(FILENAME, &error)); } snprintf(cmd, sizeof(cmd), "chmod +r %s", FILENAME); EXPECT_EQ(0, ::system(cmd)); EXPECT_EQ(0, bl.read_file(FILENAME, &error)); ::unlink(FILENAME); EXPECT_EQ((unsigned)4, bl.length()); std::string actual(bl.c_str(), bl.length()); EXPECT_EQ("ABC\n", actual); } TEST(BufferList, read_fd) { unsigned len = 4; ::unlink(FILENAME); snprintf(cmd, sizeof(cmd), "echo ABC > %s", FILENAME); EXPECT_EQ(0, ::system(cmd)); int fd = -1; bufferlist bl; EXPECT_EQ(-EBADF, bl.read_fd(fd, len)); fd = ::open(FILENAME, O_RDONLY); EXPECT_EQ(len, (unsigned)bl.read_fd(fd, len)); //EXPECT_EQ(CEPH_BUFFER_APPEND_SIZE - len, bl.front().unused_tail_length()); EXPECT_EQ(len, bl.length()); ::close(fd); ::unlink(FILENAME); } TEST(BufferList, write_file) { ::unlink(FILENAME); int mode = 0600; bufferlist bl; EXPECT_EQ(-ENOENT, bl.write_file("un/like/ly", mode)); bl.append("ABC"); EXPECT_EQ(0, bl.write_file(FILENAME, mode)); struct stat st; memset(&st, 0, sizeof(st)); ::stat(FILENAME, &st); EXPECT_EQ((unsigned)(mode | S_IFREG), st.st_mode); ::unlink(FILENAME); } TEST(BufferList, write_fd) { ::unlink(FILENAME); int fd = ::open(FILENAME, O_WRONLY|O_CREAT|O_TRUNC, 0600); bufferlist bl; for (unsigned i = 0; i < IOV_MAX * 2; i++) { bufferptr ptr("A", 1); bl.push_back(ptr); } EXPECT_EQ(0, bl.write_fd(fd)); ::close(fd); struct stat st; memset(&st, 0, sizeof(st)); ::stat(FILENAME, &st); EXPECT_EQ(IOV_MAX * 2, st.st_size); ::unlink(FILENAME); } TEST(BufferList, write_fd_offset) { ::unlink(FILENAME); int fd = ::open(FILENAME, O_WRONLY|O_CREAT|O_TRUNC, 0600); bufferlist bl; for (unsigned i = 0; i < IOV_MAX * 2; i++) { bufferptr ptr("A", 1); bl.push_back(ptr); } uint64_t offset = 200; EXPECT_EQ(0, bl.write_fd(fd, offset)); ::close(fd); struct stat st; memset(&st, 0, sizeof(st)); ::stat(FILENAME, &st); EXPECT_EQ(IOV_MAX * 2 + offset, (unsigned)st.st_size); ::unlink(FILENAME); } TEST(BufferList, crc32c) { bufferlist bl; __u32 crc = 0; bl.append("A"); crc = bl.crc32c(crc); EXPECT_EQ((unsigned)0xB3109EBF, crc); crc = bl.crc32c(crc); EXPECT_EQ((unsigned)0x5FA5C0CC, crc); } TEST(BufferList, crc32c_append) { bufferlist bl1; bufferlist bl2; for (int j = 0; j < 200; ++j) { bufferlist bl; for (int i = 0; i < 200; ++i) { char x = rand(); bl.append(x); bl1.append(x); } bl.crc32c(rand()); // mess with the cached bufferptr crc values bl2.append(bl); } ASSERT_EQ(bl1.crc32c(0), bl2.crc32c(0)); } TEST(BufferList, crc32c_zeros) { char buffer[4*1024]; for (size_t i=0; i < sizeof(buffer); i++) { buffer[i] = i; } bufferlist bla; bufferlist blb; for (size_t j=0; j < 1000; j++) { bufferptr a(buffer, sizeof(buffer)); bla.push_back(a); uint32_t crca = bla.crc32c(111); blb.push_back(a); uint32_t crcb = ceph_crc32c(111, (unsigned char*)blb.c_str(), blb.length()); EXPECT_EQ(crca, crcb); } } TEST(BufferList, crc32c_append_perf) { int len = 256 * 1024 * 1024; bufferptr a(len); bufferptr b(len); bufferptr c(len); bufferptr d(len); std::cout << "populating large buffers (a, b=c=d)" << std::endl; char *pa = a.c_str(); char *pb = b.c_str(); char *pc = c.c_str(); char *pd = c.c_str(); for (int i=0; i(bufferlist& l, bufferlist& r) // ASSERT_FALSE(a > ab); ASSERT_TRUE(ab > a); ASSERT_TRUE(ac > ab); ASSERT_FALSE(ab > ac); ASSERT_FALSE(ab > ab); // // bool operator>=(bufferlist& l, bufferlist& r) // ASSERT_FALSE(a >= ab); ASSERT_TRUE(ab >= a); ASSERT_TRUE(ac >= ab); ASSERT_FALSE(ab >= ac); ASSERT_TRUE(ab >= ab); // // bool operator<(bufferlist& l, bufferlist& r) // ASSERT_TRUE(a < ab); ASSERT_FALSE(ab < a); ASSERT_FALSE(ac < ab); ASSERT_TRUE(ab < ac); ASSERT_FALSE(ab < ab); // // bool operator<=(bufferlist& l, bufferlist& r) // ASSERT_TRUE(a <= ab); ASSERT_FALSE(ab <= a); ASSERT_FALSE(ac <= ab); ASSERT_TRUE(ab <= ac); ASSERT_TRUE(ab <= ab); // // bool operator==(bufferlist &l, bufferlist &r) // ASSERT_FALSE(a == ab); ASSERT_FALSE(ac == ab); ASSERT_TRUE(ab == ab); } TEST(BufferList, ostream) { std::ostringstream stream; bufferlist bl; const char *s[] = { "ABC", "DEF" }; for (unsigned i = 0; i < 2; i++) { bufferptr ptr(s[i], strlen(s[i])); bl.push_back(ptr); } stream << bl; std::cerr << stream.str() << std::endl; EXPECT_GT(stream.str().size(), stream.str().find("list(len=6,")); EXPECT_GT(stream.str().size(), stream.str().find("len 3 nref 1),\n")); EXPECT_GT(stream.str().size(), stream.str().find("len 3 nref 1)\n")); } TEST(BufferList, zero) { // // void zero() // { bufferlist bl; bl.append('A'); EXPECT_EQ('A', bl[0]); bl.zero(); EXPECT_EQ('\0', bl[0]); } // // void zero(unsigned o, unsigned l) // const char *s[] = { "ABC", "DEF", "GHI", "KLM" }; { bufferlist bl; bufferptr ptr(s[0], strlen(s[0])); bl.push_back(ptr); bl.zero((unsigned)0, (unsigned)1); EXPECT_EQ(0, ::memcmp("\0BC", bl.c_str(), 3)); } { bufferlist bl; for (unsigned i = 0; i < 4; i++) { bufferptr ptr(s[i], strlen(s[i])); bl.push_back(ptr); } { PrCtl unset_dumpable; EXPECT_DEATH(bl.zero((unsigned)0, (unsigned)2000), ""); } bl.zero((unsigned)2, (unsigned)5); EXPECT_EQ(0, ::memcmp("AB\0\0\0\0\0HIKLM", bl.c_str(), 9)); } { bufferlist bl; for (unsigned i = 0; i < 4; i++) { bufferptr ptr(s[i], strlen(s[i])); bl.push_back(ptr); } bl.zero((unsigned)3, (unsigned)3); EXPECT_EQ(0, ::memcmp("ABC\0\0\0GHIKLM", bl.c_str(), 9)); } { bufferlist bl; bufferptr ptr1(4); bufferptr ptr2(4); memset(ptr1.c_str(), 'a', 4); memset(ptr2.c_str(), 'b', 4); bl.append(ptr1); bl.append(ptr2); bl.zero((unsigned)2, (unsigned)4); EXPECT_EQ(0, ::memcmp("aa\0\0\0\0bb", bl.c_str(), 8)); } } TEST(BufferList, EmptyAppend) { bufferlist bl; bufferptr ptr; bl.push_back(ptr); ASSERT_EQ(bl.begin().end(), 1); } TEST(BufferList, TestPtrAppend) { bufferlist bl; char correct[MAX_TEST]; int curpos = 0; int length = random() % 5 > 0 ? random() % 1000 : 0; while (curpos + length < MAX_TEST) { if (!length) { bufferptr ptr; bl.push_back(ptr); } else { char *current = correct + curpos; for (int i = 0; i < length; ++i) { char next = random() % 255; correct[curpos++] = next; } bufferptr ptr(current, length); bl.append(ptr); } length = random() % 5 > 0 ? random() % 1000 : 0; } ASSERT_EQ(memcmp(bl.c_str(), correct, curpos), 0); } TEST(BufferList, TestDirectAppend) { bufferlist bl; char correct[MAX_TEST]; int curpos = 0; int length = random() % 5 > 0 ? random() % 1000 : 0; while (curpos + length < MAX_TEST) { char *current = correct + curpos; for (int i = 0; i < length; ++i) { char next = random() % 255; correct[curpos++] = next; } bl.append(current, length); length = random() % 5 > 0 ? random() % 1000 : 0; } ASSERT_EQ(memcmp(bl.c_str(), correct, curpos), 0); } TEST(BufferList, TestCloneNonShareable) { bufferlist bl; std::string str = "sharetest"; bl.append(str.c_str(), 9); bufferlist bl_share; bl_share.share(bl); bufferlist bl_noshare; buffer::ptr unraw = buffer::create_unshareable(10); unraw.copy_in(0, 9, str.c_str()); bl_noshare.append(unraw); bufferlist bl_copied_share = bl_share; bufferlist bl_copied_noshare = bl_noshare; // assert shared bufferlist has same buffers bufferlist::iterator iter_bl = bl.begin(); bufferlist::iterator iter_bl_share = bl_share.begin(); // ok, this considers ptr::off, but it's still a true assertion (and below) ASSERT_TRUE(iter_bl.get_current_ptr().c_str() == iter_bl_share.get_current_ptr().c_str()); // assert copy of shareable bufferlist has same buffers iter_bl = bl.begin(); bufferlist::iterator iter_bl_copied_share = bl_copied_share.begin(); ASSERT_TRUE(iter_bl.get_current_ptr().c_str() == iter_bl_copied_share.get_current_ptr().c_str()); // assert copy of non-shareable bufferlist has different buffers bufferlist::iterator iter_bl_copied_noshare = bl_copied_noshare.begin(); ASSERT_FALSE(iter_bl.get_current_ptr().c_str() == iter_bl_copied_noshare.get_current_ptr().c_str()); // assert that claim with CLAIM_ALLOW_NONSHAREABLE overrides safe-sharing bufferlist bl_claim_noshare_override; void* addr = bl_noshare.begin().get_current_ptr().c_str(); bl_claim_noshare_override.claim(bl_noshare, buffer::list::CLAIM_ALLOW_NONSHAREABLE); bufferlist::iterator iter_bl_noshare_override = bl_claim_noshare_override.begin(); ASSERT_TRUE(addr /* the original first segment of bl_noshare() */ == iter_bl_noshare_override.get_current_ptr().c_str()); } TEST(BufferList, TestCopyAll) { const static size_t BIG_SZ = 10737414; ceph::shared_ptr big( (unsigned char*)malloc(BIG_SZ), free); unsigned char c = 0; for (size_t i = 0; i < BIG_SZ; ++i) { big.get()[i] = c++; } bufferlist bl; bl.append((const char*)big.get(), BIG_SZ); bufferlist::iterator i = bl.begin(); bufferlist bl2; i.copy_all(bl2); ASSERT_EQ(bl2.length(), BIG_SZ); ceph::shared_ptr big2( (unsigned char*)malloc(BIG_SZ), free); bl2.copy(0, BIG_SZ, (char*)big2.get()); ASSERT_EQ(memcmp(big.get(), big2.get(), BIG_SZ), 0); } TEST(BufferList, InvalidateCrc) { const static size_t buffer_size = 262144; ceph::shared_ptr big( (unsigned char*)malloc(buffer_size), free); unsigned char c = 0; char* ptr = (char*) big.get(); char* inptr; for (size_t i = 0; i < buffer_size; ++i) { ptr[i] = c++; } bufferlist bl; // test for crashes (shouldn't crash) bl.invalidate_crc(); // put data into bufferlist bl.append((const char*)big.get(), buffer_size); // get its crc __u32 crc = bl.crc32c(0); // modify data in bl without its knowledge inptr = (char*) bl.c_str(); c = 0; for (size_t i = 0; i < buffer_size; ++i) { inptr[i] = c--; } // make sure data in bl are now different than in big EXPECT_NE(memcmp((void*) ptr, (void*) inptr, buffer_size), 0); // crc should remain the same __u32 new_crc = bl.crc32c(0); EXPECT_EQ(crc, new_crc); // force crc invalidate, check if it is updated bl.invalidate_crc(); EXPECT_NE(crc, bl.crc32c(0)); } TEST(BufferList, TestIsProvidedBuffer) { char buff[100]; bufferlist bl; bl.push_back(buffer::create_static(100, buff)); ASSERT_TRUE(bl.is_provided_buffer(buff)); bl.append_zero(100); ASSERT_FALSE(bl.is_provided_buffer(buff)); } TEST(BufferHash, all) { { bufferlist bl; bl.append("A"); bufferhash hash; EXPECT_EQ((unsigned)0, hash.digest()); hash.update(bl); EXPECT_EQ((unsigned)0xB3109EBF, hash.digest()); hash.update(bl); EXPECT_EQ((unsigned)0x5FA5C0CC, hash.digest()); } { bufferlist bl; bl.append("A"); bufferhash hash; EXPECT_EQ((unsigned)0, hash.digest()); bufferhash& returned_hash = hash << bl; EXPECT_EQ(&returned_hash, &hash); EXPECT_EQ((unsigned)0xB3109EBF, hash.digest()); } } /* * Local Variables: * compile-command: "cd .. ; make unittest_bufferlist && * ulimit -s unlimited ; CEPH_BUFFER_TRACK=true valgrind \ * --max-stackframe=20000000 --tool=memcheck \ * ./unittest_bufferlist # --gtest_filter=BufferList.constructors" * End: */