在图片中添加自己的文本信息(PNG及JPEG格式)

背景：

PNG格式和JPEG格式的图片都有一个表示文本信息的段，这些信息可以是任意的字符串，或者版权信息，或者工作室信息，或者其它的。本文的程序是在原始图片中添加这个段的信息。由于比较简单，不多说。

对比图片格式，小弟学习过一段时间，由于工作原因，对JPEG了解深一些，但只限于格式，像JPEG的压缩算法还没学习到。

代码如下：

/**************************************************************************

PNG：
段组成：
名称    占用字节      内容
长度       4        数据长度
段名       4       如：'IHDR'
数据       N       数据内容
CRC        4  CRC32值(段名及数据的CRC)

添加文本信息：
长度： strlen(text)
段名：'tEXt'
数据： text(内容自定)
CRC： crc32('tEXt'+text)

放置位置：位于IHDR后面
PNG头8字节，IHDR段25字节(内容13字节，其它12字节)，共33字节，
故放置33偏移量处后面。

JPEG：
组成：
名称   占用字节                  内容
段名      2       如：FF D8表示JPEG图片，FF FE表示注释段
长度      2          数据段长度+2(即这里的2字节)
数据      N                  数据内容

添加文件信息：
段名： FF FE
长度： strlen(text_len) + 2
数据： info(text)

放置位置：位于FF E0或FF E1段后面
读0x04、0x05处长度(即FFE0或FFE1的长度)，须转换成小端模式

PNG、JPEG图片为大端模式，以字节保存没问题，但数据长度必须转换成大端模式

写新文件方法：
先保存原来的开始部分，再插入文本信息，再保存剩下部分数据
**************************************************************************/

#include <stdio .h>
#include <stdlib .h>
#include <string .h>
#include  "crc.h"

const  char *text =  ">AUTHOR: Late Lee from http://www.latelee.org.\n";

#define little2big32(x) \
        ( (x& 0xff000000) >>  24 | (x& 0xff0000) >>  8 | (x& 0xff00) <<  8 | (x& 0xff) <<  24 )

#define little2big16(x) \
        ( (x& 0xff00) >>  8 | (x& 0xff) <<  8 )

#define HEAD_SIZE  33
int add_text_png( const  char *file,  const  char *new_file)
{
        FILE *in;
        FILE *out;
         int len =  0;
        size_t ret =  0;
         char *buffer;
         char *text_buf;
        size_t text_len =  0;
        size_t text_len_1 =  0;
         char text_type[ 4] = { 't',  'E',  'X',  't'};
         int text_crc =  0;

        in = fopen(file,  "rb");
         if (in == NULL)
        {
                printf( "Can not open file %s\n", file);
                 return - 1;
        }

        fseek(in,  0, SEEK_END);
        len = ftell(in);
        fseek(in,  0, SEEK_SET);

        printf( "file size: %d\n", len);

        buffer = ( char *)malloc( sizeof( char) * len);
         if (buffer == NULL)
        {
                 return - 1;
        }

         // read all to buffer
        ret = fread(buffer,  1, len, in);
         if (ret != len)
        {
                printf( "read file failed.\n");
                 return - 1;
        }

        out = fopen(new_file,  "wb");
         if (out == NULL)
        {
                printf( "open new file failed.\n");
                 return - 1;
        }

         // write 'head'
        ret = fwrite(buffer,  1, HEAD_SIZE, out);
         if (ret != HEAD_SIZE)
        {
                printf( "write head failed.\n");
                 return - 1;
        }

         /
        text_len = strlen(text);
        text_buf = ( char *)malloc( sizeof( char) * (text_len +  12));
        text_len_1 = little2big32(text_len);
        memcpy(text_buf, &text_len_1,  4);         // big endian
        memcpy(text_buf +  4, text_type,  4);
        memcpy(text_buf +  4 +  4, text, text_len);
        text_crc = crc32(( unsigned  char *)(text_buf +  4), (u32)text_len +  4);
        memcpy(text_buf +  4 +  4 + text_len, &text_crc,  4);

        printf( "text len: %d\n", text_len);

         // write information
        ret = fwrite(text_buf,  1, text_len +  12, out);
         if (ret != text_len +  12)
        {
                printf( "write head failed.\n");
                 return - 1;
        }

         // write 'left'
        ret = fwrite(buffer + HEAD_SIZE,  1, len - HEAD_SIZE, out);
         if (ret != len - HEAD_SIZE)
        {
                printf( "write head failed.\n");
                 return - 1;
        }

        fclose(in);
        fclose(out);

        free(buffer);
        free(text_buf);

        printf( "good job.\n");
        
         return  0;
}

