anydragon

CHtmlView의 C++에서 생성한 JScript를 모든 Frame에 대해 Inject하여 실행하기

CHtmlView(CDHtmlDialog, CHtmlDialog)등에서 HTML을 로드하여 실행하게 됩니다.
혹시 이런 생각을 해보신적 있는지요?

기존 HTML 소스에 동적으로 JScript를 추가하고, 그 함수를 호출받고 싶다…
단, 추가할 JScript 소스는 C++에서 명시적으로 정의한다.

즉, 동일한 URL에 대해 기존의 일반 웹 브라우저와 조금 다른 웹(즉, 뭐.. 화면 상단에 강제적인 버튼 추가와 그 처리등등…)을 표현하고 싶다.
즉, C++에서 DocumentComplete Timing때, C++에서 정의한 JScript 함수를 넣는다. Body OnLoad()에 그 함수를 대체해서 넣는다. 즉, 기존 Html이

function OnLoad() {
  alert('hello');
}

function OnLoad() {

alert('hello');

}

function OnLoad() {
  alert('hello'); // <- 기존 내용
  inject();
}

function inject() {
  alert('injected');
}

function OnLoad() {

alert('hello'); // <- 기존 내용

inject();

}

function inject() {

alert('injected');

}

와 같이 수정하면, inject()가 실행될 것입니다.
즉, C++의 DocumentComplete에서,

InjectScript(..., "function OnLoad(){alert('hello');inject();}function inject(){alert('injected');}", ...)

1	InjectScript(..., "function OnLoad(){alert('hello');inject();}function inject(){alert('injected');}", ...)

를 호출하는 것입니다.
이를 위한 소스 코드를 아래와 같이 공유합니다.
사용은 CHtmlView의 OnDocumentComplete나 DocumentComplete 때 아래 함수를 호출하면 됩니다.
(view plain을 누르시면 코드 확인이 쉽습니다.)

HRESULT InjectJScriptAllFrame(IN CHtmlView * pcWnd, IN LPCTSTR lpszJScript, IN INT nMaxRecurseFrame) {
  HRESULT hr = S_OK;
  BOOL bStackOverflow = FALSE;
  IHTMLDocument2 * pIHtmlDocument2 = NULL;

  if ((NULL == pcWnd) || (NULL == lpszJScript)) {
    hr = HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER);
    ASSERT(FALSE);
    goto FINAL;
  }

  if (NULL == pcWnd -> GetHtmlDocument()) {
    hr = HRESULT_FROM_WIN32(ERROR_NOT_READY);
    ASSERT(FALSE);
    goto FINAL;
  }

  hr = pcWnd -> GetHtmlDocument() -> QueryInterface(IID_IHTMLDocument2, (VOID ** ) & pIHtmlDocument2);
  if (SUCCEEDED(hr)) {
    if (NULL == pIHtmlDocument2) {
      hr = HRESULT_FROM_WIN32(ERROR_INTERNAL_ERROR);
      ASSERT(FALSE);
      goto FINAL;
    }

    // HtmlDocument를 Frame 별로 recursive하게 JScript를 inject한다.
    // 만약, Frame이 MAX가 되었다면, hr은 HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW)가 리턴된다.
    hr = InjectJScriptByDocumentRecurse(pIHtmlDocument2, lpszJScript, nMaxRecurseFrame, 0, & bStackOverflow);

    if (HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW) == hr) {
      // Stack overflow인 경우에는 stop~!
      pcWnd -> Stop();
    }
  }

  FINAL:

    if (NULL != pIHtmlDocument2) {
      pIHtmlDocument2 -> Release();
      pIHtmlDocument2 = NULL;
    }

  return hr;
}

