diff options
| -rw-r--r-- | lstrlib.c | 11 |
1 files changed, 8 insertions, 3 deletions
| @@ -1,5 +1,5 @@ | |||
| 1 | /* | 1 | /* |
| 2 | ** $Id: lstrlib.c,v 1.234 2015/09/28 18:05:01 roberto Exp roberto $ | 2 | ** $Id: lstrlib.c,v 1.235 2015/10/08 15:53:05 roberto Exp roberto $ |
| 3 | ** Standard library for string operations and pattern-matching | 3 | ** Standard library for string operations and pattern-matching |
| 4 | ** See Copyright Notice in lua.h | 4 | ** See Copyright Notice in lua.h |
| 5 | */ | 5 | */ |
| @@ -1339,8 +1339,13 @@ static int str_pack (lua_State *L) { | |||
| 1339 | case Kchar: { /* fixed-size string */ | 1339 | case Kchar: { /* fixed-size string */ |
| 1340 | size_t len; | 1340 | size_t len; |
| 1341 | const char *s = luaL_checklstring(L, arg, &len); | 1341 | const char *s = luaL_checklstring(L, arg, &len); |
| 1342 | luaL_argcheck(L, len == (size_t)size, arg, "wrong length"); | 1342 | if (size <= len) /* string larger than (or equal to) needed? */ |
| 1343 | luaL_addlstring(&b, s, size); | 1343 | luaL_addlstring(&b, s, size); /* truncate string to asked size */ |
| 1344 | else { /* string smaller than needed */ | ||
| 1345 | luaL_addlstring(&b, s, len); /* add it all */ | ||
| 1346 | while (len++ < size) /* pad extra space */ | ||
| 1347 | luaL_addchar(&b, LUA_PACKPADBYTE); | ||
| 1348 | } | ||
| 1344 | break; | 1349 | break; |
| 1345 | } | 1350 | } |
| 1346 | case Kstring: { /* strings with length count */ | 1351 | case Kstring: { /* strings with length count */ |
