Funzione C++ restituendo const char *

Question

Ho sviluppato una piccola funzione di supporto per una classe di test dell'unità, che prende il mio vector<unsigned char> e lo converte in un const char *. Ho scritto questo così posso passarlo alla macro ASSERT_STREQ di gtest per semplici confronti. Eccolo:

const char * convertVecToChar(std::vector<unsigned char>& source) 
{ 
    std::vector<unsigned char>::size_type size = source.size(); 
    char* data = (char*)malloc(sizeof(char) * (size + 1)); 
    memcpy(data, &source[0], size); 
    data[size] = 0; 
    return data; 
} 

Ed ecco un esempio di esso che è chiamato:

ASSERT_STREQ("de", convertVecToChar(somevector)); 

Presumo che questo è che perde tuttavia come sto chiamando malloc, ma senza chiamare delete a valle della strada?

Esiste un modo più elegante per eseguire questa operazione, che non comporta la creazione di una variabile const char * separata per ogni volta che si chiama ASSERT_STREQ all'interno di un metodo di prova?

Grazie mille in anticipo per tutte le risposte.

Chris

Answer 1

Return un std::string invece di un char* (malloc(), o new, non desiderati):

std::string convertVecToChar(std::vector<unsigned char>& source) 
{ 
    return std::string(source.begin(), source.end()); 
} 

ed impiego:

ASSERT_STREQ("de", convertVecToChar(somevector).c_str()); 

Answer 2

Si dovrebbe solo utilizzare string container - no è necessario preoccuparsi di perdite di memoria.

BTW - Poiché si utilizza C++, attenersi semplicemente a new e delete.

Answer 3

Stavo per raccomandare di usare anche std :: string, ma mi stavo chiedendo, perché non confrontare semplicemente il contenuto del vettore? È possibile accedere ai dati grezzi attraverso &source[0], in modo che si possa fare:

ASSERT_STREQ("de", (char*)&source[0]); 

Answer 4

sovraccarico operator==, allora si può solo utilizzare ASSERT_EQ:

bool operator==(const char* nullTerminatedChars, 
       const std::vector<char>& vecChars) 
{ 
    return std::string(nullTerminatedChars) == 
      std::string(vecChars.begin(), vecChars.end()); 
} 

Usa come:

std::vector<char> chars; 
ASSERT_EQ("de", chars); 

You' È necessario sovraccaricare anche operator<<(std::ostream& ..., poiché GoogleTest lo utilizza per convertire gli argomenti nell'asserzione in messaggi di errore se l'asserzione fallisce.

edit:

std::ostream& operator<<(std::ostream& os, const std::vector<char>& chars) 
{ 
    return os << std::string(chars.begin(), chars.end()); 
} 

Answer 5

source.push_back(0); 
ASSERT_STREQ("de", (char*)&source[0]); 
source.pop_back();