HRESULT InjectJScriptByDocumentRecurse(IN IHTMLDocument2 * pIHtmlDocument2, IN LPCTSTR lpszJScript, IN INT nMaxRecurseFrame, IN INT nCurDepth, OUT LPBOOL pbStackOverflow) {
  HRESULT hr = S_OK;
  HRESULT hr2 = S_OK;
  LONG nFrameCount = 0;
  LONG i = 0;
  VARIANT varIndex = {
    0,
  };
  VARIANT varDispWin = {
    0,
  };
  BSTR bstrScript = {
    0,
  };
  BSTR bstrElementType = {
    0,
  };
  BSTR bstrInsertWhere = {
    0,
  };
  IHTMLElement * pIHtmlElement = NULL;
  IHTMLElement2 * pIHtmlElement2 = NULL;
  IHTMLElement * pIHtmlElementScript = NULL;
  IHTMLScriptElement * pIHtmlScript = NULL;
  IHTMLFramesCollection2 * pIHtmlFramesCollection = NULL;
  IHTMLWindow2 * pIHtmlWindow = NULL;
  IHTMLDocument2 * pIHtmlDocument2Frame = NULL;

  if ((NULL == pIHtmlDocument2) || (NULL == lpszJScript) || (NULL == pbStackOverflow)) {
    hr = HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER);
    goto FINAL;
  }

  if (nCurDepth >= nMaxRecurseFrame) {
    hr = HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW);
    * pbStackOverflow = TRUE;
    goto FINAL;
  }

  if (TRUE == * pbStackOverflow) {
    hr = HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW);
    goto FINAL;
  }

  // insertAdjacentElement에 들어갈 Param을 결정한다.
  bstrScript = ::SysAllocString(T2COLE(lpszJScript));
  bstrElementType = ::SysAllocString(T2COLE(TEXT("script")));
  bstrInsertWhere = ::SysAllocString(T2COLE(TEXT("afterBegin")));

  hr = pIHtmlDocument2 -> get_body( & pIHtmlElement);
  if ((FAILED(hr)) || (NULL == pIHtmlElement)) {
    if ((SUCCEEDED(hr)) && (NULL == pIHtmlElement)) {
      hr = HRESULT_FROM_WIN32(ERROR_NOT_READY);
      goto FINAL;
    }

    pIHtmlElement = NULL;
    ASSERT(FALSE);
    goto FINAL;
  }

  hr = pIHtmlElement -> QueryInterface(IID_IHTMLElement2, (VOID ** ) & pIHtmlElement2);
  if (FAILED(hr)) {
    pIHtmlElement2 = NULL;
    ASSERT(FALSE);
    goto FINAL;
  }

  hr = pIHtmlDocument2 -> createElement(bstrElementType, & pIHtmlElementScript);
  if (FAILED(hr)) {
    pIHtmlElementScript = NULL;
    ASSERT(FALSE);
    goto FINAL;
  }

  hr = pIHtmlElementScript -> QueryInterface(IID_IHTMLScriptElement, (VOID ** ) & pIHtmlScript);
  if (FAILED(hr)) {
    pIHtmlScript = NULL;
    ASSERT(FALSE);
    goto FINAL;
  }

  hr = pIHtmlScript -> put_defer(VARIANT_TRUE);
  if (FAILED(hr)) {
    ASSERT(FALSE);
    goto FINAL;
  }

  hr = pIHtmlScript -> put_text(bstrScript);
  if (FAILED(hr)) {
    ASSERT(FALSE);
    goto FINAL;
  }

  hr = pIHtmlElement2 -> insertAdjacentElement(bstrInsertWhere, pIHtmlElementScript, NULL);
  if (FAILED(hr)) {
    ASSERT(FALSE);
    goto FINAL;
  }

  // 여기까지 왔다면, 일단 성공~
  hr = S_OK;

  hr2 = pIHtmlDocument2 -> get_frames( & pIHtmlFramesCollection);
  if (FAILED(hr2)) {
    // Frame이 없는 깔끔한 경우이다~
    goto FINAL;
  }

  hr2 = pIHtmlFramesCollection -> get_length( & nFrameCount);
  if (FAILED(hr2)) {
    goto FINAL;
  }

  for (i = 0; i < nFrameCount; i++) {
    varIndex.vt = VT_I4;
    varIndex.lVal = i;

    if (NULL != varDispWin.pdispVal) {
      varDispWin.pdispVal -> Release();
      varDispWin.pdispVal = NULL;
    }

  hr2 = pIHtmlFramesCollection -> item( & varIndex, & varDispWin);
  if (FAILED(hr2)) {
    continue;
  }

  if (NULL == varDispWin.pdispVal) {
    ASSERT(FALSE);
    continue;
  }

  if (NULL != pIHtmlWindow) {
    pIHtmlWindow -> Release();
    pIHtmlWindow = NULL;
  }

  hr2 = varDispWin.pdispVal -> QueryInterface(IID_IHTMLWindow2, (VOID ** ) & pIHtmlWindow);
  if (FAILED(hr2)) {
    continue;
  }

  if (NULL == pIHtmlWindow) {
    ASSERT(FALSE);
    continue;
  }

  if (NULL != pIHtmlDocument2Frame) {
    pIHtmlDocument2Frame -> Release();
    pIHtmlDocument2Frame = NULL;
  }

  hr2 = pIHtmlWindow -> get_document( & pIHtmlDocument2Frame);
  if (FAILED(hr2)) {
    continue;
  }

  if (NULL == pIHtmlDocument2Frame) {
    ASSERT(FALSE);
    continue;
  }

  // Recursive 실행~!!!
  hr2 = InjectJScriptByDocumentRecurse(pIHtmlDocument2Frame, lpszJScript, nMaxRecurseFrame, nCurDepth + 1, pbStackOverflow);

  // StackOverflow
  if (HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW) == hr2) {
    ( * pbStackOverflow) = TRUE;
    hr = HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW);
    break;
  }
}

