maxmustermann/PowerShell
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Added test for buffer not having enough space to write computer name

Browse source
This commit is contained in:
Aaron 2015-07-30 13:47:31 -07:00
parent 385fb20cdf
commit da4d4d7a21
2 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/impl/getcomputername.cpp
View file

@ -5,8 +5,8 @@
BOOL GetComputerName(LPTSTR name, LPDWORD len)
{
errno = 0;
int host = gethostname(name, HOST_NAME_MAX);
size_t len2 = *len;
int host = gethostname(name, len2);
if(host == 0)
{
return TRUE;

7
src/tests/test-getcomputername.cpp
View file

@ -1,5 +1,6 @@
#include <gtest/gtest.h>
#include "getcomputername.h"
#include <iostream>
TEST(GetComputerName,simple)
{
@ -26,11 +27,13 @@ TEST(GetComputerName,simple)
ASSERT_EQ(hostnameString,hostnameStringTest);
}
TEST(GetComputerName,buffertosmall)
TEST(GetComputerName,bufferttoosmall)
{
char hostname[HOST_NAME_MAX];
std::string hostnameFunctionTest;
DWORD hostSize = 0;
BOOL getComputerName = GetComputerName(&hostnameFunctionTest[0], &hostSize);
ASSERT_TRUE(getComputerName != 0);
ASSERT_TRUE(getComputerName == 0);
char buffer[ 256 ];
EXPECT_EQ(errno, ERROR_BUFFER_OVERFLOW);
}