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