《程序员面试宝典》移除C/C++程序代码中的注释

最近在看《程序员面试宝典》，受益匪浅，这篇文章是5.8节的代码，加了我的注释，还规范了代码，共享之。 (*^__^*) ……

#include <string.h>

#include <stdlib.h>

#include <stdio.h>

#include <fcntl.h>

#include <io.h>

/**

* 功能：移除C/C++程序代码中的注释

*

* 输入：指向C/C++程序代码的指针

*/

void remove_comment(char *buf, size_t size)

{

char *p,//扫描指针

c, //暂存扫描字符

*end; //指向文末

char *sq_start, //记录单引号(')起始位置

*dq_start; //记录双引号(")起始位置

char *lc_start, //记录行注释(//)起始位置

*bc_start; //记录块注释(/*)起始位置

size_t len;

p = buf;

end = p + size;

sq_start = NULL;

dq_start = NULL;

lc_start = NULL;

bc_start = NULL;

while (p < end)

{

c = *p;

switch (c)

{

case '\'': //遇到单引号(')

if (dq_start || lc_start || bc_start)

{

//忽略字符串与注释中的单引号

p++;

continue;

}

if (NULL == sq_start)

{

sq_start = p;

p++;

}

else

{

len = p++ - sq_start;

if (2 == len && '\\' == *(sq_start+1))

{

//忽略字符中的单引号

continue;

}

sq_start = NULL;

}

break;

case '\"': //遇到双引号(")

if (sq_start || lc_start || bc_start)

{

//忽略字符与注释中的单引号

p++;

continue;

}

if (NULL == dq_start)

{

dq_start = p;

p++;

}

else

{

len = p++ - dq_start;

if ('\\' == *(p-2))

{

//忽略字符串中的双引号

continue;

}

dq_start = NULL;

}

break;

case '/': //遇到斜杠(/)

if (sq_start || dq_start || lc_start || bc_start)

{

//忽略字符、字符串与注释中的斜杠

p++;

continue;

}

c = *(p+1);

if ('/' == c)

{

lc_start = p;

p += 2;

}

else if ('*' == c)

{

bc_start = p;

p += 2;

}

else

{

p++; //忽略除号

}

break;

case '*': //遇到星号(*)

if (sq_start || dq_start || lc_start || NULL == bc_start)

{

//忽略字符、字符串与行注释中的星号，还有忽略乘号

p++;

continue;

}

if ('/' != *(p+1))

{

//忽略块注释中的星号

p++;

continue;

}

len = (p+1) - bc_start + 1;

memset(bc_start, ' ', len);

p += 2;

bc_start = NULL;

break;

case '\n': //遇到换行符

if (NULL == lc_start)

{

//不是行注释的换行符则忽略

p++;

continue;

}

c = *(p-1);

len = ('\r'==c ? p-1 : p) - lc_start;

memset(lc_start, ' ', len);

p++;

lc_start = NULL;

break;

default:

p++;

break;

}

}

//如果最后一行是行注释

if (lc_start)

{

len = p - lc_start;

memset(lc_start, ' ', len);

}

}

int main(int argc, char *argv[])

{

int fd, n;

char buf[102400];

fd = open("C:\\uuu.txt", O_RDONLY, 0);

if (-1 == fd)

{

return -1;

}

n = read(fd, buf, sizeof(buf));

if (-1==n || 0==n)

{

close(fd);

return -1;

}

remove_comment(buf, n);

*(buf+n) = '\0';

printf(buf);

close(fd);

return 0;

}

直接用这代码测试，结果截图：