int add_text_jpeg( const  char *file,  const  char *new_file)
{
        FILE *in;
        FILE *out;
         int len =  0;
        size_t ret =  0;
         unsigned  char *buffer;         // must be unsigned char
         char *text_buf;
        size_t text_len =  0;
        size_t text_len_1 =  0;
        size_t tmp_len =  0;
         unsigned  char text_type[ 2] = { 0xFF,  0xFE};

        in = fopen(file,  "rb");
         if (in == NULL)
        {
                printf( "Can not open file %s\n", file);
                 return - 1;
        }

        fseek(in,  0, SEEK_END);
        len = ftell(in);
        fseek(in,  0, SEEK_SET);

        printf( "file size: %d\n", len);

        buffer = ( unsigned  char *)malloc( sizeof( char) * len);
         if (buffer == NULL)
        {
                 return - 1;
        }

         // read all to buffer
        ret = fread(buffer,  1, len, in);
         if (ret != len)
        {
                printf( "read file failed.\n");
                 return - 1;
        }

         // read tmp_len
        tmp_len = *(buffer +  4) <<  8 | *(buffer +  5);
        printf( "tmp len: %x\n", tmp_len);

        out = fopen(new_file,  "wb");
         if (out == NULL)
        {
                printf( "open new file failed.\n");
                 return - 1;
        }

         // write 'head'
        ret = fwrite(buffer,  1, tmp_len +  4, out);
         if (ret != tmp_len +  4)
        {
                printf( "write head failed.\n");
                 return - 1;
        }

         /
        text_len = strlen(text);
        text_buf = ( char *)malloc( sizeof( char) * (text_len +  4));
        text_len_1 = little2big16(text_len +  2);
        memcpy(text_buf, text_type,  2);         // FF FE
        memcpy(text_buf +  2, &text_len_1,  2);         // len
        memcpy(text_buf +  2 +  2, text, text_len);         // text

        printf( "text len: %d\n", text_len);

         // write information
        ret = fwrite(text_buf,  1, text_len +  4, out);
         if (ret != text_len +  4)
        {
                printf( "write text failed.\n");
                 return - 1;
        }

         // write 'left'
        ret = fwrite(buffer + tmp_len +  4,  1, len - (tmp_len +  4), out);
         if (ret != len - (tmp_len +  4))
        {
                printf( "write file failed.\n");
                 return - 1;
        }

        fclose(in);
        fclose(out);

        free(buffer);
        free(text_buf);

        printf( "good job.\n");
        
         return  0;
}

int my_strnicmp( const  char *s1,  const  char *s2, size_t len)
{
         unsigned  char c1, c2;

         const  char *tmp1 = s1 + strlen(s1) -  1;
         const  char *tmp2 = s2 + strlen(s2) -  1;
        c1 =  0;        c2 =  0;
         if (len) {
                 do {
                        c1 = *tmp1; c2 = *tmp2;
                        tmp1--; tmp2--;
                         if (!c1)
                                 break;
                         if (!c2)
                                 break;
                         if (c1 == c2)
                                 continue;
                        c1 = tolower(c1);
                        c2 = tolower(c2);
                         if (c1 != c2)
                                 break;
                }  while (--len);
        }
         return ( int)c1 - ( int)c2;
}

// a.exe foo.jpg bar.jpg
// 其中的图片格式需要一致，由用户控制
int main( int argc,  char *argv[])
{
         int ret =  0;

         if (argc !=  3)
        {
                printf( "usage: %s old.jpg|png new.jpg|png.\n", argv[ 0]);
                 return - 1;
        }

         if (!my_strnicmp(argv[ 1],  "jpg",  3))
                add_text_jpeg(argv[ 1], argv[ 2]);

         else  if (!my_strnicmp(argv[ 1],  "png",  3))
                add_text_png(argv[ 1], argv[ 2]);
         else
        {
                printf( "not support.\n");
                 return - 1;
        }
         return  0;
}