kern/kfile.c

   1 /**
   2  * \file
   3  * <!--
   4  * This file is part of BeRTOS.
   5  *
   6  * Bertos is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  * As a special exception, you may use this file as part of a free software
  21  * library without restriction.  Specifically, if other files instantiate
  22  * templates or use macros or inline functions from this file, or you compile
  23  * this file and link it with other files to produce an executable, this
  24  * file does not by itself cause the resulting executable to be covered by
  25  * the GNU General Public License.  This exception does not however
  26  * invalidate any other reasons why the executable file might be covered by
  27  * the GNU General Public License.
  28  *
  29  * Copyright 2007 Develer S.r.l. (http://www.develer.com/)
  30  *
  31  * -->
  32  *
  33  * \brief Virtual KFile I/O interface.
  34  * This module implements some generic I/O interfaces for kfile.
  35  *
  36  * \version $Id$
  37  * \author Francesco Sacchi <batt@develer.com>
  38  * \author Daniele Basile <asterix@develer.com>
  39  *
  40  */
  41
  42
  43 #include "kfile.h"
  44 #include <appconfig.h>
  45
  46 #include <cfg/debug.h>
  47 #include <mware/formatwr.h>
  48 #include <string.h>
  49
  50 /*
  51  * Sanity check for config parameters required by this module.
  52  */
  53 #if !defined(CONFIG_KFILE_GETS) || ((CONFIG_KFILE_GETS != 0) && CONFIG_KFILE_GETS != 1)
  54         #error CONFIG_KFILE_GETS must be set to either 0 or 1 in appconfig.h
  55 #endif
  56 #if !defined(CONFIG_PRINTF)
  57         #error CONFIG_PRINTF missing in appconfig.h
  58 #endif
  59
  60
  61 /**
  62  * Generic putc implementation using \a fd->write.
  63  */
  64 int kfile_putc(int _c, struct KFile *fd)
  65 {
  66         unsigned char c = (unsigned char)_c;
  67
  68         if (kfile_write(fd, &c, sizeof(c)) == sizeof(c))
  69                 return (int)((unsigned char)_c);
  70         else
  71                 return EOF;
  72 }
  73
  74 /**
  75  * Generic getc implementation using \a fd->read.
  76  */
  77 int kfile_getc(struct KFile *fd)
  78 {
  79         unsigned char c;
  80
  81         if (kfile_read(fd, &c, sizeof(c)) == sizeof(c))
  82                 return (int)((unsigned char)c);
  83         else
  84                 return EOF;
  85 }
  86
  87 #if CONFIG_PRINTF
  88 /**
  89  * Formatted write.
  90  */
  91 int kfile_printf(struct KFile *fd, const char *format, ...)
  92 {
  93         va_list ap;
  94         int len;
  95
  96         va_start(ap, format);
  97         len = _formatted_write(format, (void (*)(char, void *))kfile_putc, fd, ap);
  98         va_end(ap);
  99
 100         return len;
 101 }
 102 #endif /* CONFIG_PRINTF */
 103
 104 /**
 105  * Write a string to kfile \a fd.
 106  * \return 0 if OK, EOF in case of error.
 107  */
 108 int kfile_print(struct KFile *fd, const char *s)
 109 {
 110         while (*s)
 111         {
 112                 if (kfile_putc(*s++, fd) == EOF)
 113                         return EOF;
 114         }
 115         return 0;
 116 }
 117
 118 #if CONFIG_KFILE_GETS
 119 /**
 120  * Read a line long at most as size and put it
 121  * in buf.
 122  * \return number of chars read or EOF in case
 123  *         of error.
 124  */
 125 int kfile_gets(struct KFile *fd, char *buf, int size)
 126 {
 127         return kfile_gets_echo(fd, buf, size, false);
 128 }
 129
 130
 131 /**
 132  * Read a line long at most as size and put it
 133  * in buf, with optional echo.
 134  *
 135  * \return number of chars read, or EOF in case
 136  *         of error.
 137  */
 138 int kfile_gets_echo(struct KFile *fd, char *buf, int size, bool echo)
 139 {
 140         int i = 0;
 141         int c;
 142
 143         for (;;)
 144         {
 145                 if ((c = kfile_getc(fd)) == EOF)
 146                 {
 147                         buf[i] = '\0';
 148                         return -1;
 149                 }
 150
 151                 /* FIXME */
 152                 if (c == '\r' || c == '\n' || i >= size-1)
 153                 {
 154                         buf[i] = '\0';
 155                         if (echo)
 156                                 kfile_print(fd, "\r\n");
 157                         break;
 158                 }
 159                 buf[i++] = c;
 160                 if (echo)
 161                         kfile_putc(c, fd);
 162         }
 163
 164         return i;
 165 }
 166 #endif /* !CONFIG_KFILE_GETS */
 167
 168
 169 /**
 170  * Move \a fd file seek position of \a offset bytes from \a whence.
 171  *
 172  * This is a generic implementation of seek function, you can redefine
 173  * it in your local module is needed.
 174  */
 175 kfile_off_t kfile_genericSeek(struct KFile *fd, kfile_off_t offset, KSeekMode whence)
 176 {
 177         uint32_t seek_pos;
 178
 179         switch (whence)
 180         {
 181
 182         case KSM_SEEK_SET:
 183                 seek_pos = 0;
 184                 break;
 185         case KSM_SEEK_END:
 186                 seek_pos = fd->size - 1;
 187                 break;
 188         case KSM_SEEK_CUR:
 189                 seek_pos = fd->seek_pos;
 190                 break;
 191         default:
 192                 ASSERT(0);
 193                 return EOF;
 194                 break;
 195         }
 196
 197         /* Bound check */
 198         if (seek_pos + offset > fd->size)
 199         {
 200                 ASSERT(0);
 201                 return EOF;
 202         }
 203
 204         fd->seek_pos = seek_pos + offset;
 205
 206         return fd->seek_pos;
 207 }
 208
 209 #if CONFIG_TEST
 210
 211 /**
 212  * KFile read/write subtest.
 213  * Try to write/read in the same \a f file location \a size bytes.
 214  * \return true if all is ok, false otherwise
 215  * \note Restore file position at exit (if no error)
 216  * \note Test buffer \a buf must be filled with
 217  * the following statement:
 218  * <pre>
 219  * buf[i] = i & 0xff
 220  * </pre>
 221  */
 222 static bool kfile_rwTest(KFile *f, uint8_t *buf, size_t size)
 223 {
 224         /*
 225          * Write test buffer
 226          */
 227         if (kfile_write(f, buf, size) != size)
 228                 return false;
 229
 230         kfile_seek(f, -(kfile_off_t)size, KSM_SEEK_CUR);
 231
 232         /*
 233          * Reset test buffer
 234          */
 235         memset(buf, 0, size);
 236
 237         /*
 238          * Read file in test buffer
 239          */
 240         if (kfile_read(f, buf, size) != size)
 241                 return false;
 242         kfile_seek(f, -(kfile_off_t)size, KSM_SEEK_CUR);
 243
 244         /*
 245          * Check test result
 246          */
 247         for (size_t i = 0; i < size; i++)
 248                 if (buf[i] != (i & 0xff))
 249                         return false;
 250
 251         return true;
 252 }
 253
 254 /**
 255  * KFile read/write test.
 256  * This function write and read \a test_buf long \a size
 257  * on \a fd handler.
 258  * \a save_buf can be NULL or a buffer where to save previous file content.
 259  */
 260 bool kfile_test(KFile *fd, uint8_t *test_buf, uint8_t *save_buf, size_t size)
 261 {
 262         /*
 263          * Part of test buf size that you would write.
 264          * This var is used in test 3 to check kfile_write
 265          * when writing beyond filesize limit.
 266          */
 267         kfile_off_t len = size / 2;
 268
 269
 270         /* Fill test buffer */
 271         for (size_t i = 0; i < size; i++)
 272                 test_buf[i] = (i & 0xff);
 273
 274         /*
 275          * If necessary, user can save content,
 276          * for later restore.
 277          */
 278         if (save_buf)
 279         {
 280                 kfile_read(fd, save_buf, size);
 281                 kprintf("Saved content..form [%lu] to [%lu]\n", fd->seek_pos, fd->seek_pos + size);
 282         }
 283
 284         /* TEST 1 BEGIN. */
 285         kprintf("Test 1: write from pos 0 to [%lu]\n", size);
 286
 287         /*
 288          * Seek to addr 0.
 289          */
 290         if (kfile_seek(fd, 0, KSM_SEEK_SET) != 0)
 291                 goto kfile_test_end;
 292
 293         /*
 294          * Test read/write to address 0..size
 295          */
 296         if (!kfile_rwTest(fd, test_buf, size))
 297                 goto kfile_test_end;
 298
 299         kprintf("Test 1: ok!\n");
 300
 301         /*
 302          * Restore previous read content.
 303          */
 304         if (save_buf)
 305         {
 306                 kfile_seek(fd, 0, KSM_SEEK_SET);
 307
 308                 if (kfile_write(fd, save_buf, size) != size)
 309                         goto kfile_test_end;
 310
 311                 kprintf("Restore content..form [%lu] to [%lu]\n", fd->seek_pos, fd->seek_pos + size);
 312         }
 313         /* TEST 1 END. */
 314
 315         /* TEST 2 BEGIN. */
 316         kprintf("Test 2: write from pos [%lu] to [%lu]\n", fd->size/2 , fd->size/2 + size);
 317
 318         /*
 319          * Go to half test size.
 320          */
 321         kfile_seek(fd, (fd->size / 2), KSM_SEEK_SET);
 322
 323         /*
 324          * If necessary, user can save content
 325          * for later restore.
 326          */
 327         if (save_buf)
 328         {
 329                 kfile_read(fd, save_buf, size);
 330                 kfile_seek(fd, -(kfile_off_t)size, KSM_SEEK_CUR);
 331                 kprintf("Saved content..form [%lu] to [%lu]\n", fd->seek_pos, fd->seek_pos + size);
 332         }
 333
 334         /*
 335          * Test read/write to address filesize/2 ... filesize/2 + size
 336          */
 337         if (!kfile_rwTest(fd, test_buf, size))
 338                 goto kfile_test_end;
 339
 340         kprintf("Test 2: ok!\n");
 341
 342         /*
 343          * Restore previous content.
 344          */
 345         if (save_buf)
 346         {
 347                 kfile_seek(fd, -(kfile_off_t)size, KSM_SEEK_CUR);
 348
 349                 if (kfile_write(fd, save_buf, size) != size)
 350                         goto kfile_test_end;
 351
 352                 kprintf("Restore content..form [%lu] to [%lu]\n", fd->seek_pos, fd->seek_pos + size);
 353         }
 354
 355         /* TEST 2 END. */
 356
 357         /* TEST 3 BEGIN. */
 358         kprintf("Test 3: write outside of fd->size limit [%lu]\n", fd->size);
 359         kprintf("This test should FAIL!, you must see an assertion fail message.\n");
 360
 361         /*
 362          * Go to the Flash end
 363          */
 364         kfile_seek(fd, -len, KSM_SEEK_END);
 365
 366         /*
 367          * If necessary, user can save content,
 368          * for later restore.
 369          */
 370         if (save_buf)
 371         {
 372                 kfile_read(fd, save_buf, len);
 373                 kfile_seek(fd, -len, KSM_SEEK_CUR);
 374                 kprintf("Saved content..form [%lu] to [%lu]\n", fd->seek_pos, fd->seek_pos + len);
 375         }
 376
 377         /*
 378          * Test read/write to address (filesize - size) ... filesize
 379          */
 380         if (kfile_rwTest(fd, test_buf, size))
 381                 goto kfile_test_end;
 382
 383         kprintf("Test 3: ok!\n");
 384
 385         /*
 386          * Restore previous read content
 387          */
 388         if (save_buf)
 389         {
 390                 kfile_seek(fd, -len, KSM_SEEK_END);
 391
 392                 if (kfile_write(fd, save_buf, len) != len)
 393                         goto kfile_test_end;
 394
 395                 kprintf("Restore content..form [%lu] to [%lu]\n", fd->seek_pos, fd->seek_pos + len);
 396         }
 397
 398         /* TEST 3 END. */
 399
 400         kfile_close(fd);
 401         return true;
 402
 403 kfile_test_end:
 404         kfile_close(fd);
 405         return false;
 406 }
 407
 408 #endif /* CONFIG_TEST */