[969] | 1 | /*
|
---|
| 2 | * $Id$
|
---|
| 3 | *
|
---|
| 4 | * New generation of string handling functions safe from a memory management point of view
|
---|
| 5 | * Developped by Andree Leidenfrost
|
---|
| 6 | */
|
---|
[767] | 7 |
|
---|
[768] | 8 | #include <stdio.h>
|
---|
| 9 | #include <string.h>
|
---|
[1074] | 10 |
|
---|
| 11 | #include "mr_mem.h"
|
---|
[969] | 12 |
|
---|
[767] | 13 | /**
|
---|
[969] | 14 | * Safe alternative to standard function strok()
|
---|
[767] | 15 | * Build a partition name from a drive and a partition number.
|
---|
[969] | 16 | * @param instr
|
---|
| 17 | * @param delims
|
---|
| 18 | * @param lastpos
|
---|
| 19 | * @return @p
|
---|
[767] | 20 | * @note this function allocates memory that needs to be freed by caller
|
---|
| 21 | **/
|
---|
[969] | 22 | char *mr_strtok(char *instr, const char *delims, int *lastpos)
|
---|
| 23 | {
|
---|
[767] | 24 |
|
---|
[969] | 25 | char *token = NULL;
|
---|
| 26 | char *strptr = NULL;
|
---|
| 27 | size_t pos1 = 0;
|
---|
| 28 | size_t pos2 = 0;
|
---|
[767] | 29 |
|
---|
[969] | 30 | if (strlen(instr) <= *lastpos) {
|
---|
| 31 | *lastpos = 0;
|
---|
| 32 | return token;
|
---|
| 33 | }
|
---|
[767] | 34 |
|
---|
[969] | 35 | strptr = instr + *lastpos;
|
---|
| 36 | pos2 = strspn(strptr, delims);
|
---|
| 37 | strptr += pos2;
|
---|
| 38 | pos1 = strcspn(strptr, delims);
|
---|
[1074] | 39 | token = (char *)mr_malloc(sizeof(*token) * (pos1 + 1));
|
---|
[969] | 40 | strncpy(token, strptr, pos1);
|
---|
| 41 | token[pos1] = '\0';
|
---|
| 42 | *lastpos = *lastpos + pos1 + pos2 + 1;
|
---|
[767] | 43 |
|
---|
| 44 | return token;
|
---|
| 45 | }
|
---|
| 46 |
|
---|
| 47 |
|
---|
[969] | 48 | /**
|
---|
| 49 | * Returns the string fed to it 'inptr' with all characters to escape given
|
---|
| 50 | * in 'toesc' prepended by escaping character 'escchr'.
|
---|
| 51 | * (Prepare strings for use in system() or popen() with this function.)
|
---|
| 52 | * @param instr
|
---|
| 53 | * @param toesc
|
---|
| 54 | * @param escchr
|
---|
| 55 | * @note this function allocates memory that needs to be freed by caller
|
---|
| 56 | **/
|
---|
| 57 | char *mr_stresc(char *instr, char *toesc, const char escchr)
|
---|
| 58 | {
|
---|
| 59 |
|
---|
| 60 | char *inptr = NULL;
|
---|
| 61 | char *retstr = NULL;
|
---|
| 62 | char *retptr = NULL;
|
---|
| 63 | char *escptr = NULL;
|
---|
| 64 | int cnt = 0;
|
---|
| 65 |
|
---|
| 66 | inptr = instr;
|
---|
| 67 |
|
---|
| 68 | // Count how many characters need escaping.
|
---|
| 69 | while (*inptr != '\0') {
|
---|
| 70 | escptr = toesc;
|
---|
| 71 | while (*escptr != '\0') {
|
---|
| 72 | if (*inptr == *escptr) {
|
---|
| 73 | // Found it, skip the rest.
|
---|
| 74 | cnt++;
|
---|
| 75 | inptr++;
|
---|
| 76 | break;
|
---|
| 77 | }
|
---|
| 78 | inptr++;
|
---|
| 79 | escptr++;
|
---|
| 80 | }
|
---|
| 81 | }
|
---|
| 82 | inptr = instr;
|
---|
| 83 |
|
---|
[1074] | 84 | retstr = (char *) mr_malloc(strlen(inptr) + cnt + 1);
|
---|
[969] | 85 | retptr = retstr;
|
---|
| 86 |
|
---|
| 87 | // Prepend specified characters with escape character.
|
---|
| 88 | while (*inptr != '\0') {
|
---|
| 89 | escptr = toesc;
|
---|
| 90 | while (*escptr != '\0') {
|
---|
| 91 | if (*inptr == *escptr) {
|
---|
| 92 | // Found it, skip the rest.
|
---|
| 93 | *retptr = escchr;
|
---|
| 94 | retptr++;
|
---|
| 95 | break;
|
---|
| 96 | }
|
---|
| 97 | escptr++;
|
---|
| 98 | }
|
---|
| 99 | *retptr = *inptr;
|
---|
| 100 | retptr++;
|
---|
| 101 | inptr++;
|
---|
| 102 | }
|
---|
| 103 | *retptr = '\0';
|
---|
| 104 |
|
---|
| 105 | return retstr;
|
---|
[767] | 106 | }
|
---|