safembyt.c

来自「开放源码的编译器open watcom 1.6.0版的源代码」· C语言 代码 · 共 614 行 · 第 1/2 页

/****************************************************************************
*
*                            Open Watcom Project
*
*    Portions Copyright (c) 1983-2002 Sybase, Inc. All Rights Reserved.
*
*  ========================================================================
*
*    This file contains Original Code and/or Modifications of Original
*    Code as defined in and that are subject to the Sybase Open Watcom
*    Public License version 1.0 (the 'License'). You may not use this file
*    except in compliance with the License. BY USING THIS FILE YOU AGREE TO
*    ALL TERMS AND CONDITIONS OF THE LICENSE. A copy of the License is
*    provided with the Original Code and Modifications, and is also
*    available at www.sybase.com/developer/opensource.
*
*    The Original Code and all software distributed under the License are
*    distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
*    EXPRESS OR IMPLIED, AND SYBASE AND ALL CONTRIBUTORS HEREBY DISCLAIM
*    ALL SUCH WARRANTIES, INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF
*    MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR
*    NON-INFRINGEMENT. Please see the License for the specific language
*    governing rights and limitations under the License.
*
*  ========================================================================
*
* Description:  Non-exhaustive test of the Safer C library
*               multibyte character functions.
*
*               modelled after mbtest.c
****************************************************************************/


#define __STDC_WANT_LIB_EXT1__ 1       // Enable Safer C interfaces

#include <errno.h>
#include <locale.h>
#include <stdarg.h>
#include <mbstring.h>
#include <stdio.h>
#include <stddef.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <wchar.h>
#ifdef __SW_BW
    #include <wdefwin.h>
#endif

/* DUMMY_DBCS won't be needed now that Windows and DOS are fully supported */
//#if !defined(__NT__) && !defined(__OS2__) && !defined(__WINDOWS__)
//    #define DUMMY_DBCS
//#endif


#define VERIFY( exp )   if( !(exp) ) {                                      \
                            printf( "%s: ***FAILURE*** at line %d of %s.\n",\
                                    ProgramName, __LINE__,                  \
                                    _mbslwr(__FILE__) );                    \
                            NumErrors++;                                    \
                        }

char ProgramName[128];                          /* executable filename */
int NumErrors = 0;                              /* number of errors */

int     NumViolations = 0;  /* runtime-constraint violation counter */


/* Runtime-constraint handler for tests; doesn't abort program. */
void my_constraint_handler( const char *msg, void *ptr, errno_t error )
{
    fprintf( stderr, "Runtime-constraint in %s", msg );
    ++NumViolations;
}



/****
***** Test mbstowcs_s() and mbsrtowcs_s().
****/

void TestToWide( void )
{
    wchar_t             wcs[20];
    errno_t             rc;
    size_t              retval;
    size_t              retval2;
    mbstate_t           ps;
    const char         *pchar;
    const char         *pchar2;
    int                 violations = NumViolations;



    rc = mbstowcs_s( &retval, wcs, 20, "", 0 );
    VERIFY( rc == 0 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( violations == NumViolations );

    rc = mbstowcs_s( &retval, wcs, 20, "foo", 3 );
    VERIFY( rc == 0 );
    VERIFY( retval == 3 );
    VERIFY( violations == NumViolations );

    rc = mbstowcs_s( &retval, wcs, 20, "\x81\xFC""foo", 0 );
    VERIFY( rc == 0 );
    VERIFY( retval == 0 );
    VERIFY( violations == NumViolations );

    rc = mbstowcs_s( &retval, wcs, 20, "foo", 10 );
    VERIFY( rc == 0 );
    VERIFY( retval == 3 );
    VERIFY( violations == NumViolations );

    rc = mbstowcs_s( &retval, wcs, 6, "\x81\xFC""foo", 10 );
    VERIFY( rc == 0 );
    VERIFY( retval == 4 );
    VERIFY( wcs[1] == 'f' );
    VERIFY( wcs[2] == 'o' );
    VERIFY( wcs[3] == 'o' );
    VERIFY( wcs[4] == '\0' );
    VERIFY( violations == NumViolations );
    rc = mbstowcs_s( &retval2, NULL, 0, "\x81\xFC""foo", 10 );
    VERIFY( retval == retval2 );
    VERIFY( violations == NumViolations );

    pchar = "";
    rc = mbsrtowcs_s( &retval, wcs, 20, &pchar, 0, &ps );
    VERIFY( rc == 0 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( violations == NumViolations );

    pchar = "foo";
    rc = mbsrtowcs_s( &retval, wcs, 20, &pchar, 3, &ps );
    VERIFY( rc == 0 );
    VERIFY( retval == 3 );
    VERIFY( violations == NumViolations );

    pchar = "\x81\xFC""foo";
    rc = mbsrtowcs_s( &retval, wcs, 20, &pchar, 0, &ps );
    VERIFY( rc == 0 );
    VERIFY( retval == 0 );
    VERIFY( violations == NumViolations );

    pchar = "foo";
    rc = mbsrtowcs_s( &retval, wcs, 20, &pchar, 10, &ps );
    VERIFY( rc == 0 );
    VERIFY( retval == 3 );
    VERIFY( violations == NumViolations );

    pchar = "\x81\xFC""foo";
    rc = mbsrtowcs_s( &retval, wcs, 6, &pchar, 10, &ps );
    VERIFY( rc == 0 );
    VERIFY( retval == 4 );
    VERIFY( wcs[1] == 'f' );
    VERIFY( wcs[2] == 'o' );
    VERIFY( wcs[3] == 'o' );
    VERIFY( wcs[4] == '\0' );
    VERIFY( violations == NumViolations );
    pchar2 = "\x81\xFC""foo";
    rc = mbsrtowcs_s( &retval2, NULL, 0, &pchar2, 10, &ps );
    VERIFY( retval == retval2 );
    VERIFY( violations == NumViolations );

    /***********************************************************************/
    /*  now test runtime-constraints                                       */
    /***********************************************************************/
    rc = mbstowcs_s( NULL, wcs, 20, "\x81\xFC""foo", 10 );
    VERIFY( rc != 0 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );

    rc = mbstowcs_s( &retval, wcs, 6, NULL, 10 );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );

#if RSIZE_MAX != SIZE_MAX
    wcs[0] = L'X';
    rc = mbstowcs_s( &retval, wcs, ~0, "\x81\xFC""foo", 10 );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'X' );
    VERIFY( ++violations == NumViolations );

    rc = mbstowcs_s( &retval, wcs, 20, "\x81\xFC""foo", ~0 );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );
