wordpress优化经历（六）——wordpress图片防盗链

不愿意做鱼的小鲸鱼

已于 2023-03-20 10:33:02 修改

阅读量384

点赞数

分类专栏： wordpress 文章标签： javascript web

于 2023-03-20 10:32:43 首次发布

本文链接：https://blog.csdn.net/qq_42038623/article/details/129662349

版权

wordpress 专栏收录该内容

10 篇文章 3 订阅

订阅专栏

背景

由于我写博客的习惯是先在本地把博客写好，然后图片什么的先放在本地，之后将博客在CSDN上进行markdown排版，并且将图片上传图加进去，在CSDN上先完成博客文章的发布；最后就直接将CSDN上的排版好的文章进行复制粘贴进自己的个人博客网站上，这样做的好处就是文章不用重新排版了，然后里面的图片资源也上传到了CSDN的服务器上，之后自己网站引用图片也是使用了CSDN的图片资源。但是最近出现了一个问题，CSDN上的图片全部失效了，原因是图片被防盗链处理，请求全部被跨域访问了，导致我个人网站的所有使用过CSDN排版的博客图片全部链接失效了，这导致我很头疼，这篇文章就记录一下这次图片被防盗链事故的处理经过。

在这里插入图片描述

解决方法

WordPress很多插件或者代码都可以实现在编辑文章中自动将外链图片下载到本地，但是我试了几个效果都不是很好。
最后选择了一个比较笨一点的方法，但是效果贼好。

将下面代码加到当前主题函数模板 functions.php 中：

function ecp_save_post($post_id, $post) {
  global $wpdb;
  if($post->post_status == 'publish') {
    $p   = '/<img.*[\s]src=[\"|\'](.*)[\"|\'].*>/iU';
    $num = preg_match_all($p, $post->post_content, $matches);
    if ($num) {
      $wp_upload_dir = wp_upload_dir();
      set_time_limit(0);
      $ch = curl_init();
      curl_setopt($ch, CURLOPT_HEADER, false);
      curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);
      curl_setopt($ch, CURLOPT_SSL_VERIFYPEER, false);
      curl_setopt($ch, CURLOPT_FOLLOWLOCATION, true);
      curl_setopt($ch, CURLOPT_MAXREDIRS,20);
      curl_setopt($ch, CURLOPT_CONNECTTIMEOUT, 30);
 
      $ecp_options = $_SERVER['HTTP_HOST'];
      foreach ($matches[1] as $src) {
        if (isset($src) && strpos($src, $ecp_options) === false) {
          $file_info = wp_check_filetype(basename($src), null);
          if ($file_info['ext'] == false) {
            date_default_timezone_set('PRC');
            $file_name = date('YmdHis-').dechex(mt_rand(100000, 999999)).'.tmp';
          } else {
            $file_name = dechex(mt_rand(100000, 999999)) . '-' . basename($src);
          }
          curl_setopt($ch, CURLOPT_URL, $src);
          $file_path = $wp_upload_dir['path'] . '/' . $file_name;
          $img = fopen($file_path, 'wb');
          curl_setopt($ch, CURLOPT_FILE, $img);
          $img_data  = curl_exec($ch);
          fclose($img);
 
          if (file_exists($file_path) && filesize($file_path) > 0) {
            $t   = curl_getinfo($ch, CURLINFO_CONTENT_TYPE);
            $arr = explode('/', $t);
            if (pathinfo($file_path, PATHINFO_EXTENSION) == 'tmp') {
              $file_path = ecp_handle_ext($file_path, $arr[1], $wp_upload_dir['path'], $file_name, 'tmp');
            } elseif (pathinfo($file_path, PATHINFO_EXTENSION) == 'webp') {
              $file_path = ecp_handle_ext($file_path, $arr[1], $wp_upload_dir['path'], $file_name, 'webp');
            }
            $post->post_content  = str_replace($src, $wp_upload_dir['url'] . '/' . basename($file_path), $post->post_content);
            $attachment = ecp_get_attachment_post(basename($file_path), $wp_upload_dir['url'] . '/' . basename($file_path));
            $attach_id = wp_insert_attachment($attachment, ltrim($wp_upload_dir['subdir'] . '/' . basename($file_path), '/'), 0);
            $attach_data = wp_generate_attachment_metadata($attach_id, $file_path);
            $ss = wp_update_attachment_metadata($attach_id, $attach_data);
          }
        }
      }
      curl_close($ch);
      $wpdb->update( $wpdb->posts, array('post_content' => $post->post_content), array('ID' => $post->ID));
    }
  }
}
 
function ecp_handle_ext($file, $type, $file_dir, $file_name, $ext) {
  switch ($ext) {
    case 'tmp':
      if (rename($file, str_replace('tmp', $type, $file))) {
        if ('webp' == $type) {
          return ecp_image_convert('webp', 'jpeg', $file_dir . '/' . str_replace('tmp', $type, $file_name));
        }
        return $file_dir . '/' . str_replace('tmp', $type, $file_name);
      }
    case 'webp':
      if ('webp' == $type) {
        return ecp_image_convert('webp', 'jpeg', $file);
      } else {
        if (rename($file, str_replace('webp', $type, $file))) {
          return $file_dir . '/' . str_replace('webp', $type, $file_name);
        }
      }
    default:
      return $file;
  }
}
 
function ecp_image_convert($from='webp', $to='jpeg', $image) {
  $im = imagecreatefromwebp($image);
  if (imagejpeg($im, str_replace('webp', 'jpeg', $image), 100)) {
    try {
      unlink($image);
    } catch (Exception $e) {
      $error_msg = sprintf('Error removing local file %s: %s', $image,
        $e->getMessage());
      error_log($error_msg);
    }
  }
  imagedestroy($im);
 
  return str_replace('webp', 'jpeg', $image);
}
 
function ecp_get_attachment_post($filename, $url) {
  $file_info  = wp_check_filetype($filename, null);
  return array(
    'guid'           => $url,
    'post_type'      => 'attachement',
    'post_mime_type' => $file_info['type'],
    'post_title'     => preg_replace('/\.[^.]+$/', '', $filename),
    'post_content'   => '',
    'post_status'    => 'inherit'
  );
}
add_action('save_post', 'ecp_save_post', 120, 2);