FINAL:

  if (NULL != pIHtmlDocument2Frame) {
    pIHtmlDocument2Frame -> Release();
    pIHtmlDocument2Frame = NULL;
  }

if (NULL != pIHtmlWindow) {
  pIHtmlWindow -> Release();
  pIHtmlWindow = NULL;
}

if (NULL != varDispWin.pdispVal) {
  varDispWin.pdispVal -> Release();
  varDispWin.pdispVal = NULL;
}

if (NULL != pIHtmlScript) {
  pIHtmlScript -> Release();
  pIHtmlScript = NULL;
}

if (NULL != pIHtmlElementScript) {
  pIHtmlElementScript -> Release();
  pIHtmlElementScript = NULL;
}

if (NULL != pIHtmlElement2) {
  pIHtmlElement2 -> Release();
  pIHtmlElement2 = NULL;
}

if (NULL != pIHtmlElement) {
  pIHtmlElement -> Release();
  pIHtmlElement = NULL;
}

::SysFreeString(bstrScript);::SysFreeString(bstrElementType);::SysFreeString(bstrInsertWhere);

return hr;
}

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

243

244

245

246

247

248

249

250

251

252

253

254

255

256

257

258

259

260

HRESULT InjectJScriptAllFrame(IN CHtmlView * pcWnd, IN LPCTSTR lpszJScript, IN INT nMaxRecurseFrame) {

HRESULT hr = S_OK;

BOOL bStackOverflow = FALSE;

IHTMLDocument2 * pIHtmlDocument2 = NULL;

if ((NULL == pcWnd) || (NULL == lpszJScript)) {

hr = HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER);

ASSERT(FALSE);

goto FINAL;

}

if (NULL == pcWnd -> GetHtmlDocument()) {

hr = HRESULT_FROM_WIN32(ERROR_NOT_READY);

ASSERT(FALSE);

goto FINAL;

}

hr = pcWnd -> GetHtmlDocument() -> QueryInterface(IID_IHTMLDocument2, (VOID ** ) & pIHtmlDocument2);

