[7/9] msvcirt: Implement stdiobuf::setrwbuf
Iván Matellanes
matellanesivan at gmail.com
Fri Oct 2 08:03:25 CDT 2015
Signed-off-by: Iván Matellanes <matellanesivan at gmail.com>
---
dlls/msvcirt/msvcirt.c | 28 ++++++++++++++++++++++++++--
dlls/msvcirt/tests/msvcirt.c | 32 ++++++++++++++++++++++++++++++++
2 files changed, 58 insertions(+), 2 deletions(-)
diff --git a/dlls/msvcirt/msvcirt.c b/dlls/msvcirt/msvcirt.c
index 1d915fa..6bcd070 100644
--- a/dlls/msvcirt/msvcirt.c
+++ b/dlls/msvcirt/msvcirt.c
@@ -1580,8 +1580,32 @@ streampos __thiscall stdiobuf_seekoff(stdiobuf *this, streamoff offset, ios_seek
DEFINE_THISCALL_WRAPPER(stdiobuf_setrwbuf, 12)
int __thiscall stdiobuf_setrwbuf(stdiobuf *this, int read_size, int write_size)
{
- FIXME("(%p %d %d) stub\n", this, read_size, write_size);
- return EOF;
+ char *reserve;
+ int buffer_size = read_size + write_size;
+
+ TRACE("(%p %d %d)\n", this, read_size, write_size);
+ if (buffer_size < 0)
+ return 0;
+ if (read_size == 0 && write_size == 0) {
+ this->base.unbuffered = 1;
+ return 0;
+ }
+ /* get a new buffer */
+ reserve = MSVCRT_operator_new(buffer_size);
+ if (!reserve)
+ return 0;
+ streambuf_setb(&this->base, reserve, reserve + buffer_size, 1);
+ this->base.unbuffered = 0;
+ /* set the get/put areas */
+ if (read_size > 0)
+ streambuf_setg(&this->base, reserve, reserve + read_size, reserve + read_size);
+ else
+ streambuf_setg(&this->base, NULL, NULL, NULL);
+ if (write_size > 0)
+ streambuf_setp(&this->base, reserve + read_size, reserve + buffer_size);
+ else
+ streambuf_setp(&this->base, NULL, NULL);
+ return 1;
}
/* ?stdiofile at stdiobuf@@QAEPAU_iobuf@@XZ */
diff --git a/dlls/msvcirt/tests/msvcirt.c b/dlls/msvcirt/tests/msvcirt.c
index 8bc45c1..758062c 100644
--- a/dlls/msvcirt/tests/msvcirt.c
+++ b/dlls/msvcirt/tests/msvcirt.c
@@ -210,6 +210,7 @@ static int (*__thiscall p_strstreambuf_underflow)(strstreambuf*);
static stdiobuf* (*__thiscall p_stdiobuf_file_ctor)(stdiobuf*, FILE*);
static void (*__thiscall p_stdiobuf_dtor)(stdiobuf*);
static int (*__thiscall p_stdiobuf_overflow)(stdiobuf*, int);
+static int (*__thiscall p_stdiobuf_setrwbuf)(stdiobuf*, int, int);
static int (*__thiscall p_stdiobuf_sync)(stdiobuf*);
static int (*__thiscall p_stdiobuf_underflow)(stdiobuf*);
@@ -374,6 +375,7 @@ static BOOL init(void)
SET(p_stdiobuf_file_ctor, "??0stdiobuf@@QEAA at PEAU_iobuf@@@Z");
SET(p_stdiobuf_dtor, "??1stdiobuf@@UEAA at XZ");
SET(p_stdiobuf_overflow, "?overflow at stdiobuf@@UEAAHH at Z");
+ SET(p_stdiobuf_setrwbuf, "?setrwbuf at stdiobuf@@QEAAHHH at Z");
SET(p_stdiobuf_sync, "?sync at stdiobuf@@UEAAHXZ");
SET(p_stdiobuf_underflow, "?underflow at stdiobuf@@UEAAHXZ");
@@ -458,6 +460,7 @@ static BOOL init(void)
SET(p_stdiobuf_file_ctor, "??0stdiobuf@@QAE at PAU_iobuf@@@Z");
SET(p_stdiobuf_dtor, "??1stdiobuf@@UAE at XZ");
SET(p_stdiobuf_overflow, "?overflow at stdiobuf@@UAEHH at Z");
+ SET(p_stdiobuf_setrwbuf, "?setrwbuf at stdiobuf@@QAEHHH at Z");
SET(p_stdiobuf_sync, "?sync at stdiobuf@@UAEHXZ");
SET(p_stdiobuf_underflow, "?underflow at stdiobuf@@UAEHXZ");
@@ -2016,6 +2019,35 @@ static void test_stdiobuf(void)
ok(stb2.base.gptr == stb2.base.egptr, "wrong get pointer, expected %p got %p\n", stb2.base.egptr, stb2.base.gptr);
stb2.base.unbuffered = 1;
+ /* setrwbuf */
+ ret = (int) call_func3(p_stdiobuf_setrwbuf, &stb1, 100, 60);
+ ok(ret == 1, "expected 1 got %d\n", ret);
+ ok(stb1.base.allocated == 1, "wrong allocate value, expected 1 got %d\n", stb1.base.allocated);
+ ok(stb1.base.unbuffered == 0, "wrong unbuffered value, expected 0 got %d\n", stb1.base.unbuffered);
+ ok(stb1.base.ebuf == stb1.base.base + 160, "expected %p got %p\n", stb1.base.base + 160, stb1.base.ebuf);
+ ok(stb1.base.eback == stb1.base.base, "wrong get base, expected %p got %p\n", stb1.base.base, stb1.base.eback);
+ ok(stb1.base.gptr == stb1.base.base + 100, "wrong get pointer, expected %p got %p\n", stb1.base.base + 100, stb1.base.gptr);
+ ok(stb1.base.egptr == stb1.base.base + 100, "wrong get end, expected %p got %p\n", stb1.base.base + 100, stb1.base.egptr);
+ ok(stb1.base.pbase == stb1.base.base + 100, "wrong put base, expected %p got %p\n", stb1.base.base + 100, stb1.base.pbase);
+ ok(stb1.base.pptr == stb1.base.base + 100, "wrong put pointer, expected %p got %p\n", stb1.base.base + 100, stb1.base.pptr);
+ ok(stb1.base.epptr == stb1.base.base + 160, "wrong put end, expected %p got %p\n", stb1.base.base + 160, stb1.base.epptr);
+ ret = (int) call_func3(p_stdiobuf_setrwbuf, &stb1, -1, 64);
+ ok(ret == 0 || ret == 1, "expected 0 or 1, got %d\n", ret);
+ ret = (int) call_func3(p_stdiobuf_setrwbuf, &stb1, 32, -8);
+ ok(ret == 0 || ret == 1, "expected 0 or 1, got %d\n", ret);
+ ret = (int) call_func3(p_stdiobuf_setrwbuf, &stb1, 0, 64);
+ ok(ret == 1, "expected 1 got %d\n", ret);
+ ok(stb1.base.ebuf == stb1.base.base + 64, "expected %p got %p\n", stb1.base.base + 64, stb1.base.ebuf);
+ ok(stb1.base.eback == NULL, "wrong get base, expected %p got %p\n", NULL, stb1.base.eback);
+ ok(stb1.base.gptr == NULL, "wrong get pointer, expected %p got %p\n", NULL, stb1.base.gptr);
+ ok(stb1.base.egptr == NULL, "wrong get end, expected %p got %p\n", NULL, stb1.base.egptr);
+ ok(stb1.base.pbase == stb1.base.base, "wrong put base, expected %p got %p\n", stb1.base.base, stb1.base.pbase);
+ ok(stb1.base.pptr == stb1.base.base, "wrong put pointer, expected %p got %p\n", stb1.base.base, stb1.base.pptr);
+ ok(stb1.base.epptr == stb1.base.base + 64, "wrong put end, expected %p got %p\n", stb1.base.base + 64, stb1.base.epptr);
+ ret = (int) call_func3(p_stdiobuf_setrwbuf, &stb1, 0, 0);
+ ok(ret == 0, "expected 0 got %d\n", ret);
+ ok(stb1.base.unbuffered == 1, "wrong unbuffered value, expected 1 got %d\n", stb1.base.unbuffered);
+
call_func1(p_stdiobuf_dtor, &stb1);
call_func1(p_stdiobuf_dtor, &stb2);
fclose(file1);
--
2.1.4
More information about the wine-patches
mailing list