mbstow_s.c

来自「开放源码的编译器open watcom 1.6.0版的源代码」· C语言 代码 · 共 135 行

/****************************************************************************
*
*                            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:  Safer C Library Implementation of mbstowcs_s() _fmbstowcs_s()
*
****************************************************************************/


#include "variety.h"
#include "saferlib.h"
#include <mbstring.h>
#include "farfunc.h"



/****
***** Convert a multibyte character string to a wide character string.
****/

_WCRTLINK errno_t _NEARFAR(mbstowcs_s,_fmbstowcs_s)( size_t * __restrict retval,
                                                     wchar_t _FFAR * __restrict dst,
                                                     rsize_t dstmax,
                                                     const char _FFAR * __restrict src,
                                                     rsize_t len )
{
    int                 numChars = 0;
    int                 ret;
    errno_t             rc = -1;
    const char          *msg = NULL;
    const char _FFAR    *srcend = src + len;
    wchar_t _FFAR       *dststart = dst;
    rsize_t             maxlen;

    // Verify runtime-constraints
    // retval != NULL
    // src    != NULL
    // if dst == NULL then dstmax == 0
    // if dst != NULL then dstmax != 0
    if(__check_constraint_nullptr_msg( msg, retval ) &&
        __check_constraint_nullptr_msg( msg, src )) {
        if(dst == NULL) {                /* get required size */
            // ensure dstmax == 0
            if(__check_constraint_a_gt_b_msg( msg, dstmax, 0 )) {
                for( ;; ) {
                    if( *src != '\0' ) {
                        if(srcend < src) break;               //no null found
                        ret = _NEARFAR(mbtowc,_fmbtowc)( NULL, src, MB_LEN_MAX );
                        if( ret == -1 )  {
                            *retval = -1;
                            break;               //encoding error
                        }
                        src = _NEARFAR(_mbsinc,_fmbsinc)( src );
                        numChars++;
                    } else {
                        break;
                    }
                }
                if( ret != -1 ) {                         //no encoding error
                  *retval = numChars;
                  rc = 0;
                }
            }
        } else {                                    /* convert the string */
            // more runtime-constraints
            // len    <= RSIZE_MAX
            // dstmax <= RSIZE_MAX
            if(__check_constraint_maxsize_msg( msg, dstmax ) &&
               __check_constraint_maxsize_msg( msg, len ) &&
               __check_constraint_a_gt_b_msg( msg, 1, dstmax )) {

                for( maxlen = min(len, dstmax); maxlen > 0; maxlen-- ) {
                    if( *src != '\0' ) {
                        if(srcend < src) break;               //no null found
                        ret = _NEARFAR(mbtowc,_fmbtowc)( dst, src, MB_LEN_MAX );
                        if( ret == -1 )  {
                            *retval = -1;
                            break;               //encoding error
                        }
                        src = _NEARFAR(_mbsinc,_fmbsinc)( src );
                        dst++;
                        numChars++;
                    } else {
                        break;
                    }
                }
                if( (dstmax > len) ||
                    __check_constraint_toosmall_msg( msg, dst, maxlen )) {
                    *dst = L'\0';            // terminate string
                }
                if( (msg == NULL) && __check_constraint_a_gt_b_msg( msg, src, srcend )) {
                    if( ret != -1 ) {                         //no encoding error
                      *retval = numChars;
                      rc = 0;
                    }
                }
            }
        }
    }

    if(msg != NULL) {
        // Runtime-constraint found
        // set dst[0] to nullchar and *retval to -1
        if((dst != NULL) && (dstmax > 0) && __lte_rsizmax( dstmax ))
            *dststart = L'\0';
        if(retval != NULL)
            *retval = -1;
        // Now call the handler
        __rtct_fail( __func__, msg, NULL );
    }
    return( rc );
}