if (SUCCEEDED(hr)) {

if (NULL == pIHtmlDocument2) {

hr = HRESULT_FROM_WIN32(ERROR_INTERNAL_ERROR);

ASSERT(FALSE);

goto FINAL;

}

// HtmlDocument를 Frame 별로 recursive하게 JScript를 inject한다.

// 만약, Frame이 MAX가 되었다면, hr은 HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW)가 리턴된다.

hr = InjectJScriptByDocumentRecurse(pIHtmlDocument2, lpszJScript, nMaxRecurseFrame, 0, & bStackOverflow);

if (HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW) == hr) {

// Stack overflow인 경우에는 stop~!

pcWnd -> Stop();

}

FINAL:

if (NULL != pIHtmlDocument2) {

pIHtmlDocument2 -> Release();

pIHtmlDocument2 = NULL;

}

return hr;

}

HRESULT InjectJScriptByDocumentRecurse(IN IHTMLDocument2 * pIHtmlDocument2, IN LPCTSTR lpszJScript, IN INT nMaxRecurseFrame, IN INT nCurDepth, OUT LPBOOL pbStackOverflow) {

HRESULT hr = S_OK;

HRESULT hr2 = S_OK;

LONG nFrameCount = 0;

LONG i = 0;

VARIANT varIndex = {

};

VARIANT varDispWin = {

};

BSTR bstrScript = {

};

BSTR bstrElementType = {

};

BSTR bstrInsertWhere = {

};

IHTMLElement * pIHtmlElement = NULL;

IHTMLElement2 * pIHtmlElement2 = NULL;

IHTMLElement * pIHtmlElementScript = NULL;

IHTMLScriptElement * pIHtmlScript = NULL;

IHTMLFramesCollection2 * pIHtmlFramesCollection = NULL;

IHTMLWindow2 * pIHtmlWindow = NULL;

IHTMLDocument2 * pIHtmlDocument2Frame = NULL;

if ((NULL == pIHtmlDocument2) || (NULL == lpszJScript) || (NULL == pbStackOverflow)) {

hr = HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER);

goto FINAL;

}

if (nCurDepth >= nMaxRecurseFrame) {

hr = HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW);

* pbStackOverflow = TRUE;

goto FINAL;

}

if (TRUE == * pbStackOverflow) {

hr = HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW);

goto FINAL;

}

// insertAdjacentElement에 들어갈 Param을 결정한다.

bstrScript = ::SysAllocString(T2COLE(lpszJScript));

bstrElementType = ::SysAllocString(T2COLE(TEXT("script")));

bstrInsertWhere = ::SysAllocString(T2COLE(TEXT("afterBegin")));

hr = pIHtmlDocument2 -> get_body( & pIHtmlElement);

if ((FAILED(hr)) || (NULL == pIHtmlElement)) {

if ((SUCCEEDED(hr)) && (NULL == pIHtmlElement)) {

hr = HRESULT_FROM_WIN32(ERROR_NOT_READY);

goto FINAL;

}

pIHtmlElement = NULL;

ASSERT(FALSE);

goto FINAL;

}

hr = pIHtmlElement -> QueryInterface(IID_IHTMLElement2, (VOID ** ) & pIHtmlElement2);

if (FAILED(hr)) {

pIHtmlElement2 = NULL;

ASSERT(FALSE);

goto FINAL;

}

hr = pIHtmlDocument2 -> createElement(bstrElementType, & pIHtmlElementScript);

if (FAILED(hr)) {

pIHtmlElementScript = NULL;

ASSERT(FALSE);

goto FINAL;

}

hr = pIHtmlElementScript -> QueryInterface(IID_IHTMLScriptElement, (VOID ** ) & pIHtmlScript);

if (FAILED(hr)) {

pIHtmlScript = NULL;

ASSERT(FALSE);

goto FINAL;

}

hr = pIHtmlScript -> put_defer(VARIANT_TRUE);

if (FAILED(hr)) {

ASSERT(FALSE);

goto FINAL;

}

hr = pIHtmlScript -> put_text(bstrScript);

if (FAILED(hr)) {

ASSERT(FALSE);

goto FINAL;

}

