src/gallium/auxiliary/util/u_stream_wd.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2008 Tungsten Graphics, Inc., Cedar Park, Texas.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28 /**
  29  * @file
  30  * Stream implementation for the Windows Display driver.
  31  */
  32
  33 #include "pipe/p_config.h"
  34
  35 #if defined(PIPE_SUBSYSTEM_WINDOWS_DISPLAY)
  36
  37 #include <windows.h>
  38 #include <winddi.h>
  39
  40 #include "util/u_memory.h"
  41 #include "util/u_string.h"
  42
  43 #include "u_stream.h"
  44
  45
  46 #define MAP_FILE_SIZE (4*1024*1024)
  47
  48
  49 struct util_stream
  50 {
  51    char filename[MAX_PATH + 1];
  52    WCHAR wFileName[MAX_PATH + 1];
  53    boolean growable;
  54    size_t map_size;
  55    ULONG_PTR iFile;
  56    char *pMap;
  57    size_t written;
  58    unsigned suffix;
  59 };
  60
  61
  62 static INLINE boolean
  63 util_stream_map(struct util_stream *stream)
  64 {
  65    ULONG BytesInUnicodeString;
  66    static char filename[MAX_PATH + 1];
  67    unsigned filename_len;
  68
  69    if(stream->growable)
  70       filename_len = util_snprintf(filename,
  71                                    sizeof(filename),
  72                                    "%s.%04x",
  73                                    stream->filename,
  74                                    stream->suffix++);
  75    else
  76       filename_len = util_snprintf(filename,
  77                                    sizeof(filename),
  78                                    "%s",
  79                                    stream->filename);
  80
  81    EngMultiByteToUnicodeN(
  82          stream->wFileName,
  83          sizeof(stream->wFileName),
  84          &BytesInUnicodeString,
  85          filename,
  86          filename_len);
  87
  88    stream->pMap = EngMapFile(stream->wFileName, stream->map_size, &stream->iFile);
  89    if(!stream->pMap)
  90       return FALSE;
  91
  92    memset(stream->pMap, 0, stream->map_size);
  93    stream->written = 0;
  94
  95    return TRUE;
  96 }
  97
  98
  99 static INLINE void
 100 util_stream_unmap(struct util_stream *stream)
 101 {
 102    EngUnmapFile(stream->iFile);
 103    if(stream->written < stream->map_size) {
 104       /* Truncate file size */
 105       stream->pMap = EngMapFile(stream->wFileName, stream->written, &stream->iFile);
 106       if(stream->pMap)
 107          EngUnmapFile(stream->iFile);
 108    }
 109
 110    stream->pMap = NULL;
 111 }
 112
 113
 114 static INLINE void
 115 util_stream_full_qualified_filename(char *dst, size_t size, const char *src)
 116 {
 117    boolean need_drive, need_root;
 118
 119    if((('A' <= src[0] && src[0] <= 'Z') || ('a' <= src[0] && src[0] <= 'z')) && src[1] == ':') {
 120       need_drive = FALSE;
 121       need_root = src[2] == '\\' ? FALSE : TRUE;
 122    }
 123    else {
 124       need_drive = TRUE;
 125       need_root = src[0] == '\\' ? FALSE : TRUE;
 126    }
 127
 128    util_snprintf(dst, size,
 129                  "\\??\\%s%s%s",
 130                  need_drive ? "C:" : "",
 131                  need_root ? "\\" : "",
 132                  src);
 133 }
 134
 135
 136 struct util_stream *
 137 util_stream_create(const char *filename, size_t max_size)
 138 {
 139    struct util_stream *stream;
 140
 141    stream = CALLOC_STRUCT(util_stream);
 142    if(!stream)
 143       goto error1;
 144
 145    util_stream_full_qualified_filename(stream->filename,
 146                                        sizeof(stream->filename),
 147                                        filename);
 148
 149    if(max_size) {
 150       stream->growable = FALSE;
 151       stream->map_size = max_size;
 152    }
 153    else {
 154       stream->growable = TRUE;
 155       stream->map_size = MAP_FILE_SIZE;
 156    }
 157
 158    if(!util_stream_map(stream))
 159       goto error2;
 160
 161    return stream;
 162
 163 error2:
 164    FREE(stream);
 165 error1:
 166    return NULL;
 167 }
 168
 169
 170 static INLINE void
 171 util_stream_copy(struct util_stream *stream, const char *data, size_t size)
 172 {
 173    assert(stream->written + size <= stream->map_size);
 174    memcpy(stream->pMap + stream->written, data, size);
 175    stream->written += size;
 176 }
 177
 178
 179 boolean
 180 util_stream_write(struct util_stream *stream, const void *data, size_t size)
 181 {
 182    if(!stream)
 183       return FALSE;
 184
 185    if(!stream->pMap)
 186       return FALSE;
 187
 188    while(stream->written + size > stream->map_size) {
 189       size_t step = stream->map_size - stream->written;
 190       util_stream_copy(stream, data, step);
 191       data = (const char *)data + step;
 192       size -= step;
 193
 194       util_stream_unmap(stream);
 195       if(!stream->growable || !util_stream_map(stream))
 196          return FALSE;
 197    }
 198
 199    util_stream_copy(stream, data, size);
 200
 201    return TRUE;
 202 }
 203
 204
 205 void
 206 util_stream_flush(struct util_stream *stream)
 207 {
 208    (void)stream;
 209 }
 210
 211
 212 void
 213 util_stream_close(struct util_stream *stream)
 214 {
 215    if(!stream)
 216       return;
 217
 218    util_stream_unmap(stream);
 219
 220    FREE(stream);
 221 }
 222
 223
 224 #endif