Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

134 lines
2.4 KiB

/*++
Copyright (c) 2000 Microsoft Corporation
Module Name:
spprintf.c
Abstract:
safer sprintf variants
Author:
Jay Krell (a-JayK) November 2000
Revision History:
--*/
#include "spprecmp.h"
#pragma hdrstop
#include <stdarg.h>
#include <stdio.h>
#include "spprintf.h"
#include "spcab.h"
//
// _snprintf and co. do not write a terminal nul when the string just fits.
// These function do.
//
void
SpFormatStringVaA(
PSTR Buffer,
SIZE_T Size,
PCSTR Format,
va_list Args
)
{
if (Buffer != NULL && Size != 0)
{
Buffer[0] = 0;
Size -= 1;
if (Size != 0)
_vsnprintf(Buffer, Size, Format, Args);
Buffer[Size] = 0;
}
}
void
__cdecl
SpFormatStringA(
PSTR Buffer,
SIZE_T Size,
PCSTR Format,
...
)
{
va_list Args;
va_start(Args, Format);
SpFormatStringVaA(Buffer, Size, Format, Args);
va_end(Args);
}
void
SpFormatStringVaW(
PWSTR Buffer,
SIZE_T Size,
PCWSTR Format,
va_list Args
)
{
if (Buffer != NULL && Size != 0)
{
Buffer[0] = 0;
Size -= 1;
if (Size != 0)
_vsnwprintf(Buffer, Size, Format, Args);
Buffer[Size] = 0;
}
}
void
__cdecl
SpFormatStringW(
PWSTR Buffer,
SIZE_T Size,
PCWSTR Format,
...
)
{
va_list Args;
va_start(Args, Format);
SpFormatStringVaW(Buffer, Size, Format, Args);
va_end(Args);
}
NTSTATUS
__cdecl
SpFormatStringWToA(
PSTR Buffer,
SIZE_T Size,
PCWSTR Format,
...
)
{
va_list Args;
UNICODE_STRING UnicodeBuffer = { 0 };
ANSI_STRING AnsiBuffer = { 0 };
NTSTATUS Status = STATUS_SUCCESS;
va_start(Args, Format);
UnicodeBuffer.Buffer = (PWSTR)SpMemAlloc(Size * sizeof(UnicodeBuffer.Buffer[0]));
if (UnicodeBuffer.Buffer == NULL) {
Status = STATUS_NO_MEMORY;
goto Exit;
}
if (Size != 0) {
UnicodeBuffer.Buffer[0] = 0;
}
SpFormatStringVaW(UnicodeBuffer.Buffer, Size, Format, Args);
UnicodeBuffer.Length = (USHORT)(wcslen(UnicodeBuffer.Buffer) + 1) * sizeof(UnicodeBuffer.Buffer[0]);
AnsiBuffer.MaximumLength = (USHORT)Size * sizeof(AnsiBuffer.Buffer[0]);
Status = SpUnicodeStringToAnsiString(&AnsiBuffer, &UnicodeBuffer, FALSE);
Exit:
va_end(Args);
return Status;
}