hr = pIHtmlElement2 -> insertAdjacentElement(bstrInsertWhere, pIHtmlElementScript, NULL);

if (FAILED(hr)) {

ASSERT(FALSE);

goto FINAL;

}

// 여기까지 왔다면, 일단 성공~

hr = S_OK;

hr2 = pIHtmlDocument2 -> get_frames( & pIHtmlFramesCollection);

if (FAILED(hr2)) {

// Frame이 없는 깔끔한 경우이다~

goto FINAL;

}

hr2 = pIHtmlFramesCollection -> get_length( & nFrameCount);

if (FAILED(hr2)) {

goto FINAL;

}

for (i = 0; i < nFrameCount; i++) {

varIndex.vt = VT_I4;

varIndex.lVal = i;

if (NULL != varDispWin.pdispVal) {

varDispWin.pdispVal -> Release();

varDispWin.pdispVal = NULL;

}

hr2 = pIHtmlFramesCollection -> item( & varIndex, & varDispWin);

if (FAILED(hr2)) {

continue;

}

if (NULL == varDispWin.pdispVal) {

ASSERT(FALSE);

continue;

}

if (NULL != pIHtmlWindow) {

pIHtmlWindow -> Release();

pIHtmlWindow = NULL;

}

hr2 = varDispWin.pdispVal -> QueryInterface(IID_IHTMLWindow2, (VOID ** ) & pIHtmlWindow);

if (FAILED(hr2)) {

continue;

}

if (NULL == pIHtmlWindow) {

ASSERT(FALSE);

continue;

}

if (NULL != pIHtmlDocument2Frame) {

pIHtmlDocument2Frame -> Release();

pIHtmlDocument2Frame = NULL;

}

hr2 = pIHtmlWindow -> get_document( & pIHtmlDocument2Frame);

if (FAILED(hr2)) {

continue;

}

if (NULL == pIHtmlDocument2Frame) {

ASSERT(FALSE);

continue;

}

// Recursive 실행~!!!

hr2 = InjectJScriptByDocumentRecurse(pIHtmlDocument2Frame, lpszJScript, nMaxRecurseFrame, nCurDepth + 1, pbStackOverflow);

// StackOverflow

if (HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW) == hr2) {

( * pbStackOverflow) = TRUE;

hr = HRESULT_FROM_WIN32(ERROR_STACK_OVERFLOW);

break;

}

FINAL:

if (NULL != pIHtmlDocument2Frame) {

pIHtmlDocument2Frame -> Release();

pIHtmlDocument2Frame = NULL;

}

if (NULL != pIHtmlWindow) {

pIHtmlWindow -> Release();

pIHtmlWindow = NULL;

}

if (NULL != varDispWin.pdispVal) {

varDispWin.pdispVal -> Release();

varDispWin.pdispVal = NULL;

}

if (NULL != pIHtmlScript) {

pIHtmlScript -> Release();

pIHtmlScript = NULL;

}

if (NULL != pIHtmlElementScript) {

pIHtmlElementScript -> Release();

pIHtmlElementScript = NULL;

}

if (NULL != pIHtmlElement2) {

pIHtmlElement2 -> Release();

pIHtmlElement2 = NULL;

}

if (NULL != pIHtmlElement) {

pIHtmlElement -> Release();

pIHtmlElement = NULL;

}

::SysFreeString(bstrScript);::SysFreeString(bstrElementType);::SysFreeString(bstrInsertWhere);

return hr;

}

outlook내 icloud내 연락처의 표시방법 변경