#endif

    wcs[0] = L'X';
    rc = mbstowcs_s( &retval, wcs, 0, "\x81\xFC""foo", 10 );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'X' );
    VERIFY( ++violations == NumViolations );

    rc = mbstowcs_s( &retval, wcs, 2, "\x81\xFC""foo", 10 );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );

#if 0
    /* don't know how to force an encoding error */
    rc = mbstowcs_s( &retval, wcs, 20, "\x01\x11""foo", 10 );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );
#endif

    pchar = "\x81\xFC""foo";
    rc = mbsrtowcs_s( NULL, wcs, 20, &pchar, 10, &ps );
    VERIFY( rc != 0 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );

    rc = mbsrtowcs_s( &retval, wcs, 6, NULL, 10, &ps );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );

    pchar = NULL;
    rc = mbsrtowcs_s( &retval, wcs, 6, &pchar, 10, &ps );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );

#if RSIZE_MAX != SIZE_MAX
    wcs[0] = L'X';
    pchar = "\x81\xFC""foo";
    rc = mbsrtowcs_s( &retval, wcs, ~0, &pchar, 10, &ps );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'X' );
    VERIFY( ++violations == NumViolations );

    pchar = "\x81\xFC""foo";
    rc = mbsrtowcs_s( &retval, wcs, 20, &pchar, ~0, &ps );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );
#endif

    pchar = "\x81\xFC""foo";
    wcs[0] = L'X';
    rc = mbsrtowcs_s( &retval, wcs, 0, &pchar, 10, &ps );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'X' );
    VERIFY( ++violations == NumViolations );

    pchar = "\x81\xFC""foo";
    rc = mbsrtowcs_s( &retval, wcs, 2, &pchar, 10, &ps );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );
#if 0
    /* don't know how to force an encoding error */
    pchar = "\x01\x11""foo";
    rc = mbsrtowcs_s( &retval, wcs, 20, &pchar, 10, &ps );
    VERIFY( rc != 0 );
    VERIFY( retval == -1 );
    VERIFY( wcs[0] == L'\0' );
    VERIFY( ++violations == NumViolations );
#endif
}



/****
***** Test wctomb_s() and wcstombs_s().
****/

void TestFromWide( void )
{
    char                mbs[20];
    wchar_t             wc = L'\0';
    wchar_t             wcs[] = { 'H', 'e', 'l', 'l', 'o', '\0' };

    errno_t             rc;
    size_t              retval;
    size_t              retval2;
    int                 status;
    int                 violations = NumViolations;

    rc = wctomb_s( &status, mbs, 20, wc );
    VERIFY( rc == 0 );
    VERIFY( mbs[0] == '\0' );
    VERIFY( status == 1 );
    VERIFY( violations == NumViolations );

    rc = wctomb_s( &status, mbs, 20, L'X' );
    VERIFY( rc == 0 );
    VERIFY( mbs[0] == 'X' );
    VERIFY( status == 1 );
    VERIFY( violations == NumViolations );

    rc = wctomb_s( &status, NULL, 0, L'X' );