1/*- 2 * Copyright (c) 2002 Tim J. Robbins 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27/* 28 * Test program for wcsrtombs(), as specified by IEEE Std. 1003.1-2001 and 29 * ISO/IEC 9899:1999. 30 * 31 * The function is tested with both the "C" ("POSIX") LC_CTYPE setting and 32 * "ja_JP.eucJP". Other encodings are not tested. 33 */ 34 35#include <sys/cdefs.h> 36__FBSDID("$FreeBSD$"); 37 38#include <assert.h> 39#include <errno.h> 40#include <limits.h> 41#include <locale.h> 42#include <stdio.h> 43#include <stdlib.h> 44#include <string.h> 45#include <wchar.h> 46 47int 48main(int argc, char *argv[]) 49{ 50 wchar_t srcbuf[128]; 51 char dstbuf[128]; 52 wchar_t *src; 53 mbstate_t s; 54 55 /* 56 * C/POSIX locale. 57 */ 58 59 printf("1..1\n"); 60 61 /* Simple null terminated string. */ 62 wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf)); 63 wcscpy(srcbuf, L"hello"); 64 memset(dstbuf, 0xcc, sizeof(dstbuf)); 65 src = srcbuf; 66 memset(&s, 0, sizeof(s)); 67 assert(wcsrtombs(dstbuf, (const wchar_t **)&src, sizeof(dstbuf), 68 &s) == 5); 69 assert(strcmp(dstbuf, "hello") == 0); 70 assert((unsigned char)dstbuf[6] == 0xcc); 71 assert(src == NULL); 72 73 /* Not enough space in destination buffer. */ 74 wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf)); 75 wcscpy(srcbuf, L"hello"); 76 memset(dstbuf, 0xcc, sizeof(dstbuf)); 77 src = srcbuf; 78 memset(&s, 0, sizeof(s)); 79 assert(wcsrtombs(dstbuf, (const wchar_t **)&src, 4, 80 &s) == 4); 81 assert(memcmp(dstbuf, "hell", 4) == 0); 82 assert((unsigned char)dstbuf[5] == 0xcc); 83 assert(src == srcbuf + 4); 84 85 /* Null terminated string, internal dest. buffer */ 86 wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf)); 87 wcscpy(srcbuf, L"hello"); 88 src = srcbuf; 89 memset(&s, 0, sizeof(s)); 90 assert(wcsrtombs(NULL, (const wchar_t **)&src, sizeof(dstbuf), 91 &s) == 5); 92 93 /* Null terminated string, internal state. */ 94 wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf)); 95 wcscpy(srcbuf, L"hello"); 96 memset(dstbuf, 0xcc, sizeof(dstbuf)); 97 src = srcbuf; 98 assert(wcsrtombs(dstbuf, (const wchar_t **)&src, sizeof(dstbuf), 99 NULL) == 5); 100 assert(strcmp(dstbuf, "hello") == 0); 101 assert((unsigned char)dstbuf[6] == 0xcc); 102 assert(src == NULL); 103 104 /* Null terminated string, internal state, internal dest. buffer. */ 105 wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf)); 106 wcscpy(srcbuf, L"hello"); 107 src = srcbuf; 108 assert(wcsrtombs(NULL, (const wchar_t **)&src, 0, NULL) == 5); 109 110 /* Empty source buffer. */ 111 wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf)); 112 srcbuf[0] = L'\0'; 113 memset(dstbuf, 0xcc, sizeof(dstbuf)); 114 src = srcbuf; 115 memset(&s, 0, sizeof(s)); 116 assert(wcsrtombs(dstbuf, (const wchar_t **)&src, sizeof(dstbuf), 117 &s) == 0); 118 assert(dstbuf[0] == L'\0'); 119 120 /* Zero length destination buffer. */ 121 wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf)); 122 wcscpy(srcbuf, L"hello"); 123 memset(dstbuf, 0xcc, sizeof(dstbuf)); 124 src = srcbuf; 125 memset(&s, 0, sizeof(s)); 126 assert(wcsrtombs(dstbuf, (const wchar_t **)&src, 0, &s) == 0); 127 assert((unsigned char)dstbuf[0] == 0xcc); 128 129 /* 130 * Japanese (EUC) locale. 131 */ 132 133 assert(strcmp(setlocale(LC_CTYPE, "ja_JP.eucJP"), "ja_JP.eucJP") == 0); 134 assert(MB_CUR_MAX > 1); 135 136 wmemset(srcbuf, 0xcc, sizeof(srcbuf) / sizeof(*srcbuf)); 137 srcbuf[0] = 0xA3C1; 138 srcbuf[1] = 0x0020; 139 srcbuf[2] = 0x0042; 140 srcbuf[3] = 0x0020; 141 srcbuf[4] = 0xA3C3; 142 srcbuf[5] = 0x0000; 143 memset(dstbuf, 0xcc, sizeof(dstbuf)); 144 src = srcbuf; 145 memset(&s, 0, sizeof(s)); 146 assert(wcsrtombs(dstbuf, (const wchar_t **)&src, sizeof(dstbuf), 147 &s) == 7); 148 assert(strcmp(dstbuf, "\xA3\xC1 B \xA3\xC3") == 0); 149 assert((unsigned char)dstbuf[8] == 0xcc); 150 assert(src == NULL); 151 152 printf("ok 1 - wcsrtombs()\n"); 153 154 return (0); 155} 156