Sub 표시방법변경()
    
    Dim objOL       As Outlook.Application
    Dim objNS       As Outlook.NameSpace
    Dim objContact  As Outlook.ContactItem
    Dim objItems    As Outlook.Items
    Dim objContactsFolder As Outlook.MAPIFolder
    Dim objICloudFolder As Outlook.MAPIFolder
    Dim obj         As Object
    Dim strFirstName As String
    Dim strLastName As String
    Dim strName     As String
    
    On Error Resume Next
    
    Set objOL = CreateObject("Outlook.Application")
    Set objNS = objOL.GetNamespace("MAPI")
    Set objICloudFolder = objNS.Folders("iCloud")
    Set objICloudContacts = objICloudFolder.Folders("연락처")
    Set objItems = objICloudContacts.Items
    
    For Each obj In objItems
        If obj.Class = olContact Then
            Set objContact = obj
            
            With objContact
                strFirstName = .FirstName
                strLastName = .LastName
                
                If (.FirstName <> "" And .LastName <> "") Then
                    .FullName = .LastName + .FirstName
                End If
                
                If (strLastName = "" And strFirstName = "") Then
                    strName = .CompanyName
                Else
                    strName = strLastName + strFirstName
                End If
                
                If (.JobTitle <> "인턴") Then
                    .FileAs = strName + IIf(.JobTitle <> "" And .FirstName <> "", " " + .JobTitle + "님", " 님")
                Else
                    .FileAs = strName + " 님"
                End If
                
                If (.Email1Address <> "" And strName <> "") Then
                    .Email1DisplayName = .FileAs
                End If
                
                If (.Email2Address <> "" And strName <> "") Then
                    .Email2DisplayName = .FileAs
                End If
                
                If (.Email3Address <> "" And strName <> "") Then
                    .Email3DisplayName = .FileAs
                End If
                
                .Save
            End With
        End If
        
        Err.Clear
    Next
    
    Set objOL = Nothing
    Set objNS = Nothing
    Set obj = Nothing
    Set objContact = Nothing
    Set objItems = Nothing
    Set objICloudFolder = Nothing
    Set objICloudContacts = Nothing
    
End Sub

Sub 표시방법변경()

Dim objOL As Outlook.Application

Dim objNS As Outlook.NameSpace

Dim objContact As Outlook.ContactItem

Dim objItems As Outlook.Items

Dim objContactsFolder As Outlook.MAPIFolder

Dim objICloudFolder As Outlook.MAPIFolder

Dim obj As Object

Dim strFirstName As String

Dim strLastName As String

Dim strName As String

On Error Resume Next

Set objOL = CreateObject("Outlook.Application")

Set objNS = objOL.GetNamespace("MAPI")

Set objICloudFolder = objNS.Folders("iCloud")

Set objICloudContacts = objICloudFolder.Folders("연락처")

Set objItems = objICloudContacts.Items

For Each obj In objItems

If obj.Class = olContact Then

Set objContact = obj

With objContact

strFirstName = .FirstName

strLastName = .LastName

If (.FirstName <> "" And .LastName <> "") Then

.FullName = .LastName + .FirstName

End If

If (strLastName = "" And strFirstName = "") Then

strName = .CompanyName

Else

strName = strLastName + strFirstName

End If

If (.JobTitle <> "인턴") Then

.FileAs = strName + IIf(.JobTitle <> "" And .FirstName <> "", " " + .JobTitle + "님", " 님")

Else

.FileAs = strName + " 님"

End If

If (.Email1Address <> "" And strName <> "") Then

.Email1DisplayName = .FileAs

End If

If (.Email2Address <> "" And strName <> "") Then

.Email2DisplayName = .FileAs

End If

If (.Email3Address <> "" And strName <> "") Then

.Email3DisplayName = .FileAs

End If

.Save

End With

End If

Err.Clear

Set objOL = Nothing

Set objNS = Nothing

Set obj = Nothing

Set objContact = Nothing

Set objItems = Nothing

Set objICloudFolder = Nothing

Set objICloudContacts = Nothing

End Sub

anydragon

참고자료

최신 글

보관함

windows에서 쓸데 없이 업그레이드로 다운로드 받은 파일 삭제

CHtmlView의 C++에서 생성한 JScript를 모든 Frame에 대해 Inject하여 실행하기

outlook내 icloud내 연락처의 표시방법 변경

태그

링크

메타