Открытие потока через функцию - PullRequest
3 голосов
/ 29 июня 2011

Мне нужна помощь с не копируемой природой [io](f)stream с.

Мне нужно предоставить хакерскую оболочку около fstream s для обработки файлов с символами Юникода в их именах файлов в Windows. Для этого я разработал функцию-обертку:

bool open_ifstream( istream &stream, const string &filename )
{
#ifdef __GLIBCXX__
    FILE* result = _wfopen( convert_to_utf16(filename).c_str(), L"r" );
    if( result == 0 )
        return false;

    __gnu_cxx::stdio_filebuf<char>* buffer = new __gnu_cxx::stdio_filebuf<char>( result, std::ios_base::in, 1 );
    istream stream2(buffer);
    std::swap(stream, stream2);

#elif defined(_MSC_VER)
    stream.open( convert_to_utf16(filename) );
#endif
    return !!stream;
}

Конечно, линия std::swap является виновником. Я также попытался вернуть поток из функции, но это приводит к той же проблеме. Конструктор копирования std::istream равен delete d. Я также попробовал std::move, но это не помогло. Как мне обойти эту проблему?

РЕДАКТИРОВАТЬ: Я наконец нашел хороший способ для Keep It Simple (TM) и все же функциональный, благодаря идее @ tibur. Она все еще хакерская в том смысле, что она зависит от используемой библиотеки Windows Standard C ++, но так как используется только две настоящие , для меня это не проблема.

#include <fstream>
#include <memory>
#if _WIN32
# if __GLIBCXX__
#  include<ext/stdio_filebuf.h>
unique_ptr<istream> open_ifstream( const string &filename )
{
    FILE* c_file = _wfopen( convert_to_utf16(filename).c_str(), L"r" );
    __gnu_cxx::stdio_filebuf<char>* buffer = new __gnu_cxx::stdio_filebuf<char>( c_file, std::ios_base::in, 1 );

    return std::unique_ptr<istream>( new istream(buffer) );
}
# elif _MSC_VER
unique_ptr<ifstream> open_ifstream( const string &filename )
{
    return unique_ptr<ifstream>(new ifstream( convert_to_utf16(filename)) );
}
# else
# error unknown fstream implementation
# endif
#else
unique_ptr<ifstream> open_ifstream( const string &filename )
{
    return unique_ptr<ifstream>(new ifstream(filename) );
}
#endif

А в коде пользователя:

auto stream_ptr( open_ifstream(filename) );
auto &stream = *stream_ptr;
if( !stream )
    return emit_error( "Unable to open nectar file: " + filename );

Зависит от C ++ 0x <memory> и ключевого слова auto. Конечно, вы не можете просто получить close полученную переменную stream, но деструктор GNU Libstdc ++ std::istream позаботится о закрытии файла, поэтому никакого дополнительного управления памятью нигде не требуется.

Ответы [ 4 ]

3 голосов
/ 29 июня 2011

А как же:

ifstream * open_ifstream(const string &filename);
3 голосов
/ 29 июня 2011

Не могли бы вы просто использовать функцию-член rdbuf для прямой установки буфера stream?

2 голосов
/ 29 июня 2011

Умеренно ненавязчивая идея:

#include <iconv.h>
#include <algorithm>

void windowify(std::string & filename)
{
#ifdef WIN32
  assert(filename.length() < 1000);

  wchar_t wbuf[1000];
  char    cbuf[1000];
  char * ip = &cbuf[0];
  char * op = reinterpret_cast<char*>(&wbuf[0]);

  size_t ib = filename.length(), ob = 1000;

  std::fill(cbuf + filename.length(), cbuf + 1000, 0);
  std::copy(filename.begin(), filename.end(), cbuf);

  iconv_t cd = iconv_open("WCHAR_T", "UTF-8");
  iconv(cd, &ip, &ib, &op, &ob);
  iconv_close(cd);

  wchar_t sfnbuf[1000];
  std::fill(cbuf, cbuf + 1000, 0);

  ib = GetShortPathNameW(wbuf, sfnbuf, 1000);
  ob = 1000;
  ip = reinterpret_cast<char*>(&wbuf[0]);
  op = &cbuf[0];

  cd = iconv_open("UTF-8", "WCHAR_T");
  iconv(cd, &ip, &ib, &op, &ob);
  iconv_close(cd);

  filename = std::string(cbuf);
#endif
}

Использование:

std::string filename = getFilename();
windowify(filename);
std::ifstream infile(filename.c_str());
1 голос
/ 14 апреля 2015

Я бы предложил небольшое улучшение: используйте _wopen (или _wsopen_s) вместо _wfopen. Вы получите файловый дескриптор (int), который вы можете передать stdio_filebuf вместо FILE*. Таким образом, вы должны избегать утечки любого ресурса (как указано marcin)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...