用C语言实现PHP的dirname函数

写一个程序要用到，于是整理了一下。在PHP源码中的ext/standard/string.c中。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <windows.h>

/***********************************
Author  : Demon
Website : https://demon.tw
E-mail  : 380401911@qq.com
***********************************/

#define IS_SLASH_P(c)  (*(c) == '/' || (*(c) == '\\' && !IsDBCSLeadByte(*(c-1))))
#define DEFAULT_SLASH  '\\'


char *estrndup(const char *s, size_t length)
{
    char *p;

    p = (char *) malloc(length+1);
    if (p == NULL) {
        return p;
    }
    memcpy(p, s, length);
    p[length] = 0;
    return p;
}


/* {{{ php_dirname
   Returns directory name component of path */
size_t php_dirname(char *path, size_t len)
{
    register char *end = path + len - 1;
    unsigned int len_adjust = 0;

    /* Note that on Win32 CWD is per drive (heritage from CP/M).
     * This means dirname("c:foo") maps to "c:." or "c:" - which means CWD on C: drive.
     */
    if ((2 <= len) && isalpha((int)((unsigned char *)path)[0]) && (':' == path[1])) {
        /* Skip over the drive spec (if any) so as not to change */
        path += 2;
        len_adjust += 2;
        if (2 == len) {
            /* Return "c:" on Win32 for dirname("c:").
             * It would be more consistent to return "c:." 
             * but that would require making the string *longer*.
             */
            return len;
        }
    }

    if (len == 0) {
        /* Illegal use of this function */
        return 0;
    }

    /* Strip trailing slashes */
    while (end >= path && IS_SLASH_P(end)) {
        end--;
    }
    if (end < path) {
        /* The path only contained slashes */
        path[0] = DEFAULT_SLASH;
        path[1] = '\0';
        return 1 + len_adjust;
    }

    /* Strip filename */
    while (end >= path && !IS_SLASH_P(end)) {
        end--;
    }
    if (end < path) {
        path[0] = '.';
        path[1] = '\0';
        return 1 + len_adjust;
    }

    /* Strip slashes which came before the file name */
    while (end >= path && IS_SLASH_P(end)) {
        end--;
    }
    if (end < path) {
        path[0] = DEFAULT_SLASH;
        path[1] = '\0';
        return 1 + len_adjust;
    }
    *(end+1) = '\0';

    return (size_t)(end + 1 - path) + len_adjust;
}
/* }}} */

/* {{{ proto string dirname(string path)
   Returns the directory name component of the path */
char *dirname(char *str)
{
    char *ret;
    size_t ret_len;

    ret = estrndup(str, strlen(str));
    ret_len = php_dirname(ret, strlen(str));

    return ret;
}
/* }}} */

int main(void)
{
    char *file = dirname("/etc/passwd");
    printf("%s\n", file);
    free(file);
    return 0;
}

随机文章:

1. Python pdfplumber内存泄露问题解决方案
2. OpenWrt搭建KMS服务器激活Windows和Office
3. OpenWrt挂载USB储存设备实现Samba共享
4. DOSLNK.EXE下载
5. 邪恶的eval和new Function

这篇文章发布于 2011年03月13日，星期日，22:36，归类于 程序设计。 您可以跟踪这篇文章的评论通过 RSS 2.0 feed。 您可以留下评论，或者从您的站点trackback。